mongo db

mongo.js

var MongoClient = require('mongodb').MongoClient;
var mg = require('../config').mongodb;

var DB = function(){
  this.url = "mongodb://" + mg['host'] + ":" + mg['port'] + "/" + mg['database'];
};

module.exports = new DB;

DB.prototype.connect = function(cb){
  MongoClient.connect(this.url, function(err, db) {
    if(err) {
      console.log(err);
    } else {
      cb && cb(db);
    }
  });
};

DB.prototype.insert = function(table, obj, cb){
  this.connect(function(db){
    db.collection(table).insertOne(obj, function(err, data){
      if(err) {
        console.log(err);
      } else {
        cb && cb(data);
        db.close();
      }
    });
  });
};

DB.prototype.find = function(table, condition, cb) {
  this.connect(function(db) {
    var data = db.collection(table).find(condition);
    cb && cb(data);
    db.close();
  });
};

DB.prototype.update = function(table, condition, obj){
  this.connect(function(db){
    db.collection(table).updateOne(condition, obj, function(err, data) {
      if(err) {
        console.log(err);
      } else {
        cb && cb(data);
        db.close();
      }
    });
  });
};

DB.prototype.remove = function(table, condition) {
  this.connect(function(db){
    db.collection(table).deleteMany(condition, function(err, data) {
      if(err) {
        console.log(err);
      } else {
        cb && cb(data);
        db.close();
      }
    });
  });
};