wok-server 0.1.0

package/dist/mysql/manager/ops/delete.js

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deleteMany = exports.deleteById = void 0;
+const exception_1 = require("../../exception");
+const utils_1 = require("../utils");
+const criteria_1 = require("./criteria");
+/**
+ * 按 id 删除.
+ *
+ * @param config
+ * @param connection
+ * @param mapping
+ * @param type
+ * @param id
+ */
+async function deleteById(config, connection, table, id) {
+    const res = await (0, utils_1.promiseQuery)(config, connection, `delete from ?? where ?? = ?`, [
+        table.tableName,
+        table.id,
+        id
+    ]);
+    const ok = res;
+    return ok.affectedRows === 1;
+}
+exports.deleteById = deleteById;
+/**
+ * 按条件删除，返回被删除的记录数.
+ * 务必谨慎使用，大批量删除容易带来性能问题，造成线上事故.
+ * @param config
+ * @param connection
+ * @param table
+ * @param criteria
+ */
+async function deleteMany(config, connection, opts) {
+    let sql = 'delete from ?? ';
+    const values = [opts.table.tableName];
+    let query = (0, criteria_1.buildQuery)(opts.criteria);
+    if (!query) {
+        throw new exception_1.MysqlException('No valid criteria specified.');
+    }
+    if (query) {
+        sql += ` where ${query.sql} `;
+        values.push(...query.values);
+    }
+    // 排序
+    if (opts.orderBy && opts.orderBy.length) {
+        opts.orderBy.forEach((orderBy, idx) => {
+            const [field, sort] = orderBy;
+            if (idx == 0) {
+                sql += ` order by ?? ${sort} `;
+            }
+            else {
+                sql += ` , ?? ${sort} `;
+            }
+            values.push(field);
+        });
+    }
+    // 数量限制
+    if (opts.limit) {
+        sql += ` limit ${opts.limit} `;
+    }
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    return res.affectedRows;
+}
+exports.deleteMany = deleteMany;

package/dist/mysql/manager/ops/exist.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.existsBy = exports.existsById = void 0;
+const criteria_1 = require("./criteria");
+const utils_1 = require("../utils");
+async function existsById(config, connection, table, id) {
+    const res = await (0, utils_1.promiseQuery)(config, connection, `select 1 from ?? where ?? = ?`, [
+        table.tableName,
+        table.id,
+        id
+    ]);
+    const rows = res;
+    return !!rows.length;
+}
+exports.existsById = existsById;
+async function existsBy(config, connection, table, criteria) {
+    let query = criteria ? (0, criteria_1.buildQuery)(criteria) : undefined;
+    let sql = `select 1 from ?? `;
+    if (query) {
+        sql += ` where ${query.sql} `;
+    }
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, [table.tableName].concat(query ? query.values : []));
+    const rows = res;
+    return !!rows.length;
+}
+exports.existsBy = existsBy;

package/dist/mysql/manager/ops/find.js

@@ -0,0 +1,111 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findFirst = exports.findByIdIn = exports.find = exports.findAll = exports.findById = void 0;
+const criteria_1 = require("./criteria");
+const utils_1 = require("../utils");
+/**
+ * 按 id 查询
+ * @param connection
+ * @param mapping
+ * @param type
+ * @param id
+ */
+async function findById(config, connection, table, id) {
+    const res = await (0, utils_1.promiseQuery)(config, connection, `select * from ?? where ?? = ?`, [
+        table.tableName,
+        table.id,
+        id
+    ]);
+    const rows = res;
+    if (!rows.length) {
+        return null;
+    }
+    return rows[0];
+}
+exports.findById = findById;
+/**
+ * 查询表中所有数据。需谨慎操作，全表查询有可能会产生大列表问题。
+ * @param connection
+ * @param mapping
+ * @param type
+ * @returns
+ */
+async function findAll(config, connection, table) {
+    const res = await (0, utils_1.promiseQuery)(config, connection, `select * from ??`, [table.tableName]);
+    return res;
+}
+exports.findAll = findAll;
+/**
+ * 条件查询
+ * @param opts
+ * @returns
+ */
+async function find(config, conn, opts) {
+    let query = opts.criteria ? (0, criteria_1.buildQuery)(opts.criteria) : undefined;
+    const values = [];
+    let sql = `select * from ?? `;
+    values.push(opts.table.tableName);
+    if (query) {
+        sql += ` where ${query.sql} `;
+        values.push(...query.values);
+    }
+    // 排序
+    if (opts.orderBy && opts.orderBy.length) {
+        opts.orderBy.forEach((orderBy, idx) => {
+            const [field, sort] = orderBy;
+            if (idx == 0) {
+                sql += ` order by ?? ${sort} `;
+            }
+            else {
+                sql += ` , ?? ${sort} `;
+            }
+            values.push(field);
+        });
+    }
+    // 数量限制
+    if (opts.limit) {
+        sql += ` limit ${opts.limit} `;
+        if (opts.offset) {
+            sql += ` offset ${opts.offset}`;
+        }
+    }
+    const res = await (0, utils_1.promiseQuery)(config, conn, sql, values);
+    return res;
+}
+exports.find = find;
+/**
+ * 根据 id 列表查询记录
+ * @param connection
+ * @param table
+ * @param ids
+ * @returns
+ */
+async function findByIdIn(config, connection, table, ids) {
+    if (!ids.length) {
+        return [];
+    }
+    return find(config, connection, {
+        table,
+        criteria: c => c.in(table.id, ids)
+    });
+}
+exports.findByIdIn = findByIdIn;
+/**
+ * 查找符合条件的第一条记录
+ * @param conn
+ * @param table
+ * @param criteria
+ * @returns
+ */
+async function findFirst(config, conn, table, criteria) {
+    let query = criteria ? (0, criteria_1.buildQuery)(criteria) : undefined;
+    let sql = `select * from ?? `;
+    if (query) {
+        sql += ` where ${query.sql} `;
+    }
+    sql += ' limit 1';
+    const res = await (0, utils_1.promiseQuery)(config, conn, sql, [table.tableName].concat(query ? query.values : []));
+    const list = res;
+    return list.length ? list[0] : null;
+}
+exports.findFirst = findFirst;

