wok-server 0.5.0 → 0.7.0

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

@@ -1,216 +1,222 @@
-"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");
-const utils_2 = 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, (0, utils_2.processColumnValue)(data[col])]),
-        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];
-        // undefined 表示不参与更新，作用是方便编写一些特殊的逻辑，比如特定情况下不更新
-        if (val === undefined) {
-            continue;
-        }
-        // 兼容将值设置成 null 的情况，和 ['setNull’] 等同
-        if (val === null) {
-            updateFragList.push(' ?? = NULL ');
-            values.push(column);
-            continue;
-        }
-        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;
-            }
-            if (val[0] === 'set') {
-                updateFragList.push(' ?? = ? ');
-                values.push(column, (0, utils_2.processColumnValue)(val[1]));
-                continue;
-            }
-        }
-        updateFragList.push(' ?? = ? ');
-        values.push(column, (0, utils_2.processColumnValue)(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;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateMany = exports.updateOne = exports.partialUpdate = exports.updatorToSql = exports.update = void 0;
+const exception_1 = require("../../exception");
+const criteria_1 = require("./criteria");
+const utils_1 = require("../utils");
+const utils_2 = require("./utils");
+const order_by_1 = require("./order-by");
+/**
+ * 更新
+ * @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, (0, utils_2.processColumnValue)(data[col])]),
+        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
+ * @param autoUpdateTime 是否自动添加更新时间
+ * @returns
+ */
+function updatorToSql(table, updater) {
+    const values = [];
+    const updateFragList = [];
+    for (const column in updater) {
+        // 过滤掉 id
+        if (column === table.id) {
+            continue;
+        }
+        // 过滤掉 createdDate / updatedDate（自动处理）
+        if ((table.createdDate && column === table.createdDate.column)
+            || (table.updatedDate && column === table.updatedDate.column)) {
+            continue;
+        }
+        const val = updater[column];
+        // undefined 表示不参与更新
+        if (val === undefined) {
+            continue;
+        }
+        // 0.7.0 版本开始，null 和 undefined 一样被忽略更新
+        // 如需设置字段为 NULL，请使用 ['setNull']
+        if (val === null) {
+            continue;
+        }
+        if (Array.isArray(val)) {
+            if (val[0] === 'setNull') {
+                updateFragList.push(' ?? = NULL ');
+                values.push(column);
+                continue;
+            }
+            if (val[0] === 'inc') {
+                const incBy = val.length === 1 ? 1 : val[1];
+                updateFragList.push(' ?? = ?? + ? ');
+                values.push(column, column, incBy);
+                continue;
+            }
+            if (val[0] === 'now') {
+                updateFragList.push(' ?? = NOW() ');
+                values.push(column);
+                continue;
+            }
+            if (val[0] === 'set') {
+                updateFragList.push(' ?? = ? ');
+                values.push(column, (0, utils_2.processColumnValue)(val[1]));
+                continue;
+            }
+            if (val[0] === 'concat') {
+                updateFragList.push(' ?? = CONCAT(IFNULL(??, \'\'), ?) ');
+                values.push(column, column, val[1]);
+                continue;
+            }
+            if (val[0] === 'expr') {
+                updateFragList.push(' ?? = ' + val[1] + ' ');
+                values.push(column, ...(val[2] || []));
+                continue;
+            }
+        }
+        updateFragList.push(' ?? = ? ');
+        values.push(column, (0, utils_2.processColumnValue)(val));
+    }
+    // 如果没有有效更新字段，抛出异常
+    if (updateFragList.length === 0) {
+        throw new exception_1.MysqlException(`No effective fields to update (null values are ignored since v0.7.0), table: ${table.tableName}, updater: ${JSON.stringify(updater)}`);
+    }
+    // 自动添加更新时间
+    if (table.updatedDate) {
+        const updatedDate = table.updatedDate.type === 'date' ? new Date() : new Date().getTime();
+        updateFragList.push(' ?? = ?');
+        values.push(table.updatedDate.column, updatedDate);
+    }
+    return { sql: updateFragList.join(','), values };
+}
+exports.updatorToSql = updatorToSql;
+/**
+ * 部分更新
+ * @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');
+    }
+    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);
+    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);
+    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);
+    sql += ` set ${convertRes.sql} `;
+    values.push(...convertRes.values);
+    sql += ` where ${mysqlQuery.sql} `;
+    values.push(...mysqlQuery.values);
+    // 排序
+    if (opts.orderBy && opts.orderBy.length) {
+        const ob = (0, order_by_1.buildOrderBy)(opts.orderBy);
+        sql += ob.sql;
+        values.push(...ob.values);
+    }
+    // 数量限制
+    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/ops/upsert.js

@@ -0,0 +1,178 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.upsertWithUpdater = exports.upsertMany = exports.upsert = void 0;
+const utils_1 = require("../utils");
+const insert_1 = require("./insert");
+const update_1 = require("./update");
+/**
+ * Upsert 单条数据
+ * 如果主键冲突则更新，否则插入
+ * @param config
+ * @param connection
+ * @param table
+ * @param data
+ * @returns
+ */
+async function upsert(config, connection, table, data) {
+    let columnsSet = new Set();
+    if (data[table.id]) {
+        columnsSet.add(table.id);
+    }
+    table.columns.forEach(col => columnsSet.add(col));
+    const now = new Date();
+    const nowTimestamp = now.getTime();
+    if (table.createdDate) {
+        const createdData = table.createdDate.type === 'date' ? now : nowTimestamp;
+        data[table.createdDate.column] = createdData;
+        columnsSet.add(table.createdDate.column);
+    }
+    if (table.updatedDate) {
+        const updatedDate = table.updatedDate.type === 'date' ? now : nowTimestamp;
+        data[table.updatedDate.column] = updatedDate;
+        columnsSet.add(table.updatedDate.column);
+    }
+    const columns = Array.from(columnsSet);
+    // 构建 insert values
+    const fragList = [];
+    const insertValues = [];
+    for (const col of columns) {
+        const { frag, values: vs } = (0, insert_1.processInsertValue)(data[col]);
+        fragList.push(frag);
+        insertValues.push(...vs);
+    }
+    const insertSql = `insert into ??(${columns.map(() => '??').join(',')}) values(${fragList.join(',')})`;
+    // 构建 on duplicate key update（排除 id）
+    const updateColumns = columns.filter(col => col !== table.id);
+    const updateFragments = [];
+    const updateValues = [];
+    for (const col of updateColumns) {
+        const { frag, values: vs } = (0, insert_1.processInsertValue)(data[col]);
+        updateFragments.push(`?? = ${frag}`);
+        updateValues.push(col, ...vs);
+    }
+    const updateSql = ` on duplicate key update ${updateFragments.join(',')}`;
+    const sql = insertSql + updateSql;
+    const values = [table.tableName, ...columns, ...insertValues, ...updateValues];
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    if (packet.insertId && (data[table.id] === undefined || data[table.id] === null)) {
+        data[table.id] = packet.insertId;
+    }
+    return data;
+}
+exports.upsert = upsert;
+/**
+ * Upsert 多条数据
+ * 如果主键冲突则更新，否则插入
+ * @param config
+ * @param connection
+ * @param table
+ * @param list
+ * @returns 影响的行数
+ */
+async function upsertMany(config, connection, table, list) {
+    if (!list.length) {
+        return 0;
+    }
+    let columnsSet = new Set();
+    if (list[0][table.id]) {
+        columnsSet.add(table.id);
+    }
+    table.columns.forEach(col => columnsSet.add(col));
+    const now = new Date();
+    const nowTimestamp = now.getTime();
+    let createdData = undefined;
+    if (table.createdDate) {
+        createdData = table.createdDate.type === 'date' ? now : nowTimestamp;
+        columnsSet.add(table.createdDate.column);
+    }
+    let updatedDate = undefined;
+    if (table.updatedDate) {
+        updatedDate = table.updatedDate.type === 'date' ? now : nowTimestamp;
+        columnsSet.add(table.updatedDate.column);
+    }
+    const columns = Array.from(columnsSet);
+    let sql = `insert into ??(${columns.map(() => '??').join(',')}) values`;
+    const values = [table.tableName, ...columns];
+    list.forEach((data, idx) => {
+        if (idx > 0) {
+            sql += ',';
+        }
+        const fragList = [];
+        const rowValues = [];
+        if (table.createdDate) {
+            data[table.createdDate.column] = createdData;
+        }
+        if (table.updatedDate) {
+            data[table.updatedDate.column] = updatedDate;
+        }
+        for (const col of columns) {
+            const { frag, values: vs } = (0, insert_1.processInsertValue)(data[col]);
+            fragList.push(frag);
+            rowValues.push(...vs);
+        }
+        sql += `(${fragList.join(',')})`;
+        values.push(...rowValues);
+    });
+    const updateColumns = columns.filter(col => col !== table.id);
+    sql += ` on duplicate key update ${updateColumns.map(() => '?? = values(??)').join(',')}`;
+    updateColumns.forEach(col => values.push(col, col));
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    return packet.affectedRows;
+}
+exports.upsertMany = upsertMany;
+/**
+ * Upsert 单条数据（支持自定义更新器）
+ * 如果主键冲突则按自定义逻辑更新，否则插入
+ * @param config
+ * @param connection
+ * @param table
+ * @param data 插入的数据
+ * @param updater 冲突时的更新器
+ * @returns
+ */
+async function upsertWithUpdater(config, connection, table, data, updater) {
+    let columnsSet = new Set();
+    if (data[table.id]) {
+        columnsSet.add(table.id);
+    }
+    table.columns.forEach(col => {
+        if (data[col] !== undefined) {
+            columnsSet.add(col);
+        }
+    });
+    const now = new Date();
+    const nowTimestamp = now.getTime();
+    if (table.createdDate) {
+        const createdData = table.createdDate.type === 'date' ? now : nowTimestamp;
+        data[table.createdDate.column] = createdData;
+        columnsSet.add(table.createdDate.column);
+    }
+    if (table.updatedDate) {
+        const updatedDate = table.updatedDate.type === 'date' ? now : nowTimestamp;
+        data[table.updatedDate.column] = updatedDate;
+        columnsSet.add(table.updatedDate.column);
+    }
+    const columns = Array.from(columnsSet);
+    // 构建 insert values
+    const fragList = [];
+    const insertValues = [];
+    for (const col of columns) {
+        const { frag, values: vs } = (0, insert_1.processInsertValue)(data[col]);
+        fragList.push(frag);
+        insertValues.push(...vs);
+    }
+    const insertSql = `insert into ??(${columns.map(() => '??').join(',')}) values(${fragList.join(',')})`;
+    const convertRes = (0, update_1.updatorToSql)(table, updater);
+    const updateSql = ` on duplicate key update ${convertRes.sql}`;
+    const sql = insertSql + updateSql;
+    const values = [table.tableName, ...columns, ...insertValues, ...convertRes.values];
+    const res = await (0, utils_1.promiseQuery)(config, connection, sql, values);
+    const packet = res;
+    if (packet.insertId && (data[table.id] === undefined || data[table.id] === null)) {
+        data[table.id] = packet.insertId;
+    }
+    return data;
+}
+exports.upsertWithUpdater = upsertWithUpdater;