详解nodejs操作mongodb数据库封装DB类

这个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 + "https://www.jb51.net/" + 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();

这个类库使用方法如下:

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/wwywjp.html