Mongodb 使用可选方法连接对象

Question

我想创建一个 mongodb 对象，它有一个可选的方法来在连接后创建 Collection。

我有这样的东西（mongo.js）：

import url from 'url';
import { MongoClient } from 'mongodb';

const getDbName = mongourl => {
  const mongoPathName = url.parse(mongourl).pathname;
  return mongoPathName.substring(mongoPathName.lastIndexOf('/') + 1);
}

export let db = null;

(async function() {

  const dbURL = process.env.MONGO_URL;
  const dbName = getDbName(dbURL);

  const client = new MongoClient(dbURL, { useNewUrlParser: true });

  try {

    await client.connect();
    db = client.db(dbName);

  } catch (err) {
    console.log(err);
  }

})();

如果我想连接到 db，我只是这样做：

import { db } from './libs/mongo';

但我的问题是如何添加可选方法，以便在连接到数据库后创建集合？ 我想得到类似的东西：

import { db } from './libs/mongo';

db.then(() => { // if db is connected ->
 //createCollection or something else
})

我想用 Promise 来做，但我不知道怎么做。 我想至少在两个地方使用这个库。 1、与base连接，在第二个（单独）连接并创建集合（setup）

Answer 1

您的import 本身不应引起副作用。

在mongo.js 模块中，我建议导出一个包含async 方法的类或对象以返回数据库连接。

这是我在自己的一个项目中使用的方法，其中构造函数返回一个新对象，该对象最初只有一个 connect 方法，当调用该方法时，会将其他与数据库相关的函数附加到该对象。

class Database {

    constructor(settings) {
        this.connect = async () => {
            let db = await MongoClient.connect(url, settings);

            this.someDbMethod = () => {
                return db.collection('foo').find(...);
            }

            return this;
        }
    }
}

有用法：

let db = await new Database(url, settings).connect();
let rec = await db.someDbMethod();

构造函数不能是异步的，因此我使用辅助函数来执行实际连接。