package/dist/mysql/manager/ops/index.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+// 各类操作封装
+tslib_1.__exportStar(require("./count"), exports);
+tslib_1.__exportStar(require("./criteria"), exports);
+tslib_1.__exportStar(require("./delete"), exports);
+tslib_1.__exportStar(require("./exist"), exports);
+tslib_1.__exportStar(require("./find"), exports);
+tslib_1.__exportStar(require("./insert"), exports);
+tslib_1.__exportStar(require("./modify"), exports);
+tslib_1.__exportStar(require("./paginate"), exports);
+tslib_1.__exportStar(require("./query"), exports);
+tslib_1.__exportStar(require("./update"), exports);

package/dist/mysql/manager/ops/insert.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.insertMany = exports.insert = void 0;
+const exception_1 = require("../../exception");
+const utils_1 = require("../utils");
+/**
+ * 为表插入数据
+ * @param connection
+ * @param table
+ * @param data
+ * @returns
+ */
+async function insert(config, connection, table, data) {
+    // 插入后的新数据
+    // 列信息，使用 set 防止 columns 中重复配置 id 和更新创建时间列
+    let columnsSet = new Set();
+    // 判定下 id ，如果有值，才在 insert 语句中出现 id 列，否则不出现
+    if (data[table.id]) {
+        columnsSet.add(table.id);
+    }
+    table.columns.forEach(col => columnsSet.add(col));
+    if (table.createdDate) {
+        const createdData = table.createdDate.type === 'date' ? new Date() : new Date().getTime();
+        data[table.createdDate.column] = createdData;
+        columnsSet.add(table.createdDate.column);
+    }
+    if (table.updatedDate) {
+        const updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime();
+        data[table.updatedDate.column] = updatedDate;
+        columnsSet.add(table.updatedDate.column);
+    }
+    const columns = Array.from(columnsSet);
+    // 构建 sql
+    const sql = `insert into ??(${columns.map(() => '??').join(',')}) values(${columns
+        .map(() => '?')
+        .join(',')})`;
+    const values = [table.tableName, ...columns, ...columns.map(col => data[col])];
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    if (packet.affectedRows !== 1) {
+        throw new exception_1.MysqlException(`Insert failed，table：${table.tableName}，primary key: ${data[table.id]}`);
+    }
+    // 自动生成的id处理
+    if (packet.insertId && (data[table.id] === undefined || data[table.id] === null)) {
+        data[table.id] = packet.insertId;
+    }
+    return data;
+}
+exports.insert = insert;
+/**
+ * 一次插入多条记录
+ * @param connection 连接
+ * @param table 表
+ * @param list 要插入的记录列表
+ */
+async function insertMany(config, connection, table, list) {
+    if (!list.length) {
+        return;
+    }
+    // 列信息，使用 set 防止 columns 中重复配置 id 和更新创建时间列
+    let columnsSet = new Set();
+    // 批量插入必须统一使用 id 或不使用 id，以第一条记录为准
+    if (list[0][table.id]) {
+        columnsSet.add(table.id);
+    }
+    table.columns.forEach(col => columnsSet.add(col));
+    let createdData = undefined;
+    if (table.createdDate) {
+        createdData = table.createdDate.type === 'date' ? new Date() : new Date().getTime();
+        columnsSet.add(table.createdDate.column);
+    }
+    let updatedDate = undefined;
+    if (table.updatedDate) {
+        updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime();
+        columnsSet.add(table.updatedDate.column);
+    }
+    const columns = Array.from(columnsSet);
+    // 构建 sql
+    let sql = `insert into ??(${columns.map(() => '??').join(',')}) values`;
+    const values = [table.tableName, ...columns];
+    list.forEach((data, idx) => {
+        if (idx > 0) {
+            sql += ',';
+        }
+        sql += `(${columns.map(() => '?').join(',')})`;
+        if (table.createdDate) {
+            data[table.createdDate.column] = createdData;
+        }
+        if (table.updatedDate) {
+            data[table.updatedDate.column] = updatedDate;
+        }
+        values.push(...columns.map(col => data[col]));
+    });
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const rsh = res;
+    if (rsh.affectedRows !== list.length) {
+        throw new exception_1.MysqlException(`Insert many for table ${table.tableName} failed,expected to insert ${list.length} records, actually ${rsh.affectedRows}.`);
+    }
+    // insert 插入多条记录是不会返回生成的 id 的，所以就没有办法自动处理
+}
+exports.insertMany = insertMany;

