10

我正在使用“mongodb”模块开发 nodejs / mongodb 应用程序。该应用程序启动与

node main.js

在 main.js 中，我连接到 db 并将连接保持在“db”全局变量中。然后在“服务器”的内部方法中使用“db”。我想避免将 'db' 作为全局变量，但没有找到正确的方法。

我当前的 main.js：

var server      = require('./lib/server');
var MongoClient = require('mongodb').MongoClient;
var Server      = require('mongodb').Server;
var mongoClient = new MongoClient(new Server(HOST, PORT));
db = null;

// Database connection
mongoClient.open(function(err, mongoClient) {
  if(!err){
    // Database selection
    db = mongoClient.db(DB);

    // Launch web server
    server.start(); // usage of 'db' in this part 

  } else {
    console.log(err.message);
    process.exit(1);
  }
});

任何想法更清洁的方式？

更新

我终于在 connection.js 中创建了一个模块：

var config      = require('../config/config');
var url         = 'mongodb://' + config.db.host + ':' + config.db.port + '/' + config.db.name;
var MongoClient = require('mongodb').MongoClient;
var db          = null;

module.exports = function(cb){
  if(db){
    cb(db);
    return;
  }

  MongoClient.connect(url, function(err, conn) {
    if(err){
      console.log(err.message);
      throw new Error(err);
    } else {
      db = conn; 
      cb(db);
    }
  });
}

每次我需要获得连接时，我都会调用：

var connection = require('./connection');
connection(function(db){
  // doing some stuff with the db
});

这工作得很好。

这种方法有任何潜在的失败吗？

4

2 回答 2

6

我通常包含一个项目实用程序文件，其中包含许多这些东西，只是为了方便。它作为一个伪全局函数起作用，但没有全局变量带来的许多常见问题。

例如，

项目Utils.js

module.exports = {

  initialize: function(next){
    // initialization actions, there can be many of these
    this.initializeDB(next);
  },

  initializeDb: function(next){
    mongoClient.open(function(err, mongoClient) {
      if(err) return next(err);
      module.exports.db = mongoClient.db(DB);
      next();
    });
  }
}

应用程序.js

var projectUtils = require('projectUtils');

// (snip)
projectUtils.initialize(function(err) {
  if(err) throw err; // bad DB initialization
  // After this point and inside any of your routes,
  // projectUtils.db is available for use.
  app.listen(port);
}

通过使用异步 initialize() 函数，您可以确保在启动服务器之前完成所有数据库连接、文件 I/O 等。

于 2013-08-02T03:48:32.003 回答
3

例如，您可以创建一个类似于提供程序的包装器并将其放在 provider.js 中。

Provider = function (db_name, host, port, username, password) {
    var that = this;
    var conn = generate_url(db_name, host, port, username, password); // you need to implement your version of generate_url()

    MongoClient.connect(conn, function (err, db) {
        if (err) {
            throw err;
        }
        that.db = db;
    });
};

//add the data access functions
Provider.prototype.getCollection = function (collectionName, callback) {
    this.db.collection(collectionName, collectionOptions, callback);
};

exports.Provider = Provider;

这是您使用提供程序的方式：

var db = new Provider(db_name, host, port, username, password);
db.getCollection('collection name', callback);
于 2013-08-01T22:09:04.530 回答