nodejs 작업 mongodb 데이터베이스 패키지 DB 클래스

10450 단어 nodejsmongodb
이 DB 류 도 제 가 3 개의 실제 프로젝트 응용 을 겪 은 셈 입 니 다.지금 공유 하고 필요 한 것 은 비판 을 참고 하 십시오.
위의 주석 은 모두 매우 상세 합 니 다.저 는 nodejs 플러그 인 mongoose 를 사 용 했 습 니 다.mongoose 로 mongodb 를 조작 하 는 것 이 매우 편리 합 니 다.
mongoose 에 대한 설 치 는 npm install-g mongoose 입 니 다.
이 DB 클래스 의 데이터베이스 설정 은 auth 인증 을 기반 으로 합 니 다.데이터베이스 에 계 정과 비밀번호 가 없 으 면 비어 있 으 면 됩 니 다.

/**
 * mongoose   (  mongodb)
 */

var fs = require('fs');
var path = require('path');
var mongoose = require('mongoose');
var logger = require('pomelo-logger').getLogger('mongodb-log');

var options = {
  db_user: "game",
  db_pwd: "12345678",
  db_host: "192.168.2.20",
  db_port: 27017,
  db_name: "dbname"
};

var dbURL = "mongodb://" + options.db_user + ":" + options.db_pwd + "@" + options.db_host + ":" + options.db_port + "/" + options.db_name;
mongoose.connect(dbURL);

mongoose.connection.on('connected', function (err) {
  if (err) logger.error('Database connection failure');
});

mongoose.connection.on('error', function (err) {
  logger.error('Mongoose connected error ' + err);
});

mongoose.connection.on('disconnected', function () {
  logger.error('Mongoose disconnected');
});

process.on('SIGINT', function () {
  mongoose.connection.close(function () {
    logger.info('Mongoose disconnected through app termination');
    process.exit(0);
  });
});

var DB = function () {
  this.mongoClient = {};
  var filename = path.join(path.dirname(__dirname).replace('app', ''), 'config/table.json');
  this.tabConf = JSON.parse(fs.readFileSync(path.normalize(filename)));
};

/**
 *    mongoose model
 * @param table_name    (    )
 */
DB.prototype.getConnection = function (table_name) {
  if (!table_name) return;
  if (!this.tabConf[table_name]) {
    logger.error('No table structure');
    return false;
  }

  var client = this.mongoClient[table_name];
  if (!client) {
    //         
    var nodeSchema = new mongoose.Schema(this.tabConf[table_name]);

    //  model
    client = mongoose.model(table_name, nodeSchema, table_name);

    this.mongoClient[table_name] = client;
  }
  return client;
};

/**
 *     
 * @param table_name   
 * @param fields    
 * @param callback     
 */
DB.prototype.save = function (table_name, fields, callback) {
  if (!fields) {
    if (callback) callback({msg: 'Field is not allowed for null'});
    return false;
  }

  var err_num = 0;
  for (var i in fields) {
    if (!this.tabConf[table_name][i]) err_num ++;
  }
  if (err_num > 0) {
    if (callback) callback({msg: 'Wrong field name'});
    return false;
  }

  var node_model = this.getConnection(table_name);
  var mongooseEntity = new node_model(fields);
  mongooseEntity.save(function (err, res) {
    if (err) {
      if (callback) callback(err);
    } else {
      if (callback) callback(null, res);
    }
  });
};

/**
 *     
 * @param table_name   
 * @param conditions         {_id: id, user_name: name}
 * @param update_fields        {age: 21, sex: 1}
 * @param callback     
 */
DB.prototype.update = function (table_name, conditions, update_fields, callback) {
  if (!update_fields || !conditions) {
    if (callback) callback({msg: 'Parameter error'});
    return;
  }
  var node_model = this.getConnection(table_name);
  node_model.update(conditions, {$set: update_fields}, {multi: true, upsert: true}, function (err, res) {
    if (err) {
      if (callback) callback(err);
    } else {
      if (callback) callback(null, res);
    }
  });
};

/**
 *       (     )
 * @param table_name     
 * @param conditions      {_id: id, user_name: name}
 * @param update_fields        {$set: {id: 123}}
 * @param callback     
 */
DB.prototype.updateData = function (table_name, conditions, update_fields, callback) {
  if (!update_fields || !conditions) {
    if (callback) callback({msg: 'Parameter error'});
    return;
  }
  var node_model = this.getConnection(table_name);
  node_model.findOneAndUpdate(conditions, update_fields, {multi: true, upsert: true}, function (err, data) {
    if (callback) callback(err, data);
  });
};

