cgserver 6.0.7 → 6.0.8

package/dist/lib/Database/MongoManager.js

@@ -0,0 +1,401 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GMongoMgr = exports.MgReturn = exports.MrResult = exports.MongoBaseModel = void 0;
+const FrameworkConfig_1 = require("../Config/FrameworkConfig");
+const Log_1 = require("../Logic/Log");
+const mongo = require("mongodb");
+const _error_1 = require("../Config/_error_");
+const Core_1 = require("../Core/Core");
+class MongoBaseModel {
+    _id;
+}
+exports.MongoBaseModel = MongoBaseModel;
+class MrResult {
+    /**
+     * select 的数据量
+     */
+    length = 0;
+    /**
+     * 插入数据的自增id
+     */
+    insertId = null;
+    insertIds = null;
+    /**
+     * update 更新数据的影响条数
+     */
+    changedRows = null;
+    /**
+     * 插入或删除数据的影响条数
+     */
+    affectedRows = null;
+}
+exports.MrResult = MrResult;
+class MgReturn {
+    error = null;
+    result = new MrResult();
+    list = [];
+}
+exports.MgReturn = MgReturn;
+exports.GMongoMgr = null;
+class MongoManager {
+    _init_cbs = [];
+    _mongo = null;
+    get mongo() {
+        return this._mongo;
+    }
+    _mongo_init_succ = false;
+    _inited = false;
+    get isValid() {
+        return this._inited;
+    }
+    constructor() {
+    }
+    async init() {
+        if (this._inited) {
+            return false;
+        }
+        if (this._mongo) {
+            return true;
+        }
+        if (!FrameworkConfig_1.GFCfg.db.mongo || !FrameworkConfig_1.GFCfg.db.mongo.open) {
+            return true;
+        }
+        this._inited = true;
+        Log_1.GLog.info("mongo config=" + JSON.stringify(FrameworkConfig_1.GFCfg.db.mongo), true);
+        let client = new mongo.MongoClient("mongodb://" + FrameworkConfig_1.GFCfg.db.mongo.host + ":" + FrameworkConfig_1.GFCfg.db.mongo.port, FrameworkConfig_1.GFCfg.db.mongo.options);
+        await client.connect();
+        this.onConnect();
+        this._mongo = client.db(FrameworkConfig_1.GFCfg.db.mongo.database);
+        for (let i = 0; i < this._init_cbs.length; ++i) {
+            this._init_cbs[i]();
+        }
+        return true;
+    }
+    registerInitCb(cb) {
+        this._init_cbs.push(cb);
+    }
+    onConnect() {
+        this._mongo_init_succ = true;
+        Log_1.GLog.info("mogo has connected!");
+    }
+    onEnd() {
+        this._mongo_init_succ = false;
+        Log_1.GLog.info("mogo has ended!");
+    }
+    onError(err) {
+        Log_1.GLog.info("Error " + err);
+    }
+    async getAutoIds(key) {
+        if (!this._mongo) {
+            return -1;
+        }
+        let collection = "auto_ids";
+        let col = this._mongo.collection(collection);
+        try {
+            let where = this._convertWhere({ _id: key });
+            let rs = await col.findOneAndUpdate(where, { $inc: { id: 1 } }, { upsert: true });
+            if (rs.ok) {
+                return rs.value.id + 1;
+            }
+            return 0 + 1;
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+        }
+        return -1;
+    }
+    _convertWhere(where) {
+        if (!where || !where._id) {
+            return;
+        }
+        if (Core_1.core.isString(where._id) && where._id.length == 24) {
+            let _id = where._id;
+            try {
+                where._id = new mongo.ObjectId(_id);
+            }
+            catch (e) {
+                where._id = _id;
+            }
+        }
+        return where;
+    }
+    /**
+     * 获取单条消息
+     * @param collection
+     */
+    async findOne(collection, property = {}, where = {}, sort) {
+        this._convertWhere(where);
+        let rs = { errcode: null, one: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let one = null;
+        try {
+            let col = this._mongo.collection(collection);
+            if (!sort) {
+                one = await col.findOne(where, property);
+            }
+            else {
+                one = await (col.find(where, property).sort(sort).limit(1).next());
+            }
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.one = one;
+        return rs;
+    }
+    async findMany(collection, property = {}, where = {}, sort, skip = 0, limit = 0) {
+        this._convertWhere(where);
+        let rs = { errcode: null, list: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let list = [];
+        try {
+            let col = this._mongo.collection(collection);
+            let cursor = col.find(where, property);
+            if (sort) {
+                cursor = cursor.sort(sort);
+            }
+            if (skip) {
+                cursor = cursor.skip(skip);
+            }
+            if (limit) {
+                cursor = cursor.limit(limit);
+            }
+            list = await cursor.toArray();
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.list = list;
+        return rs;
+    }
+    async findCount(collection, property, where) {
+        this._convertWhere(where);
+        let rs = { errcode: null, count: -1 };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let count = -1;
+        try {
+            let col = this._mongo.collection(collection);
+            count = await col.countDocuments(where || {}, property);
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.count = count;
+        return rs;
+    }
+    async deleteOne(collection, where) {
+        this._convertWhere(where);
+        let rs = { errcode: null, count: -1 };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let del_rs = null;
+        try {
+            let col = this._mongo.collection(collection);
+            del_rs = await col.deleteOne(where || {});
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        if (del_rs) {
+            rs.count = del_rs.deletedCount;
+        }
+        return rs;
+    }
+    async deleteMany(collection, where) {
+        this._convertWhere(where);
+        let rs = { errcode: null, count: -1 };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let del_rs = null;
+        try {
+            let col = this._mongo.collection(collection);
+            del_rs = await col.deleteMany(where || {});
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        if (del_rs) {
+            rs.count = del_rs.deletedCount;
+        }
+        return rs;
+    }
+    /**
+     * 插入数据
+     * @param collection
+     * @param data
+     */
+    async insertOne(collection, data) {
+        let rs = { errcode: null, rs: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let in_rs = null;
+        try {
+            let col = this._mongo.collection(collection);
+            in_rs = await col.insertOne(data);
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.rs = in_rs;
+        return rs;
+    }
+    async insertManay(collection, data) {
+        let rs = { errcode: null, rs: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let in_rs = null;
+        try {
+            let col = this._mongo.collection(collection);
+            in_rs = await col.insertMany(data);
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.rs = in_rs;
+        return rs;
+    }
+    async updateOne(collection, model, where, upsert = false) {
+        let _id = model["_id"];
+        delete model["_id"];
+        this._convertWhere(where);
+        let rs = { errcode: null, rs: null };
+        if (!this._mongo) {
+            if (_id) {
+                model["_id"] = _id;
+            }
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let up_rs = null;
+        try {
+            let updateModel = null;
+            let firstKey = Object.keys(model)[0];
+            if (!firstKey.startsWith("$")) {
+                updateModel = { $set: model };
+            }
+            else {
+                updateModel = model;
+            }
+            let col = this._mongo.collection(collection);
+            up_rs = await col.updateOne(where, updateModel, { upsert: upsert });
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.rs = up_rs;
+        if (up_rs?.upsertedId || _id) {
+            model["_id"] = up_rs?.upsertedId || _id;
+        }
+        return rs;
+    }
+    async updateMany(collection, model, where, upsert = false) {
+        this._convertWhere(where);
+        let rs = { errcode: null, rs: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let up_rs = null;
+        try {
+            let updateModel = null;
+            let firstKey = Object.keys(model)[0];
+            if (!firstKey.startsWith("$")) {
+                updateModel = { $set: model };
+            }
+            else {
+                updateModel = model;
+            }
+            let col = this._mongo.collection(collection);
+            up_rs = await col.updateMany(where, updateModel, { upsert: upsert });
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.rs = up_rs;
+        return rs;
+    }
+    async createIndex(collection, index, callback) {
+        let rs = { errcode: null, rs: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let i_rs = null;
+        try {
+            let col = this._mongo.collection(collection);
+            i_rs = await col.createIndex(index, callback);
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.rs = i_rs;
+        return rs;
+    }
+    async simpleAggregate(collection, property = {}, where = {}, size, random_size) {
+        this._convertWhere(where);
+        let rs = { errcode: null, list: null };
+        if (!this._mongo) {
+            rs.errcode = _error_1.EErrorCode.No_Mongo;
+            return rs;
+        }
+        let list = [];
+        try {
+            let col = this._mongo.collection(collection);
+            let params = [];
+            params.push({ '$match': where || {} });
+            params.push({ '$project': property || {} });
+            if (random_size) {
+                params.push({ '$sample': { 'size': random_size } });
+            }
+            let agg = col.aggregate(params);
+            if (size) {
+                list = await agg.limit(size).toArray();
+            }
+            else {
+                list = await agg.toArray();
+            }
+        }
+        catch (e) {
+            Log_1.GLog.error(e);
+            rs.errcode = _error_1.EErrorCode.Mongo_Error;
+        }
+        rs.list = list;
+        return rs;
+    }
+    aggregate(collection, pipeline, options) {
+        if (!this._mongo) {
+            return;
+        }
+        let col = this._mongo.collection(collection);
+        let agg = col.aggregate(pipeline, options);
+        return agg;
+    }
+}
+exports.GMongoMgr = new MongoManager();

package/dist/lib/Database/MysqlManager.js

@@ -0,0 +1,180 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GMysqlMgr = exports.SqlReturns = exports.SqlReturn = exports.SqlResult = void 0;
+const _error_1 = require("./../Config/_error_");
+const Log_1 = require("../Logic/Log");
+const FrameworkConfig_1 = require("../Config/FrameworkConfig");
+const mysql = require("mysql");
+class SqlResult {
+    /**
+     * select 的数据量
+     */
+    length = 0;
+    /**
+     * 插入数据的自增id
+     */
+    insertId = null;
+    /**
+     * update 更新数据的影响条数
+     */
+    changedRows = null;
+    /**
+     * 插入或删除数据的影响条数
+     */
+    affectedRows = null;
+}
+exports.SqlResult = SqlResult;
+class SqlReturn {
+    error = null;
+    results = null;
+    fields = null;
+    list = null;
+}
+exports.SqlReturn = SqlReturn;
+class SqlReturns {
+    error = null;
+    srs = [];
+}
+exports.SqlReturns = SqlReturns;
+exports.GMysqlMgr = null;
+class MysqlManager {
+    _init_cbs = [];
+    _pool = null;
+    get isValid() {
+        return !!this._pool;
+    }
+    constructor() {
+    }
+    async init() {
+        return new Promise(async (resolve) => {
+            if (this._pool
+                || !FrameworkConfig_1.GFCfg.db.mysql
+                || !FrameworkConfig_1.GFCfg.db.mysql.open) {
+                resolve(null);
+                return;
+            }
+            this._pool = mysql.createPool({
+                connectionLimit: 100,
+                host: FrameworkConfig_1.GFCfg.db.mysql.host,
+                port: FrameworkConfig_1.GFCfg.db.mysql.port,
+                user: FrameworkConfig_1.GFCfg.db.mysql.user,
+                password: FrameworkConfig_1.GFCfg.db.mysql.password,
+                database: FrameworkConfig_1.GFCfg.db.mysql.database,
+                supportBigNumbers: true,
+                charset: "utf8mb4"
+            });
+            console.log("mysql config=" + JSON.stringify(FrameworkConfig_1.GFCfg.db.mysql));
+            //这个的初始化位置不能变，必须位于cbs前，pool后
+            await DBCache_1.GDBCache.init();
+            resolve(null);
+            for (let i = 0; i < this._init_cbs.length; ++i) {
+                this._init_cbs[i]();
+            }
+        });
+    }
+    registerInitCb(cb) {
+        this._init_cbs.push(cb);
+    }
+    query(sqlStr, values, no_err_log) {
+        return new Promise((resolve, reject) => {
+            let sr = new SqlReturn();
+            if (!this._pool) {
+                //表示没有开通数据库，不用记录错误日志
+                sr.error = _error_1.EErrorCode.No_Mysql;
+                Log_1.GLog.error(sr.error);
+                resolve(sr);
+                return;
+            }
+            this._pool.getConnection((err, conn) => {
+                if (err) {
+                    sr.error = err;
+                    Log_1.GLog.error(err);
+                    resolve(sr);
+                    return;
+                }
+                conn.query(sqlStr, values, (error, results, fields) => {
+                    sr.error = error;
+                    sr.results = results;
+                    if (sr.results && sr.results.length > 0) {
+                        sr.list = [];
+                        for (let i = 0; i < sr.results.length; ++i) {
+                            sr.list.push(sr.results[i]);
+                        }
+                    }
+                    sr.fields = fields;
+                    conn.release();
+                    if (error && !no_err_log) {
+                        Log_1.GLog.error(error);
+                    }
+                    resolve(sr);
+                });
+            });
+        });
+    }
+    transaction(sqls) {
+        return new Promise((resolve, reject) => {
+            let srs = new SqlReturns();
+            this._pool.getConnection((err, conn) => {
+                if (err) {
+                    srs.error = err;
+                    Log_1.GLog.error(err);
+                    resolve(srs);
+                    return;
+                }
+                conn.beginTransaction((err) => {
+                    if (err) {
+                        srs.error = err;
+                        Log_1.GLog.error(err);
+                        conn.release();
+                        resolve(srs);
+                        return;
+                    }
+                    let funcs = [];
+                    for (let i = 0; i < sqls.length; ++i) {
+                        let sql = sqls[i];
+                        funcs.push(() => {
+                            conn.query(sql.sql, sql.values, (error, results, fields) => {
+                                srs.srs.push(results);
+                                if (error) {
+                                    return conn.rollback(() => {
+                                        Log_1.GLog.error(error);
+                                        srs.error = error;
+                                        conn.release();
+                                        resolve(srs);
+                                    });
+                                }
+                                else {
+                                    if (funcs.length > 0) {
+                                        let f = funcs.shift();
+                                        f();
+                                    }
+                                    else {
+                                        conn.commit((error) => {
+                                            if (error) {
+                                                return conn.rollback(() => {
+                                                    srs.error = error;
+                                                    Log_1.GLog.error(error);
+                                                    conn.release();
+                                                    resolve(srs);
+                                                });
+                                            }
+                                            else {
+                                                conn.release();
+                                                resolve(srs);
+                                            }
+                                        });
+                                    }
+                                }
+                            });
+                        });
+                    }
+                    let f = funcs.shift();
+                    f();
+                });
+            });
+        });
+    }
+}
+exports.GMysqlMgr = new MysqlManager();
+//该import必须放在GMysqlMgr实例化之后，否则version基类找不到GMysqlMgr
+const DBCache_1 = require("./Decorator/DBCache");