1. 程式人生 > >nodejs操作mongodb資料庫封裝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();

這個類庫使用方法如下:

//先包含進來
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往往需要重啟遊戲服務的。