/**
 *     
 * @param table_name   
 * @param conditions         {_id: id}
 * @param callback     
 */
DB.prototype.remove = function (table_name, conditions, callback) {
  var node_model = this.getConnection(table_name);
  node_model.remove(conditions, function (err, res) {
    if (err) {
      if (callback) callback(err);
    } else {
      if (callback) callback(null, res);
    }
  });
};

/**
 *     
 * @param table_name   
 * @param conditions     
 * @param fields      
 * @param callback     
 */
DB.prototype.find = function (table_name, conditions, fields, callback) {
  var node_model = this.getConnection(table_name);
  node_model.find(conditions, fields || null, {}, function (err, res) {
    if (err) {
      callback(err);
    } else {
      callback(null, res);
    }
  });
};

/**
 *       
 * @param table_name   
 * @param conditions     
 * @param callback     
 */
DB.prototype.findOne = function (table_name, conditions, callback) {
  var node_model = this.getConnection(table_name);
  node_model.findOne(conditions, function (err, res) {
    if (err) {
      callback(err);
    } else {
      callback(null, res);
    }
  });
};

/**
 *   _id       
 * @param table_name   
 * @param _id         ObjectId   。
 * @param callback     
 */
DB.prototype.findById = function (table_name, _id, callback) {
  var node_model = this.getConnection(table_name);
  node_model.findById(_id, function (err, res){
    if (err) {
      callback(err);
    } else {
      callback(null, res);
    }
  });
};

/**
 *           
 * @param table_name   
 * @param conditions     
 * @param callback     
 */
DB.prototype.count = function (table_name, conditions, callback) {
  var node_model = this.getConnection(table_name);
  node_model.count(conditions, function (err, res) {
    if (err) {
      callback(err);
    } else {
      callback(null, res);
    }
  });
};

/**
 *                     
 * @param table_name   
 * @param field       
 * @param conditions     
 * @param callback     
 */
DB.prototype.distinct = function (table_name, field, conditions, callback) {
  var node_model = this.getConnection(table_name);
  node_model.distinct(field, conditions, function (err, res) {
    if (err) {
      callback(err);
    } else {
      callback(null, res);
    }
  });
};

/**
 *     
 * @param table_name   
 * @param conditions      {a:1, b:2}
 * @param options   :{fields: "a b c", sort: {time: -1}, limit: 10}
 * @param callback     
 */
DB.prototype.where = function (table_name, conditions, options, callback) {
  var node_model = this.getConnection(table_name);
  node_model.find(conditions)
    .select(options.fields || '')
    .sort(options.sort || {})
    .limit(options.limit || {})
    .exec(function (err, res) {
      if (err) {
        callback(err);
      } else {
        callback(null, res);
      }
    });
};

module.exports = new DB();
이 라 이브 러 리 의 사용 방법 은 다음 과 같다.

//     
var MongoDB = require('./mongodb');

//      
MongoDB.findOne('user_info', {_id: user_id}, function (err, res) {
  console.log(res);
});

//      
MongoDB.find('user_info', {type: 1}, {}, function (err, res) {
  console.log(res);
});

//           
MongoDB.updateData('user_info', {_id: user_info._id}, {$set: update_data}, function(err, user_info) {
   callback(null, user_info);
});

//    
MongoDB.remove('user_data', {user_id: 1});

먼저 이런 예 를 들 어 더 많은 것 을 직접 시도 해 보 세 요!
그 중에서 설정 중의 config/table.json 은 데이터베이스 집합의 설정 항목 으로 구 조 는 다음 과 같다.

{
"user_stats_data": {
    "user_id": "Number",
    "platform": "Number",
    "user_first_time": "Number",
    "create_time": "Number"
  },
  "room_data": {
    "room_id": "String",
    "room_type": "Number",
    "user_id": "Number",
    "player_num": "Number",
    "diamond_num": "Number",
    "normal_settle": "Number",
    "single_settle": "Number",
    "create_time": "Number"
  },
  "online_data": {
    "server_id": "String",
    "pf": "Number",
    "player_num": "Number",
    "room_list": "String",
    "update_time": "Number"
  }
}

필드 를 추가 할 때마다 이 table.json 에 추가 하 는 것 을 기억 하 세 요.nodejs 이 서버 의 변경 으로 인해 table.json 을 변경 하려 면 게임 서 비 스 를 다시 시작 해 야 합 니 다.
이상 이 바로 본 고의 모든 내용 입 니 다.여러분 의 학습 에 도움 이 되 고 저 희 를 많이 응원 해 주 셨 으 면 좋 겠 습 니 다.

좋은 웹페이지 즐겨찾기