package/dist/mysql/manager/ops/modify.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.modify = void 0;
+const utils_1 = require("../utils");
+async function modify(config, connection, sql, values) {
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values || []);
+    const pck = res;
+    return pck.affectedRows;
+}
+exports.modify = modify;

package/dist/mysql/manager/ops/paginate.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.paginate = void 0;
+const count_1 = require("./count");
+const find_1 = require("./find");
+async function paginate(config, conn, opts) {
+    const pn = opts.pn && opts.pn >= 1 ? opts.pn : 1;
+    const limit = opts.pz && opts.pz >= 1 && opts.pz <= 1000 ? opts.pz : 20;
+    const offset = (pn - 1) * limit;
+    const list = await (0, find_1.find)(config, conn, {
+        table: opts.table,
+        criteria: opts.criteria,
+        offset,
+        limit,
+        orderBy: opts.orderBy
+    });
+    const total = await (0, count_1.count)(config, conn, opts.table, opts.criteria);
+    return {
+        total,
+        list
+    };
+}
+exports.paginate = paginate;

package/dist/mysql/manager/ops/query.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.query = void 0;
+const utils_1 = require("../utils");
+async function query(config, connection, sql, values) {
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values || []);
+    return res;
+}
+exports.query = query;

package/dist/mysql/manager/ops/update.js

