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

package/src/service/curd/fixCfgModel.ts

@@ -0,0 +1,120 @@
+import * as _ from 'lodash';
+import { CTX_VISITOR_ACCOUNT_TYPE, CTX_VISITOR_AVATAR, CTX_VISITOR_ID, CTX_VISITOR_NICKNAME, IRequestCfgModel2 } from '@/models/bizmodels';
+import { KeysOfSimpleSQL } from '@/libs/crud-pro/models/keys';
+
+function isSimpleQuery(sqlSimpleName: KeysOfSimpleSQL): boolean {
+  return sqlSimpleName === KeysOfSimpleSQL.SIMPLE_QUERY || sqlSimpleName === KeysOfSimpleSQL.SIMPLE_QUERY_COUNT || sqlSimpleName === KeysOfSimpleSQL.SIMPLE_QUERY_PAGE || sqlSimpleName === KeysOfSimpleSQL.SIMPLE_QUERY_ONE;
+}
+
+function fixCfgModel(cfgModel: IRequestCfgModel2) {
+  if (!cfgModel.method) {
+    cfgModel.method = 'anonymous';
+  }
+
+  if (!cfgModel.updateCfg) {
+    cfgModel.updateCfg = {};
+  }
+
+  const sqlSimpleName = cfgModel.sqlSimpleName;
+
+  const enableStandardUpdateCfg = cfgModel.enableStandardUpdateCfg; // 用于设置data字段
+  const enableStandardUpdateCfgCondition = cfgModel.enableStandardUpdateCfgCondition; // 用于设置condition字段
+
+  // 彻底关闭
+  if (enableStandardUpdateCfg === false) {
+    return;
+  }
+
+  const getDataCfgArray = () => {
+    // 明确有配置
+    if (Array.isArray(enableStandardUpdateCfg)) {
+      if (enableStandardUpdateCfg.includes('null')) {
+        return [];
+      }
+      return enableStandardUpdateCfg;
+    }
+    // update 、insert 添加 by
+    return ['by']; // 创建/修改语句默认添加by
+  };
+
+  const getConditionCfgArray = () => {
+    // 明确有配置
+    if (Array.isArray(enableStandardUpdateCfgCondition)) {
+      if (enableStandardUpdateCfgCondition.includes('null')) {
+        return [];
+      }
+      return enableStandardUpdateCfgCondition;
+    }
+    return [];
+  };
+
+  const dataCfgArray = getDataCfgArray();
+  const conditionCfgArray = getConditionCfgArray();
+
+  const mapping = {
+    'condition.created_by': { contextAsString: CTX_VISITOR_ID },
+    'condition.created_account_type': { contextAsString: CTX_VISITOR_ACCOUNT_TYPE },
+
+    'data.created_by': { contextAsString: CTX_VISITOR_ID },
+    'data.created_avatar': { contextAsString: CTX_VISITOR_AVATAR },
+    'data.created_nickname': { contextAsString: CTX_VISITOR_NICKNAME },
+    'data.created_account_type': { contextAsString: CTX_VISITOR_ACCOUNT_TYPE },
+
+    'data.modified_by': { contextAsString: CTX_VISITOR_ID },
+    'data.modified_avatar': { contextAsString: CTX_VISITOR_AVATAR },
+    'data.modified_nickname': { contextAsString: CTX_VISITOR_NICKNAME },
+    'data.modified_account_type': { contextAsString: CTX_VISITOR_ACCOUNT_TYPE },
+  };
+
+  const buildUpdateCfgBy = (cfgArray: string[], prefix: string): any => {
+    const obj: any = {};
+    if (Array.isArray(cfgArray)) {
+      for (let i = 0; i < cfgArray.length; i++) {
+        const suffix = cfgArray[i];
+        if (suffix) {
+          const key = `${prefix}_${suffix}`;
+          obj[key] = mapping[key];
+        }
+      }
+    }
+    return obj;
+  };
+
+  // 查询语句
+  if (isSimpleQuery(sqlSimpleName)) {
+    const updateCfgObj = buildUpdateCfgBy(conditionCfgArray, 'condition.created'); // 查询用户本人创建的
+    _.merge(cfgModel.updateCfg, updateCfgObj);
+  }
+
+  // 删除语句
+  if (sqlSimpleName === KeysOfSimpleSQL.SIMPLE_DELETE) {
+    const updateCfgObj = buildUpdateCfgBy(conditionCfgArray, 'condition.created'); // 删除用户本人创建的
+    _.merge(cfgModel.updateCfg, updateCfgObj);
+  }
+
+  // 插入语句
+  if (sqlSimpleName === KeysOfSimpleSQL.SIMPLE_INSERT) {
+    const updateCfgObj1 = buildUpdateCfgBy(dataCfgArray, 'data.created');
+    _.merge(cfgModel.updateCfg, updateCfgObj1);
+  }
+
+  // 更新语句
+  if (sqlSimpleName === KeysOfSimpleSQL.SIMPLE_UPDATE) {
+    const updateCfgObj1 = buildUpdateCfgBy(dataCfgArray, 'data.modified');
+    const updateCfgObj2 = buildUpdateCfgBy(conditionCfgArray, 'condition.created'); // 更新用户本人创建的
+    _.merge(cfgModel.updateCfg, updateCfgObj1);
+    _.merge(cfgModel.updateCfg, updateCfgObj2);
+  }
+
+  // 插入或更新
+  if (sqlSimpleName === KeysOfSimpleSQL.SIMPLE_INSERT_OR_UPDATE || sqlSimpleName === KeysOfSimpleSQL.SIMPLE_INSERT_ON_DUPLICATE_UPDATE) {
+    const updateCfgObj1 = buildUpdateCfgBy(dataCfgArray, 'data.created');
+    const updateCfgObj2 = buildUpdateCfgBy(dataCfgArray, 'data.modified');
+    const updateCfgObj3 = buildUpdateCfgBy(conditionCfgArray, 'condition.created'); // 更新用户本人创建的
+    _.merge(cfgModel.updateCfg, updateCfgObj1);
+    _.merge(cfgModel.updateCfg, updateCfgObj2);
+    _.merge(cfgModel.updateCfg, updateCfgObj3);
+  }
+}
+
+export { fixCfgModel };

