nodejs mysql db类_nodejs mongodb 数据库封装DB类 -转

使用到了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();

mongoose操作类(封装mongodb)

这个类库使用方法如下：

//先包含进来

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"

}

}