monogram/lib/database.js

TAO (aspect-oriented) modeling for MongoDB

const Collection = require('./collection');
const { MongoClient } = require('mongodb');
const co = require('co');

class Database {
  static connect(uri, options) {
    return co(function * () {
      const db = new Database();
      const dbOrClient = yield MongoClient.connect(uri, options);
      db.db = dbOrClient.constructor.name === 'MongoClient' ?
        dbOrClient.db() :
        dbOrClient;
      db.client = dbOrClient.constructor.name === 'MongoClient' ?
        dbOrClient :
        null;
      db.collections = {};
      return db;
    });
  }

  collection(name, archetype) {
    if (this.collections[name]) {
      return this.collections[name];
    }
    this.collections[name] = new Collection(this.db.collection(name), archetype, { isMongoDBDriver3: this.client != null });
    return this.collections[name];
  }

  dropDatabase() {
    return this.db.dropDatabase();
  }
}

module.exports = Database;