midway-fatcms 0.0.1-beta.22 → 0.0.1-beta.25

package/src/service/proxyapi/ProxyApiLoadService.ts

@@ -1,174 +1,174 @@
-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);
-      }
-    }
-  }
-
-}
+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);
+      }
+    }
+  }
+
+}