Question

我正在使用Node anb Mongodb Native来建立应用程序。我正在处理一个db模块，我可以要求并在其他模块中调用，这样我最终只使用一个连接。模块db.js从这段代码开始：

var _db = null;
var getDb = module.exports.getDb = function(callback) {
  if (_db) {
    console.log('_db returned');
    return callback(null, _db);
  }
  MongoClient.connect('mongodb://localhost:' + config.db.port + '/' + config.db.name, {native_parser: true}, function (err, db) {
    if (err) return callback(err);
    console.log('_db created');

    _db = db;
    callback(err, _db);
    });
};

在我需要数据库连接的其他模块中，我这样做

db.getDb(function (err, connection) {
  // Do something with connection
});

工作正常。但令人不快的问题是，如果我的代码在很短的时间内多次调用getDb，我最终会得到几个连接副本。就像我在需要数据库连接的所有模块的最开始时进行db.js要求和getDb调用一样

我现在考虑通过排队来控制对getDb的调用，这样只有绝对的第一次调用才能创建连接并将其保存在_db中。所有后来的调用都将获得创建的连接_db作为回报。我相信Async queue会帮助我...

问题是，我不明白我是如何使用异步队列编写的。文档有点模糊，我在网上找不到更好的例子。也许你可以给我一些提示。这就是我到目前为止所得到的......

var dbCalls = async.queue(function (task, callback) {
  if (_db) {
    console.log('_db returned');
    return callback(null, _db);
  }
  MongoClient.connect('mongodb://localhost:' + config.db.port + '/' + config.db.name, {native_parser: true}, function (err, db) {
    if (err) return callback(err);
    console.log('Connected to mongodb://localhost:' + config.db.port + '/' + config.db.name);

    _db = db;
    callback(null, _db);
  });
}, 1);

// I guess this .push() must be the exposed (exported) API for other modules to get a connection, but how do I return it to them,  
dbCalls.push(null, function (err) {
  console.log('finished processing foo');
});

dbCalls.push(null, function (err) {
  console.log('finished processing bar');
});

我不理解作为第一个参数传递给.push()的对象如果适用，我应该使用什么？现在它的null如何将连接和可能的错误一直传递给进行调用的模块？

Answer 1

没有async.queue的快速而肮脏的解决方案：

var _db      = null;
var _err     = null;
var _queue   = [];
var _pending = false;

var getDb = module.exports.getDb = function(callback) {
  if (_err || _db) {
    console.log('_db returned');
    return callback(_err, _db);
  } else if (_pending) { // already a connect() request pending
    _queue.push(callback);
  } else {
    _pending = true;
    _queue.push(callback);
    MongoClient.connect(..., function (err, db) {
      _err = err;
      _db = db;
      _queue.forEach(function(queuedCallback) {
        queuedCallback(err, db);
      });
    });
};

使用异步队列控制数据库连接请求

1 个答案: