封装mongodb数据库操作

let MongoClient = require('mongodb').MongoClient;
let url = 'mongodb://localhost:27017';
class DB {
  constructor(database,collection){
    this.database = database;
    this.collection = collection;
    this.connection;
    this.db;
  }

  /**
   * 
   * @returns 返回连接数据库成功后的响应
   */
  start(){
    return new Promise((res,rej) => {
      MongoClient.connect(url,(err,db) => {
        if(err) throw err;
        this.db = db;
        this.connection = db.db(this.database).collection(this.collection);
        res();
      })
    })
  }

  /**
   * 
   * @param {插入对象} obj 
   */
  insert(obj){
    this.start().then(() => {
      if(!(obj instanceof Array)){
        obj = [obj]
      }
      this.connection.insertMany(obj,(err,res) => {
        if(err) throw err;
        console.log('插入成功');
        this.db.close();
      })
    });
  }

  /**
   * 
   * @param {条件语句} where 
   * @param {修改后数据} data 
   * @param {是否全部更新} flag 
   */
  update(where,data,flag){
    this.start().then( () => {
      flag = flag ? true : false;
      this.connection.update(where,{$set:data},{multi: flag},(err,res) => {
        if(err) throw err;
        flag ? console.log('更新多行成功') : console.log('更新一条成功');
        this.db.close();
      })
    })
  }
  
  /**
   * 
   * @param {条件语句} where 
   * @param {是否多行删除} flag 
   */
  delete(where,flag){
    this.start().then(() => {
      if(!flag){
        this.connection.deleteOne(where,(err,res) => {
          if(err) throw err;
          console.log('删除单个成功')
          this.db.close();
        })
      }else{
        this.connection.deleteMany(where,(err,res) => {
          if(err) throw err;
          console.log('删除多个成功')
          this.db.close();
        })
      }
    })
  }

  /**
   * 
   * @param {条件语句} where 
   * @param {排序方式} sort 
   * @param {限制返回条数} limit 
   * @param {跳过返回条数} skip 
   */
  find(where,sort,limit,skip){
    sort = sort || 1;
    limit = limit || 0;
    skip = skip || 0;
    this.start().then( () => {
      this.connection.find(where,(err,res) => {
        if(err) throw err;
        res.each((error,data) => {
          for(let key in data){
            console.log(`${key}:${data[key]}`);
          }
        })
      }).sort(sort).limit(limit).skip(skip)
    })
  }
}

export default DB;