Node.js Mongodb本机驱动程序连接共享

时间:2012-08-20 12:29:13

标签: javascript node.js mongodb

main.js

var http = require('http');
var UserModel = require('./models/user.js');
var server = http.createServer(function(req, res){
  UserModel.create({
  }), function(e, o){
    if(e) { console.log(e); } else {
    } console.log(o); }
  });
}).listen(3000);

connections.js

var mongo = require('mongodb');

module.exports = {
    dbMain: new mongo.Db('main', new mongo.Server('127.0.0.1', 27017, { auto_reconnect: true }, {})),
    dbLog: new mongo.Db('log', new mongo.Server('127.0.0.1', 27017, { auto_reconnect: true }, {}))
};

/models/user.js

var mongodb = require('mongodb');
var db = require('./connections.js').dbMain;

module.exports = {
  create: function(newData, callback){
    db.open(function(e, db){
      db.collection('users', function(e, collection){
        collection.insert(newData, callback);
      });
    });
  }
}

当我使用上面的代码时,服务器崩溃的问题是,在请求进入的第二次,我们仍然打开了数据库连接,所以我们将db.close添加到我们的Users.create函数中。

  create: function(newData, callback){
    db.open(function(e, db){
      db.collection('users', function(e, collection){
        collection.insert(newData, function(e, o){
          db.close(); // Voila.
          callback(e, o);
        });
      });
    });
  }

在这个阶段,服务器CAN仍然崩溃,因为多个连接打开,我不明白为什么或如何发生这种情况但是确实如此。

如何将我的项目组织到模型中(我不想使用Mongoose,我的验证是在不同的模型中进行的,所以Mongoose对我来说是一种矫枉过正)?另外我如何处理项目中的连接?

2 个答案:

答案 0 :(得分:8)

你可以拥有一个很好地包装所有这一切的库 - 这意味着只有一个与数据库的连接将被打开,并且第二个请求将返回相同的(开放)连接 - 如果你每秒获得1000+ ,这是一个成败的问题(即不重新打开每个请求的连接)......

<强> users.js

var connections = require('./connections.js');

var serverCache = connections('127.0.0.1', 27017); 

module.exports = {
  create: function(newData, callback){
    serverCache('main', 'users', function(e, collection){
      collection.insert(newData, callback);
    })
  }
}

<强> connections.js

var mongo = require('mongodb');

// a mongo connection cache
// pass in host & port
// it returns a function accepting dbName, collectionName & callback
var mongoCache = function(host, port){

  // keep our open connections
  var mongoDatabases = {};

  var ensureDatabase = function(dbName, readyCallback){
    // check if we already have this db connection open
    if(mongoDatabases[dbName]){
      readyCallback(null, mongoDatabases[dbName]);
      return;
    }

    // get the connection
    var server = new mongo.Server(host, port, {auto_reconnect: true});

    // get a handle on the database
    var db = new mongo.Db(dbName, server);
    db.open(function(error, databaseConnection){
      if(error) throw error;

      // add the database to the cache
      mongoDatabases[dbName] = databaseConnection;

      // remove the database from the cache if it closes
      databaseConnection.on('close', function(){
        delete(mongoDatabases[dbName]);
      })

      // return the database connection
      readyCallback(error, databaseConnection);
    })
  }

  var ensureCollection = function(dbName, collectionName, readyCallback){

    ensureDatabase(dbName, function(error, databaseConnection){
      if(error) throw error;

      databaseConnection.createCollection(collectionName, function(error, collection) {
        if(error) throw error;

        // return the collection finally
        readyCallback(error, collection);
      })

    })
  }

  return ensureCollection;
}

module.exports = mongoCache;

答案 1 :(得分:2)

我目前正在使用具有多个http请求的全局连接。在过去,我创建了一个复杂的库,它创建了几个与MongoDB的连接,并为每个连接随机选择一个。

后来我发现本机驱动程序可以为我做这个,这很漂亮。目前我正在使用单个对象,驱动程序选择将连接发送给查询。

var srvOpts = {
    auto_reconnect: true,
    poolSize: 10,
};

var conn = new Mongo.Server("localhost", 27017, srvOpts),
    db = new Mongo.Db("dbname", conn, {});

db.open(function (){});

你可以这是一个好主意,我想把这个想法复制到我正在使用的Redis驱动程序中,但是我的时间很短,所以我怀疑我会很快做的。