NodeJS中使用MongoDB的简单封装

1. 使用示例   
   
2. /*
   
3. 使用示例：
   
4. */
   
5. var DB = require('./DB.js').DB;
   
6. var ObjectId = require('mongodb').ObjectID;
   
7. var docName = 'User';
   
8. DB(docName, function (db, collection) {
   
9.         collection.bulkWrite([{insertOne:{document:message}}], function (err, result) {
   
10.                 assert.equal(null, err);
    
11.                 invokeCallback.call(callback, result);
    
12.                 db.close();
    
13.         });
    
14. });
    
15. 连接封装部分   
    
16. var mongo = require('mongodb').MongoClient;
    
17. var assert = require('assert');
    
18. 
    
19. var url = 'mongodb://localhost:27017/test';
    
20. 
    
21. module.exports.DB = function (name, callback) {
    
22.     mongo.connect(url, function (err, db) {
    
23.         assert.equal(null, err);
    
24.         var collection = db.collection(name);
    
25.         if (typeof collection === 'undefined') {
    
26.             db.createCollection(name);
    
27.             collection = db.collection(name);
    
28.         }
    
29.         if (typeof callback === 'function') {
    
30.             callback(db, collection);
    
31.         }
    
32.     });
    
33. };

复制代码

来源：http://www.open-open.com/code/view/1454897340917