NodeJS中使用MongoDB的簡單封裝

使用示例    

/* 
使用示例：
*/
var DB = require('./DB.js').DB;
var ObjectId = require('mongodb').ObjectID;
var docName = 'User';
DB(docName, function (db, collection) {
    collection.bulkWrite([{insertOne:{document:message}}], function (err, result) {
        assert.equal(null, err);
        invokeCallback.call(callback, result);
        db.close();
    });
});

連接封裝部分    

var mongo = require('mongodb').MongoClient;
var assert = require('assert');

var url = 'mongodb://localhost:27017/test';

module.exports.DB = function (name, callback) {
    mongo.connect(url, function (err, db) {
        assert.equal(null, err);
        var collection = db.collection(name);
        if (typeof collection === 'undefined') {
            db.createCollection(name);
            collection = db.collection(name);
        }
        if (typeof callback === 'function') {
            callback(db, collection);
        }
    });
};