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

这个DB类也算是我经历了3个实际项目应用的，现分享出来，有需要的请借鉴批评。

上面的注释都挺详细的，我使用到了nodejs的插件mongoose，用mongoose操作mongodb其实蛮方便的。

关于mongoose的安装就是 npm install -g mongoose

这个DB类的数据库配置是基于auth认证的，如果您的数据库没有账号与密码则留空即可。

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;
}

/**

更新数据
@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);
}
});
};

/**

更新数据方法(带操作符的)
@param table_name 数据表名
@param conditions 更新条件 {_id: id,user_name: name}
@param update_fields 更新的操作符 {$set: {id: 123}}
@param callback 回调方法
*/
DB.prototype.updateData = function (table_name,callback) {
if (!update_fields || !conditions) {
if (callback) callback({msg: 'Parameter error'});
return;
}
var node_model = this.getConnection(table_name);
node_model.findOneAndUpdate(conditions,data) {
if (callback) callback(err,data);
});
};

/**

删除数据
@param table_name 表名
@param conditions 删除需要的条件 {_id: id}
@param callback 回调方法
*/
DB.prototype.remove = function (table_name,callback) {
var node_model = this.getConnection(table_name);
node_model.remove(conditions,res);
}
});
};

/**

查询数据
@param table_name 表名
@param conditions 查询条件
@param fields 待返回字段
@param callback 回调方法
*/
DB.prototype.find = function (table_name,callback) {
var node_model = this.getConnection(table_name);
node_model.find(conditions,fields || null,{},res) {
if (err) {
callback(err);
} else {
callback(null,res);
}
});
};

/**

查询单条数据
@param table_name 表名
@param conditions 查询条件
@param callback 回调方法
*/
DB.prototype.findOne = function (table_name,callback) {
var node_model = this.getConnection(table_name);
node_model.findOne(conditions,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,res){
if (err) {
callback(err);
} else {
callback(null,res);
}
});
};

/**

返回符合条件的文档数
@param table_name 表名
@param conditions 查询条件
@param callback 回调方法
*/
DB.prototype.count = function (table_name,callback) {
var node_model = this.getConnection(table_name);
node_model.count(conditions,res);
}
});
};

/**

查询符合条件的文档并返回根据键分组的结果
@param table_name 表名
@param field 待返回的键值
@param conditions 查询条件
@param callback 回调方法
*/
DB.prototype.distinct = function (table_name,field,callback) {
var node_model = this.getConnection(table_name);
node_model.distinct(field,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,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();

这个类库使用方法如下：

//查询一条数据
MongoDB.findOne('user_info',{_id: user_id},res) {
console.log(res);
});

//查询多条数据
MongoDB.find('user_info',{type: 1},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 是数据库集合的配置项，结构如下：

记得每次给添加字段时，要往这个table.json里面添加。由于nodejs这个服务器的改动，更改table.json往往需要重启游戏服务的。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持编程之家。

原文链接：https://www.f2er.com/nodejs/40091.html

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

猜你在找的Node.js相关文章