midway-fatcms 0.0.1-beta.7 → 0.0.1-beta.71

package/src/libs/crud-pro/services/CrudProExecuteSqlService.ts

@@ -1,279 +1,264 @@
-import * as _ from 'lodash';
-import {PoolClient} from 'pg'
-import {CrudProServiceBase} from './CrudProServiceBase';
-import {SqlCfgModel} from '../models/SqlCfgModel';
-import {KeyOfCrudTypes, KeysOfCustomSQL, KeysOfSqlResPicker, SqlDbType} from '../models/keys';
-import {CommonException, Exceptions} from '../exceptions';
-import {pickAndConvertPgRows} from '../utils/sqlConvert/convertPgType';
-import {replaceQuestionMarks} from "../utils/sqlConvert/convertPgSql";
-import {replaceQuestionMarksForMssql} from "../utils/sqlConvert/convertMsSql";
-import {ModelUtils} from '../utils/ModelUtils';
-import {MixinUtils} from '../utils/MixinUtils';
-import {FuncContext} from '../models/FuncContext';
-import {ExecuteContext} from "../models/ExecuteContext";
-
-const { checkFuncCfgValid } = ModelUtils;
-const { isEmpty } = MixinUtils;
-
-class CrudProExecuteSqlService extends CrudProServiceBase {
-  async executeSqlCfgModels(): Promise<void> {
-    const exeCtx = this.getExecuteContext();
-    const sqlCfgModels = exeCtx.getSqlCfgModels();
-    for (let i = 0; i < sqlCfgModels.length; i++) {
-      const sqlCfgModel = sqlCfgModels[i];
-      const checkResult = this.executeSqlCfgModelPreCheck(sqlCfgModel);
-      if (checkResult) {
-        await this.executeSqlCfgModel(sqlCfgModel);
-      }
-    }
-  }
-
-  private async executeSqlCfgModel(sqlCfgModel: SqlCfgModel) {
-    this.logger.info('executeSqlCfgModel1', sqlCfgModel.executeSql, sqlCfgModel.executeSqlArgs);
-
-    const exeCtx = this.getExecuteContext();
-    const connection = await this.getTxConnectionBySqlCfg(sqlCfgModel);
-    const executeSqlArgs = this.handleExecuteSqlArgsByResModel(exeCtx, sqlCfgModel.executeSqlArgs || [])
-
-
-    let sqlRes: any;
-    let queryRes : any;
-
-    if (sqlCfgModel.sqlDdType === SqlDbType.postgres) { // import { PoolClient } from 'pg';
-
-      const pgClient: PoolClient = connection as any;
-      const pgSql = replaceQuestionMarks(sqlCfgModel.executeSql);
-
-      this.logger.info('executeSqlCfgModel_postgres', pgSql, executeSqlArgs, sqlCfgModel.resPicker);
-
-
-      queryRes = await pgClient.query({
-        text: pgSql,
-        values: executeSqlArgs || []
-      });
-      sqlRes = pickAndConvertPgRows(queryRes)
-
-    }
-
-    else if (sqlCfgModel.sqlDdType === SqlDbType.sqlserver) {
-
-      const mssql = replaceQuestionMarksForMssql(sqlCfgModel.executeSql);
-
-      this.logger.info('executeSqlCfgModel_sqlserver', mssql, executeSqlArgs, sqlCfgModel.resPicker);
-      // SQLServer
-      queryRes = await connection.query(mssql, executeSqlArgs);
-      sqlRes = _.get(queryRes,'recordsets[0]') || []
-    }
-
-    else {
-
-      this.logger.info('executeSqlCfgModel_mysql', sqlCfgModel.executeSql, executeSqlArgs, sqlCfgModel.resPicker);
-
-
-      // MYSQL
-      queryRes = await connection.query(sqlCfgModel.executeSql, executeSqlArgs);
-      sqlRes = queryRes[0]
-    }
-
-    const resObject = this.toQueryResByResPicker(sqlRes, queryRes, sqlCfgModel);
-    exeCtx.setResModelItem(sqlCfgModel.resName, resObject);
-  }
-
-
-  private handleExecuteSqlArgsByResModel(exeCtx: ExecuteContext, executeSqlArgs: any[]): any[] {
-    return executeSqlArgs.map((arg)=> {
-      if (arg && typeof arg === 'object' && typeof arg.___GENERATE_GET_RES_ATTR___ === 'string' ) {
-        const word: string = arg.___GENERATE_GET_RES_ATTR___;
-        if (word.startsWith(KeysOfCustomSQL.SQL_PICK_RES_AS_NUMBER)) {
-          const resName = MixinUtils.removeStringPrefix(word, KeysOfCustomSQL.SQL_PICK_RES_AS_NUMBER)
-          this.logger.info('handleExecuteSqlArgsByResModel resName', resName);
-          this.logger.info('handleExecuteSqlArgsByResModel getResModel', exeCtx.getResModel());
-          return Number(exeCtx.getResModelItemLodash(resName) || 0);
-        }
-        if (word.startsWith(KeysOfCustomSQL.SQL_PICK_RES_AS_STRING)) {
-          const resName = MixinUtils.removeStringPrefix(word, KeysOfCustomSQL.SQL_PICK_RES_AS_STRING)
-          return String(exeCtx.getResModelItemLodash(resName) || "");
-        }
-      }
-      return arg;
-    })
-  }
-
-
-  private executeSqlCfgModelPreCheck(sqlCfgModel: SqlCfgModel): boolean {
-    const crudType = sqlCfgModel.getCrudType();
-
-    if (KeyOfCrudTypes.EMPTY_SQL === crudType) {
-      // 允许空SQL，没有需要执行的正常。
-      return false;
-    }
-
-    if (KeyOfCrudTypes.NOT_CRUD === crudType) {
-      // 只支持增删改查，其他语句不支持。
-      throw new CommonException(Exceptions.CFG_NOT_SUPPORT_THE_SQL, sqlCfgModel.executeSql);
-    }
-
-    // 此SQL不需要执行
-    if (!this.isNeedExecute(sqlCfgModel)) {
-      return false;
-    }
-
-    // 执行之前，根据配置校验一下此SQL是否可以执行
-    this.assertByValidate(sqlCfgModel);
-
-    return true;
-  }
-
-  private assertByValidate(sqlCfgModel: SqlCfgModel) {
-    const exeCtx = this.getExecuteContext();
-    const funcCfg = sqlCfgModel.validate;
-    if (!funcCfg) {
-      return; // 没有配置。
-    }
-
-    if (!checkFuncCfgValid(funcCfg)) {
-      throw new CommonException(Exceptions.CFG_SQL_CFG_EXECUTE_VALIDATE_EMPTY, {
-        funcCfg: funcCfg,
-        originSql: sqlCfgModel.originSql,
-      });
-    }
-
-    const errMsg = funcCfg.message || '执行前校验失败';
-
-    let result;
-    try {
-      const exeFunCtx = new FuncContext();
-
-      exeFunCtx.currentValue = null;
-      exeFunCtx.reqModel = exeCtx.getReqModel();
-      exeFunCtx.resModel = exeCtx.getResModel();
-
-      result = this.serviceHub.executeFuncCfg(funcCfg, exeFunCtx); // 可以抛出异常
-    } catch (e) {
-      this.logger.debug('assertByValidate', e);
-      throw new CommonException(Exceptions.RUN_EXECUTE_VALIDATE, {
-        funcCfg: funcCfg,
-        originSql: sqlCfgModel.originSql,
-        error: '' + e,
-      });
-    }
-
-    if (!(typeof result === 'boolean')) {
-      // 自定义的校验函数，必须返回boolean类型的树枝
-      throw new CommonException(Exceptions.CFG_IS_NEED_EXECUTE_FUNC_MUST_RETURN_BOOL, {
-        funcCfg: funcCfg,
-        originSql: sqlCfgModel.originSql,
-      });
-    }
-
-    if (result === false) {
-      throw new CommonException(Exceptions.RUN_EXECUTE_VALIDATE, errMsg);
-    }
-  }
-
-  /**
-   * 判断此SQL是否需要执行
-   * @param sqlCfgModel
-   * @param exeCtx
-   * @private
-   */
-  private isNeedExecute(sqlCfgModel: SqlCfgModel) {
-    const exeCtx = this.getExecuteContext();
-    const funcCfg = sqlCfgModel.executeWhen;
-    if (!funcCfg || Object.keys(funcCfg).length === 0) {
-      return true; // 没有配置。默认，需要执行。
-    }
-
-    if (!checkFuncCfgValid(funcCfg)) {
-      throw new CommonException(Exceptions.CFG_CHECK_EXECUTE_WHEN_EMPTY, {
-        funcCfg: funcCfg,
-        originSql: sqlCfgModel.originSql,
-      });
-    }
-
-    let result;
-
-    try {
-      const exeFunCtx = new FuncContext();
-
-      exeFunCtx.currentValue = null;
-      exeFunCtx.reqModel = exeCtx.getReqModel();
-      exeFunCtx.resModel = exeCtx.getResModel();
-
-      result = this.serviceHub.executeFuncCfg(funcCfg, exeFunCtx);
-    } catch (e) {
-      this.logger.debug('isNeedExecute', e);
-      throw new CommonException(Exceptions.RUN_IS_NEED_EXECUTE_ERR, {
-        funcCfg: funcCfg,
-        originSql: sqlCfgModel.originSql,
-        error: '' + e,
-      });
-    }
-
-    if (!(typeof result === 'boolean')) {
-      // 自定义的校验函数，必须返回boolean类型的树枝
-      throw new CommonException(Exceptions.CFG_IS_NEED_EXECUTE_FUNC_MUST_RETURN_BOOL, {
-        funcCfg: funcCfg,
-        originSql: sqlCfgModel.originSql,
-      });
-    }
-
-    return result === true;
-  }
-
-  private toQueryResByResPicker(sqlRes: any, queryRes: any, sqlCfgModel: SqlCfgModel) {
-    const resPicker = sqlCfgModel.resPicker;
-
-    // 返回第一行
-    if (KeysOfSqlResPicker.RESULT_FIRST_ROW === resPicker) {
-      const rows = sqlRes as any[];
-      if (rows && rows.length > 0) {
-        return rows[0];
-      }
-      return null;
-    }
-
-    // $ResultSet[0].total_count ， 只取第一行的的total_count
-    if (KeysOfSqlResPicker.RESULT_TOTAL_COUNT === resPicker) {
-      const rows = sqlRes as any[];
-      if (isEmpty(rows)) {
-        return 0;
-      }
-      const map0 = rows[0];
-      return Number(map0['total_count'] || 0);
-    }
-
-    // 增删改res的内容是修改结果：包括： insert\delete\update
-    if (KeysOfSqlResPicker.UPDATE_RESULT === resPicker) {
-      if (sqlCfgModel.sqlDdType === SqlDbType.postgres) {
-        return {
-          insertId: _.get(queryRes,'rows[0].id'),
-          affectedRows: _.get(queryRes,'rowCount')
-        }
-      }
-
-      if (sqlCfgModel.sqlDdType === SqlDbType.sqlserver) {
-        return {
-          insertId: _.get(queryRes,'recordset[0].id'),
-          affectedRows: _.get(queryRes,'rowsAffected[0]')
-        }
-      }
-
-      return sqlRes;
-    }
-
-    //其他配置：形如： sqlRes[0].total_count
-    if (typeof resPicker === 'string') {
-      if (resPicker.startsWith('sqlRes')){
-        return _.get({ sqlRes }, resPicker);
-      } else {
-        throw  new CommonException(Exceptions.RUN_EXECUTE_VALIDATE, 'resPicker必须是以sqlRes开头')
-      }
-    }
-
-    return sqlRes;
-  }
-
-}
-
-export { CrudProExecuteSqlService };
+import * as _ from 'lodash';
+import { PoolClient } from 'pg';
+import { CrudProServiceBase } from './CrudProServiceBase';
+import { SqlCfgModel } from '../models/SqlCfgModel';
+import { KeyOfCrudTypes, KeysOfCustomSQL, KeysOfSqlResPicker, SqlDbType } from '../models/keys';
+import { CommonException, Exceptions } from '../exceptions';
+import { pickAndConvertRowsByMix } from '../utils/sqlConvert/convertMix';
+import { replaceQuestionMarks } from '../utils/sqlConvert/convertPgSql';
+import { replaceQuestionMarksForMssql } from '../utils/sqlConvert/convertMsSql';
+import { ModelUtils } from '../utils/ModelUtils';
+import { MixinUtils } from '../utils/MixinUtils';
+import { FuncContext } from '../models/FuncContext';
+import { ExecuteContext } from '../models/ExecuteContext';
+
+const { checkFuncCfgValid } = ModelUtils;
+const { isEmpty } = MixinUtils;
+
+class CrudProExecuteSqlService extends CrudProServiceBase {
+  async executeSqlCfgModels(): Promise<void> {
+    const exeCtx = this.getExecuteContext();
+    const sqlCfgModels = exeCtx.getSqlCfgModels();
+    for (let i = 0; i < sqlCfgModels.length; i++) {
+      const sqlCfgModel = sqlCfgModels[i];
+      const checkResult = this.executeSqlCfgModelPreCheck(sqlCfgModel);
+      if (checkResult) {
+        await this.executeSqlCfgModel(sqlCfgModel);
+      }
+    }
+  }
+
+  private async executeSqlCfgModel(sqlCfgModel: SqlCfgModel) {
+    this.logger.info('[CrudProExecuteSqlService] executeSqlCfgModel_before', sqlCfgModel.executeSql, sqlCfgModel.executeSqlArgs);
+
+    const exeCtx = this.getExecuteContext();
+    const connection = await this.getTxConnectionBySqlCfg(sqlCfgModel);
+    const executeSqlArgs = this.handleExecuteSqlArgsByResModel(exeCtx, sqlCfgModel.executeSqlArgs || []);
+
+    let queryRes: any;
+
+    if (sqlCfgModel.sqlDbType === SqlDbType.postgres) {
+      // import { PoolClient } from 'pg';
+
+      const pgClient: PoolClient = connection as any;
+      const pgSql = replaceQuestionMarks(sqlCfgModel.executeSql, sqlCfgModel.isNativeSQL);
+      queryRes = await pgClient.query({
+        text: pgSql,
+        values: executeSqlArgs || [],
+      });
+
+      this.logger.debug('[CrudProExecuteSqlService] executeSqlCfgModel_postgres', pgSql, executeSqlArgs, sqlCfgModel.resPicker);
+
+    } else if (sqlCfgModel.sqlDbType === SqlDbType.sqlserver) {
+      // SQLServer
+
+      const mssql = replaceQuestionMarksForMssql(sqlCfgModel.executeSql, sqlCfgModel.isNativeSQL);
+      queryRes = await connection.query(mssql, executeSqlArgs);
+      this.logger.debug('[CrudProExecuteSqlService] executeSqlCfgModel_sqlserver', mssql, executeSqlArgs, sqlCfgModel.resPicker);
+
+    } else {
+      // MYSQL
+      queryRes = await connection.query(sqlCfgModel.executeSql, executeSqlArgs);
+      this.logger.debug('[CrudProExecuteSqlService] executeSqlCfgModel_mysql', sqlCfgModel.executeSql, executeSqlArgs, sqlCfgModel.resPicker);
+    }
+
+    const sqlRes = pickAndConvertRowsByMix(queryRes, sqlCfgModel.sqlDbType);
+    const resObject = this.toQueryResByResPicker(sqlRes, queryRes, sqlCfgModel);
+    exeCtx.setResModelItem(sqlCfgModel.resName, resObject);
+  }
+
+  private handleExecuteSqlArgsByResModel(exeCtx: ExecuteContext, executeSqlArgs: any[]): any[] {
+    return executeSqlArgs.map(arg => {
+      if (arg && typeof arg === 'object' && typeof arg.___GENERATE_GET_RES_ATTR___ === 'string') {
+        const word: string = arg.___GENERATE_GET_RES_ATTR___;
+        if (word.startsWith(KeysOfCustomSQL.SQL_PICK_RES_AS_NUMBER)) {
+          const resName = MixinUtils.removeStringPrefix(word, KeysOfCustomSQL.SQL_PICK_RES_AS_NUMBER);
+          this.logger.info('handleExecuteSqlArgsByResModel resName', resName);
+          this.logger.debug('handleExecuteSqlArgsByResModel getResModel', exeCtx.getResModel());
+          return Number(exeCtx.getResModelItemLodash(resName) || 0);
+        }
+        if (word.startsWith(KeysOfCustomSQL.SQL_PICK_RES_AS_STRING)) {
+          const resName = MixinUtils.removeStringPrefix(word, KeysOfCustomSQL.SQL_PICK_RES_AS_STRING);
+          return String(exeCtx.getResModelItemLodash(resName) || '');
+        }
+      }
+      return arg;
+    });
+  }
+
+  private executeSqlCfgModelPreCheck(sqlCfgModel: SqlCfgModel): boolean {
+    const crudType = sqlCfgModel.getCrudType();
+
+    if (KeyOfCrudTypes.EMPTY_SQL === crudType) {
+      // 允许空SQL，没有需要执行的正常。
+      return false;
+    }
+
+    if (KeyOfCrudTypes.NOT_CRUD === crudType) {
+      // 只支持增删改查，其他语句不支持。
+      throw new CommonException(Exceptions.CFG_NOT_SUPPORT_THE_SQL, sqlCfgModel.executeSql);
+    }
+
+    // 此SQL不需要执行
+    if (!this.isNeedExecute(sqlCfgModel)) {
+      return false;
+    }
+
+    // 执行之前，根据配置校验一下此SQL是否可以执行
+    this.assertByValidate(sqlCfgModel);
+
+    return true;
+  }
+
+  private assertByValidate(sqlCfgModel: SqlCfgModel) {
+    const exeCtx = this.getExecuteContext();
+    const funcCfg = sqlCfgModel.validate;
+    if (!funcCfg) {
+      return; // 没有配置。
+    }
+
+    if (!checkFuncCfgValid(funcCfg)) {
+      throw new CommonException(Exceptions.CFG_SQL_CFG_EXECUTE_VALIDATE_EMPTY, {
+        funcCfg: funcCfg,
+        originSql: sqlCfgModel.originSql,
+      });
+    }
+
+    const errMsg = funcCfg.message || '执行前校验失败';
+
+    let result;
+    try {
+      const exeFunCtx = new FuncContext();
+
+      exeFunCtx.currentValue = null;
+      exeFunCtx.reqModel = exeCtx.getReqModel();
+      exeFunCtx.resModel = exeCtx.getResModel();
+
+      result = this.serviceHub.executeFuncCfg(funcCfg, exeFunCtx); // 可以抛出异常
+    } catch (e) {
+      this.logger.debug('assertByValidate', e);
+      throw new CommonException(Exceptions.RUN_EXECUTE_VALIDATE, {
+        funcCfg: funcCfg,
+        originSql: sqlCfgModel.originSql,
+        error: '' + e,
+      });
+    }
+
+    if (!(typeof result === 'boolean')) {
+      // 自定义的校验函数，必须返回boolean类型的树枝
+      throw new CommonException(Exceptions.CFG_IS_NEED_EXECUTE_FUNC_MUST_RETURN_BOOL, {
+        funcCfg: funcCfg,
+        originSql: sqlCfgModel.originSql,
+      });
+    }
+
+    if (result === false) {
+      throw new CommonException(Exceptions.RUN_EXECUTE_VALIDATE, errMsg);
+    }
+  }
+
+  /**
+   * 判断此SQL是否需要执行
+   * @param sqlCfgModel
+   * @param exeCtx
+   * @private
+   */
+  private isNeedExecute(sqlCfgModel: SqlCfgModel) {
+    const exeCtx = this.getExecuteContext();
+    const funcCfg = sqlCfgModel.executeWhen;
+    if (!funcCfg || Object.keys(funcCfg).length === 0) {
+      return true; // 没有配置。默认，需要执行。
+    }
+
+    if (!checkFuncCfgValid(funcCfg)) {
+      throw new CommonException(Exceptions.CFG_CHECK_EXECUTE_WHEN_EMPTY, {
+        funcCfg: funcCfg,
+        originSql: sqlCfgModel.originSql,
+      });
+    }
+
+    let result;
+
+    try {
+      const exeFunCtx = new FuncContext();
+
+      exeFunCtx.currentValue = null;
+      exeFunCtx.reqModel = exeCtx.getReqModel();
+      exeFunCtx.resModel = exeCtx.getResModel();
+
+      result = this.serviceHub.executeFuncCfg(funcCfg, exeFunCtx);
+    } catch (e) {
+      this.logger.debug('isNeedExecute', e);
+      throw new CommonException(Exceptions.RUN_IS_NEED_EXECUTE_ERR, {
+        funcCfg: funcCfg,
+        originSql: sqlCfgModel.originSql,
+        error: '' + e,
+      });
+    }
+
+    if (!(typeof result === 'boolean')) {
+      // 自定义的校验函数，必须返回boolean类型的树枝
+      throw new CommonException(Exceptions.CFG_IS_NEED_EXECUTE_FUNC_MUST_RETURN_BOOL, {
+        funcCfg: funcCfg,
+        originSql: sqlCfgModel.originSql,
+      });
+    }
+
+    return result === true;
+  }
+
+  private toQueryResByResPicker(sqlRes: any, queryRes: any, sqlCfgModel: SqlCfgModel) {
+    const resPicker = sqlCfgModel.resPicker;
+
+    // 返回第一行
+    if (KeysOfSqlResPicker.RESULT_FIRST_ROW === resPicker) {
+      const rows = sqlRes as any[];
+      if (rows && rows.length > 0) {
+        return rows[0];
+      }
+      return null;
+    }
+
+    // $ResultSet[0].total_count ， 只取第一行的的total_count
+    if (KeysOfSqlResPicker.RESULT_TOTAL_COUNT === resPicker) {
+      const rows = sqlRes as any[];
+      if (isEmpty(rows)) {
+        return 0;
+      }
+      const map0 = rows[0];
+      return Number(map0['total_count'] || 0);
+    }
+
+    // 增删改res的内容是修改结果：包括： insert\delete\update
+    if (KeysOfSqlResPicker.UPDATE_RESULT === resPicker) {
+      if (sqlCfgModel.sqlDbType === SqlDbType.postgres) {
+        return {
+          insertId: _.get(queryRes, 'rows[0].id'),
+          affectedRows: _.get(queryRes, 'rowCount'),
+        };
+      }
+
+      if (sqlCfgModel.sqlDbType === SqlDbType.sqlserver) {
+        return {
+          insertId: _.get(queryRes, 'recordset[0].id'),
+          affectedRows: _.get(queryRes, 'rowsAffected[0]'),
+        };
+      }
+
+      return sqlRes;
+    }
+
+    //其他配置：形如： sqlRes[0].total_count
+    if (typeof resPicker === 'string') {
+      if (resPicker.startsWith('sqlRes')) {
+        return _.get({ sqlRes }, resPicker);
+      } else {
+        throw new CommonException(Exceptions.RUN_EXECUTE_VALIDATE, 'resPicker必须是以sqlRes开头');
+      }
+    }
+
+    return sqlRes;
+  }
+}
+
+export { CrudProExecuteSqlService };