@@ -0,0 +1,201 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateMany = exports.updateOne = exports.partialUpdate = exports.update = void 0;
+const exception_1 = require("../../exception");
+const criteria_1 = require("./criteria");
+const utils_1 = require("../utils");
+/**
+ * 更新
+ * @param config
+ * @param connection
+ * @param mapping
+ * @param data
+ * @returns
+ */
+async function update(config, connection, table, data) {
+    // 列信息
+    let columns = [...table.columns];
+    // 创建时间不更新
+    if (table.createdDate) {
+        const { column } = table.createdDate;
+        columns = columns.filter(col => col !== column);
+    }
+    if (table.updatedDate) {
+        const updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime();
+        data[table.updatedDate.column] = updatedDate;
+        columns.push(table.updatedDate.column);
+    }
+    // 构建 sql
+    const sql = `update ?? set ${columns.map(() => ' ?? = ? ').join(',')} where ?? = ?`;
+    // 值
+    const values = [
+        table.tableName,
+        ...columns.flatMap(col => [col, data[col] || null]),
+        table.id,
+        data[table.id]
+    ];
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    if (packet.affectedRows !== 1) {
+        throw new exception_1.MysqlException(`Failed to update record, possibly due to non-existent record，table：${table.tableName}，primary key: ${data[table.id]}`);
+    }
+    return data;
+}
+exports.update = update;
+/**
+ * 转换更新器
+ * @param table
+ * @param updater
+ * @returns
+ */
+function updatorToSql(table, updater) {
+    const values = [];
+    // 更新操作
+    const updateFragList = [];
+    // 更新时间
+    if (table.updatedDate) {
+        const updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime();
+        updateFragList.push(' ?? = ?');
+        values.push(table.updatedDate.column, updatedDate);
+    }
+    for (const column in updater) {
+        // 过滤掉id
+        if (column === table.id) {
+            continue;
+        }
+        const val = updater[column];
+        if (Array.isArray(val)) {
+            // set null
+            if (val[0] === 'setNull') {
+                updateFragList.push(' ?? = NULL ');
+                values.push(column);
+                continue;
+            }
+            if (val[0] === 'inc') {
+                updateFragList.push(' ?? = ?? + ? ');
+                values.push(column, column, val[1]);
+                continue;
+            }
+            continue;
+        }
+        updateFragList.push(' ?? = ? ');
+        values.push(column, val);
+    }
+    return { sql: updateFragList.join(','), values };
+}
+/**
+ * 部分更新
+ * @param connection
+ * @param mapping
+ * @param type
+ * @param data
+ */
+async function partialUpdate(config, connection, table, data) {
+    if (!data[table.id]) {
+        throw new exception_1.MysqlException(`Can't do a partial update, the data to be updated does not contain a primary key，table: ${table.tableName}，column：${JSON.stringify(data)}`);
+    }
+    const id = data[table.id];
+    if (typeof id !== 'string' && typeof id !== 'number') {
+        throw new exception_1.MysqlException('Primary key can only be of string or number type');
+    }
+    if (Object.keys(data).length < 2) {
+        throw new exception_1.MysqlException(`Can't do a partial update, data must contain at least one column outside of the primary key，table: ${table.tableName}，column：${JSON.stringify(data)}`);
+    }
+    const fieldNames = Object.keys(data);
+    for (const name of fieldNames) {
+        if (name !== table.id && !table.columns.some(col => col === name)) {
+            throw new exception_1.MysqlException(`Can't do a partial update，there are unconfigured columns in the data，table： ${table.tableName}，unconfigured column：${name}`);
+        }
+    }
+    let sql = ` update ?? `;
+    const values = [table.tableName];
+    // 更新操作
+    const convertRes = updatorToSql(table, data);
+    if (!convertRes.sql) {
+        throw new exception_1.MysqlException('No fields were specified to be updated!');
+    }
+    values.push(...convertRes.values);
+    sql += ` set ${convertRes.sql} where ?? = ?`;
+    values.push(table.id, id);
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    return packet.affectedRows === 1;
+}
+exports.partialUpdate = partialUpdate;
+/**
+ * 更新指定的一条记录
+ * @param config
+ * @param connection
+ * @param table
+ * @param query
+ * @param data
+ */
+async function updateOne(config, connection, table, query, updater) {
+    const values = [];
+    const mysqlQuery = (0, criteria_1.buildQuery)(query);
+    if (!mysqlQuery) {
+        throw new exception_1.MysqlException('No valid criteria specified.');
+    }
+    let sql = ` update ?? `;
+    values.push(table.tableName);
+    // 更新操作
+    const convertRes = updatorToSql(table, updater);
+    if (!convertRes.sql) {
+        throw new exception_1.MysqlException('No fields were specified to be updated!');
+    }
+    sql += ` set ${convertRes.sql} `;
+    values.push(...convertRes.values);
+    sql += ` where ${mysqlQuery.sql} limit 1`;
+    values.push(...mysqlQuery.values);
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    return packet.affectedRows === 1;
+}
+exports.updateOne = updateOne;
+/**
+ * 更新所有匹配条件的记录
+ * @param config
+ * @param connection 连接
+ * @param table 表
+ * @param query 查询条件
+ * @param updater 更新操作，支持置空和递增（需要使用元组）
+ */
+async function updateMany(config, connection, opts) {
+    const values = [];
+    const mysqlQuery = (0, criteria_1.buildQuery)(opts.query);
+    if (!mysqlQuery) {
+        throw new exception_1.MysqlException('No valid criteria specified.');
+    }
+    let sql = ` update ?? `;
+    values.push(opts.table.tableName);
+    // 更新操作
+    const convertRes = updatorToSql(opts.table, opts.updater);
+    if (!convertRes.sql) {
+        throw new exception_1.MysqlException('No fields were specified to be updated!');
+    }
+    sql += ` set ${convertRes.sql} `;
+    values.push(...convertRes.values);
+    sql += ` where ${mysqlQuery.sql} `;
+    values.push(...mysqlQuery.values);
+    // 排序
+    if (opts.orderBy && opts.orderBy.length) {
+        opts.orderBy.forEach((orderBy, idx) => {
+            const [field, sort] = orderBy;
+            if (idx == 0) {
+                sql += ` order by ?? ${sort} `;
+            }
+            else {
+                sql += ` , ?? ${sort} `;
+            }
+            values.push(field);
+        });
+    }
+    // 数量限制
+    if (opts.limit) {
+        sql += ` limit ${opts.limit} `;
+    }
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    return packet.affectedRows;
+}
+exports.updateMany = updateMany;

