nodejs操作mongodb資料庫封裝DB類
阿新 • • 發佈:2018-12-01
我使用到了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往往需要重啟遊戲服務的。