package/src/service/proxyapi/ProxyApiLoadService.ts

@@ -1,173 +1,170 @@
-import { Inject, Provide } from '@midwayjs/core';
-import { Context } from '@midwayjs/koa';
-import * as _ from 'lodash';
-import { BaseService } from '../base/BaseService';
-import { getRouteTrie, RouteTrie } from './RouteTrie';
-import { RouteHandler } from './RouteHandler';
-import { CurdProService } from '../curd/CurdProService';
-import { SystemTables} from '../../models/SystemTables';
-import { KeysOfSimpleSQL } from '../../libs/crud-pro/models/keys';
-import { IProxyApiEntity, IUpstreamInfo } from '../../models/SystemEntities';
-import { parseJsonObject } from '../../libs/utils/functions';
-import { BALANCE_STRATEGY } from './ProxyApiUtils';
-import { WeightedRandom } from './WeightedRandom';
-import { WeightedRoundRobin } from './WeightedRoundRobin';
-import { IScheduleService } from '../../interface';
-import { WorkbenchService } from '../WorkbenchService';
-import { CommonException } from '../../libs/crud-pro/exceptions';
-import { GLOBAL_STATIC_CONFIG } from '@/libs/global-config/global-config';
-
-function toUpstreamInfo(config_content: any): IUpstreamInfo {
-  const obj = parseJsonObject(config_content);
-  if (!obj) {
-    return {
-      balanceStrategy: BALANCE_STRATEGY.ROUND_ROBIN,
-      targetList: [],
-    };
-  }
-  const targetList = _.get(obj, 'data.targetList', []);
-  const balanceStrategy = _.get(obj, 'data.baseInfo.balanceStrategy');
-  if (Array.isArray(targetList)) {
-    const targetListParsed = targetList
-      .map((item: any) => {
-        return {
-          origin: item.origin,
-          weight: item.weight || 1,
-        };
-      })
-      .filter(item => {
-        const origin: string = item?.origin || '';
-        return origin.startsWith('http://') || origin.startsWith('https://');
-      });
-
-    return {
-      balanceStrategy,
-      targetList: targetListParsed,
-    };
-  }
-
-  return {
-    balanceStrategy: BALANCE_STRATEGY.ROUND_ROBIN,
-    targetList: [],
-  };
-}
-
-/**
- * 接口代理。 数据加载
- */
-@Provide()
-export class ProxyApiLoadService extends BaseService implements IScheduleService {
-  @Inject()
-  protected ctx: Context;
-
-  @Inject()
-  private curdProService: CurdProService;
-
-  @Inject()
-  private workbenchService: WorkbenchService;
-
-  /**
-   * 创建路由树。刷新路由缓存
-   * @param workbench_code
-   */
-  public async buildRouteTrie(workbench_code: string): Promise<RouteTrie> {
-    const upstreamMap: Record<string, IUpstreamInfo> = await this.buildUpstreamMap();
-    const routeTrie = getRouteTrie(workbench_code);
-    routeTrie.clearRoute();
-    const entities = await this.loadProxyApiEntity(workbench_code, upstreamMap);
-    for (let i = 0; i < entities.length; i++) {
-      const entity = entities[i];
-      if (entity.status === 1) {
-        const path_prefix2 = '/' + entity.upstream_code + entity.path_prefix;
-        routeTrie.addRoute(path_prefix2, new RouteHandler(entity));
-      }
-    }
-    return routeTrie;
-  }
-
-  private async buildUpstreamMap() {
-    
-    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
-
-    //       biz_tag={'upstream'}
-    const res = await this.curdProService.executeCrudByCfg(
-      {
-        condition: {
-          biz_tag: 'upstream',
-        },
-      },
-      {
-        sqlTable: SystemTables.sys_configs,
-        sqlDatabase: SystemDbName,
-        sqlDdType: SystemDbType,
-        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
-      }
-    );
-    const upstreamMap: Record<string, IUpstreamInfo> = {};
-    const rows = res.getResRows() as any[];
-    for (let i = 0; i < rows.length; i++) {
-      const row = rows[i];
-      const { config_code, config_content } = row || {};
-      upstreamMap[config_code] = toUpstreamInfo(config_content);
-    }
-    return upstreamMap;
-  }
-
-  private async loadProxyApiEntity(workbench_code: string, upstreamMap: Record<string, IUpstreamInfo>): Promise<IProxyApiEntity[]> {
-    
-    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
-
-    
-    const res = await this.curdProService.executeCrudByCfg(
-      {
-        condition: {},
-      },
-      {
-        sqlTable: SystemTables.sys_proxyapi,
-        sqlDatabase: SystemDbName,
-        sqlDdType: SystemDbType,
-        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
-      }
-    );
-
-    const rows0 = res.getResRows() as IProxyApiEntity[];
-
-    const rows = rows0.filter(s => {
-      if (!s.workbench_code_array) {
-        throw new CommonException('IS_SUPPORT_CURRENT_WORKBENCH', '未配置支持的站点字段');
-      }
-      return this.workbenchService.isMatchWorkbenchCode(s.workbench_code_array, workbench_code);
-    });
-
-    for (let i = 0; i < rows.length; i++) {
-      const row = rows[i];
-      const upstream_code = row.upstream_code;
-      this.addUpstreamInfo(row, upstream_code, upstreamMap);
-    }
-    return rows;
-  }
-
-  private addUpstreamInfo(row: IProxyApiEntity, upstream_code: string, upstreamMap: Record<string, IUpstreamInfo>) {
-    const upstreamInfo = upstreamMap[upstream_code];
-    if (!upstreamInfo) {
-      return;
-    }
-    row.targetIndex = 0;
-    row.targetList = upstreamInfo.targetList || [];
-    row.balanceStrategy = upstreamInfo.balanceStrategy || BALANCE_STRATEGY.ROUND_ROBIN;
-    row.weightedRandom = new WeightedRandom(upstreamInfo.targetList || []);
-    row.weightedRoundRobin = new WeightedRoundRobin(upstreamInfo.targetList || []);
-  }
-
-  public async runBySchedule(): Promise<any> {
-    const workbenchList = await this.workbenchService.getAllWorkbenchInfoList();
-    for (let i = 0; i < workbenchList.length; i++) {
-      const workbenchInfo = workbenchList[i];
-      try {
-        await this.buildRouteTrie(workbenchInfo.workbench_code);
-      } catch (e) {
-        this.logError('ProxyApiLoadService_runBySchedule', e);
-      }
-    }
-  }
-}
+import { Inject, Provide } from '@midwayjs/core';
+import { Context } from '@midwayjs/koa';
+import * as _ from 'lodash';
+import { BaseService } from '../base/BaseService';
+import { getRouteTrie, RouteTrie } from './RouteTrie';
+import { RouteHandler } from './RouteHandler';
+import { CurdProService } from '../curd/CurdProService';
+import { SystemTables } from '../../models/SystemTables';
+import { KeysOfSimpleSQL } from '../../libs/crud-pro/models/keys';
+import { IProxyApiEntity, IUpstreamInfo } from '../../models/SystemEntities';
+import { parseJsonObject } from '../../libs/utils/functions';
+import { BALANCE_STRATEGY } from './ProxyApiUtils';
+import { WeightedRandom } from './WeightedRandom';
+import { WeightedRoundRobin } from './WeightedRoundRobin';
+import { IScheduleService } from '../../interface';
+import { WorkbenchService } from '../WorkbenchService';
+import { CommonException } from '../../libs/crud-pro/exceptions';
+import { GLOBAL_STATIC_CONFIG } from '@/libs/global-config/global-config';
+
+function toUpstreamInfo(config_content: any): IUpstreamInfo {
+  const obj = parseJsonObject(config_content);
+  if (!obj) {
+    return {
+      balanceStrategy: BALANCE_STRATEGY.ROUND_ROBIN,
+      targetList: [],
+    };
+  }
+  const targetList = _.get(obj, 'data.targetList', []);
+  const balanceStrategy = _.get(obj, 'data.baseInfo.balanceStrategy');
+  if (Array.isArray(targetList)) {
+    const targetListParsed = targetList
+      .map((item: any) => {
+        return {
+          origin: item.origin,
+          weight: item.weight || 1,
+        };
+      })
+      .filter(item => {
+        const origin: string = item?.origin || '';
+        return origin.startsWith('http://') || origin.startsWith('https://');
+      });
+
+    return {
+      balanceStrategy,
+      targetList: targetListParsed,
+    };
+  }
+
+  return {
+    balanceStrategy: BALANCE_STRATEGY.ROUND_ROBIN,
+    targetList: [],
+  };
+}
+
+/**
+ * 接口代理。 数据加载
+ */
+@Provide()
+export class ProxyApiLoadService extends BaseService implements IScheduleService {
+  @Inject()
+  protected ctx: Context;
+
+  @Inject()
+  private curdProService: CurdProService;
+
+  @Inject()
+  private workbenchService: WorkbenchService;
+
+  /**
+   * 创建路由树。刷新路由缓存
+   * @param workbench_code
+   */
+  public async buildRouteTrie(workbench_code: string): Promise<RouteTrie> {
+    const upstreamMap: Record<string, IUpstreamInfo> = await this.buildUpstreamMap();
+    const routeTrie = getRouteTrie(workbench_code);
+    routeTrie.clearRoute();
+    const entities = await this.loadProxyApiEntity(workbench_code, upstreamMap);
+    for (let i = 0; i < entities.length; i++) {
+      const entity = entities[i];
+      if (entity.status === 1) {
+        const path_prefix2 = '/' + entity.upstream_code + entity.path_prefix;
+        routeTrie.addRoute(path_prefix2, new RouteHandler(entity));
+      }
+    }
+    return routeTrie;
+  }
+
+  private async buildUpstreamMap() {
+    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
+
+    //       biz_tag={'upstream'}
+    const res = await this.curdProService.executeCrudByCfg(
+      {
+        condition: {
+          biz_tag: 'upstream',
+        },
+      },
+      {
+        sqlTable: SystemTables.sys_configs,
+        sqlDatabase: SystemDbName,
+        sqlDbType: SystemDbType,
+        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
+      }
+    );
+    const upstreamMap: Record<string, IUpstreamInfo> = {};
+    const rows = res.getResRows() as any[];
+    for (let i = 0; i < rows.length; i++) {
+      const row = rows[i];
+      const { config_code, config_content } = row || {};
+      upstreamMap[config_code] = toUpstreamInfo(config_content);
+    }
+    return upstreamMap;
+  }
+
+  private async loadProxyApiEntity(workbench_code: string, upstreamMap: Record<string, IUpstreamInfo>): Promise<IProxyApiEntity[]> {
+    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
+
+    const res = await this.curdProService.executeCrudByCfg(
+      {
+        condition: {},
+      },
+      {
+        sqlTable: SystemTables.sys_proxyapi,
+        sqlDatabase: SystemDbName,
+        sqlDbType: SystemDbType,
+        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
+      }
+    );
+
+    const rows0 = res.getResRows() as IProxyApiEntity[];
+
+    const rows = rows0.filter(s => {
+      if (!s.workbench_code_array) {
+        throw new CommonException('IS_SUPPORT_CURRENT_WORKBENCH', '未配置支持的站点字段');
+      }
+      return this.workbenchService.isMatchWorkbenchCode(s.workbench_code_array, workbench_code);
+    });
+
+    for (let i = 0; i < rows.length; i++) {
+      const row = rows[i];
+      const upstream_code = row.upstream_code;
+      this.addUpstreamInfo(row, upstream_code, upstreamMap);
+    }
+    return rows;
+  }
+
+  private addUpstreamInfo(row: IProxyApiEntity, upstream_code: string, upstreamMap: Record<string, IUpstreamInfo>) {
+    const upstreamInfo = upstreamMap[upstream_code];
+    if (!upstreamInfo) {
+      return;
+    }
+    row.targetIndex = 0;
+    row.targetList = upstreamInfo.targetList || [];
+    row.balanceStrategy = upstreamInfo.balanceStrategy || BALANCE_STRATEGY.ROUND_ROBIN;
+    row.weightedRandom = new WeightedRandom(upstreamInfo.targetList || []);
+    row.weightedRoundRobin = new WeightedRoundRobin(upstreamInfo.targetList || []);
+  }
+
+  public async runBySchedule(): Promise<any> {
+    const workbenchList = await this.workbenchService.getAllWorkbenchInfoList();
+    for (let i = 0; i < workbenchList.length; i++) {
+      const workbenchInfo = workbenchList[i];
+      try {
+        await this.buildRouteTrie(workbenchInfo.workbench_code);
+      } catch (e) {
+        this.logError('ProxyApiLoadService_runBySchedule', e);
+      }
+    }
+  }
+}