package/dist/mysql/manager/tx-strict.js

@@ -0,0 +1,98 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MysqlStrictTxSession = void 0;
+const tx_1 = require("./tx");
+const exception_1 = require("../exception");
+/**
+ * 严格 mysql 事务会话，会禁用一些操作.
+ */
+class MysqlStrictTxSession extends tx_1.MysqlTxSession {
+    #opsCount = 0;
+    constructor(config, conn) {
+        super(config, conn);
+    }
+    /**
+     * 为操作计数，检查是否操作次数过多.
+     */
+    #checkAndAddOpsCount() {
+        if (this.#opsCount >= 10) {
+            throw new exception_1.MysqlException('Too many operations in a strict transaction.');
+        }
+        this.#opsCount++;
+    }
+    findById(table, id) {
+        this.#checkAndAddOpsCount();
+        return super.findById(table, id);
+    }
+    findByIdIn(table, ids) {
+        if (ids.length > 100) {
+            throw new exception_1.MysqlException(`The augument ids length(${ids.length}) passed to findByIdIn in a strict transaction is too large .`);
+        }
+        this.#checkAndAddOpsCount();
+        return super.findByIdIn(table, ids);
+    }
+    existsBy(table, criteria) {
+        this.#checkAndAddOpsCount();
+        return super.existsBy(table, criteria);
+    }
+    existsById(table, id) {
+        this.#checkAndAddOpsCount();
+        return super.existsById(table, id);
+    }
+    deleteById(table, id) {
+        this.#checkAndAddOpsCount();
+        return super.deleteById(table, id);
+    }
+    deleteOne(table, criteria) {
+        this.#checkAndAddOpsCount();
+        return super.deleteOne(table, criteria);
+    }
+    deleteMany(opts) {
+        throw new exception_1.MysqlException('Prohibited to use deleteBy in a strict transaction.');
+    }
+    findAll(table) {
+        throw new exception_1.MysqlException('Prohibited to use findAll in a strict transaction.');
+    }
+    findFirst(table, criteria) {
+        this.#checkAndAddOpsCount();
+        return super.findFirst(table, criteria);
+    }
+    insert(table, data) {
+        this.#checkAndAddOpsCount();
+        return super.insert(table, data);
+    }
+    insertMany(table, list) {
+        throw new exception_1.MysqlException('Prohibited to use insertMany in a strict transaction.');
+    }
+    update(table, data) {
+        this.#checkAndAddOpsCount();
+        return super.update(table, data);
+    }
+    updateMany(opts) {
+        throw new exception_1.MysqlException('Prohibited to use updateMany in a strict transaction.');
+    }
+    updateOne(table, query, updater) {
+        this.#checkAndAddOpsCount();
+        return super.updateOne(table, query, updater);
+    }
+    partialUpdate(table, data) {
+        this.#checkAndAddOpsCount();
+        return super.partialUpdate(table, data);
+    }
+    find(opts) {
+        throw new exception_1.MysqlException('Prohibited to use find in a strict transaction.');
+    }
+    count(table, criteria) {
+        throw new exception_1.MysqlException('Prohibited to use count in a strict transaction.');
+    }
+    paginate(opts) {
+        throw new exception_1.MysqlException('Prohibited to use paginate in a strict transaction.');
+    }
+    query(sql, values) {
+        throw new exception_1.MysqlException('Prohibited to use query in a strict transaction.');
+    }
+    modify(sql, values) {
+        throw new exception_1.MysqlException('Prohibited to use modify in a strict transaction.');
+    }
+}
+exports.MysqlStrictTxSession = MysqlStrictTxSession;

package/dist/mysql/manager/tx.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MysqlTxSession = void 0;
+const base_1 = require("./base");
+const exception_1 = require("../exception");
+/**
+ * mysql 事务会话.
+ */
+class MysqlTxSession extends base_1.BaseMysqlManager {
+    /**
+     * 中止标识
+     */
+    #aborted = false;
+    constructor(config, conn) {
+        super({ config, connection: conn });
+    }
+    queryWithConnection(queryFn) {
+        if (this.#aborted) {
+            throw new exception_1.MysqlException('Session has been aborted!');
+        }
+        return super.queryWithConnection(queryFn);
+    }
+    /**
+     * 中止，被中止后的会话不能再进行任何操作
+     */
+    abort() {
+        this.#aborted = true;
+    }
+}
+exports.MysqlTxSession = MysqlTxSession;