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

package/src/models/userSession.ts

@@ -1,216 +1,221 @@
-import * as _ from 'lodash';
-import { KeysOfAuthType } from '../libs/crud-pro/models/keys';
-import { MixinUtils } from '../libs/crud-pro/utils/MixinUtils';
-import { CommonException, Exceptions } from '../libs/crud-pro/exceptions';
-import { parseStringTrimArray } from '../libs/utils/functions';
-
-const sessionCookieCfg = {
-  signed: true,
-  encrypt: true,
-  httpOnly: true,
-  maxAge: 1000 * 3600 * 24 * 365,
-};
-
-const SESSION_ID_KEY = 'fatcmssessionid';
-const SYS_ACCOUNT_TYPE = 'fatcms';
-
-interface ISessionInfo {
-  nickName: string;
-  avatar: string;
-  roleCodes: string[];
-  functionCodes: string[];
-  loginName: string;
-  sessionId: string;
-  accountId: string;
-  workbenchCode: string;
-  accountType: string;
-}
-
-interface IConsumerUserInfo {
-  workbenchCode: string;
-  accountType: string;
-  accountId: string;
-  nickName: string;
-  loginName: string;
-  avatar: string;
-}
-
-interface IBatchCheckProps {
-  permissionCodeList?: string[];
-  roleCodeList?: string[];
-}
-
-interface IBatchCheckRes {
-  permissionCodeResult: Record<string, boolean>;
-  roleCodeResult: Record<string, boolean>;
-}
-
-function createEmptySessionInfo(superAdmin: boolean): ISessionInfo {
-  return {
-    nickName: superAdmin ? 'SA' : '',
-    avatar: '',
-    roleCodes: [],
-    functionCodes: [],
-    loginName: '',
-    sessionId: '',
-    accountId: '',
-    workbenchCode: '',
-    accountType: '',
-  };
-}
-
-class UserSessionInfo {
-  private readonly sessionInfo: ISessionInfo;
-
-  // 是否是超级管理员；拥有一切权限。紧急情况下使用
-  private readonly superAdmin: boolean;
-  constructor(sessionInfo: ISessionInfo, superAdmin?: boolean) {
-    this.superAdmin = superAdmin === true;
-    if (sessionInfo) {
-      this.sessionInfo = sessionInfo;
-    } else {
-      this.sessionInfo = createEmptySessionInfo(superAdmin);
-    }
-  }
-
-  getSessionInfo(): ISessionInfo {
-    return this.sessionInfo;
-  }
-
-  getSessionValue(key: string): any {
-    return _.get(this.sessionInfo, key);
-  }
-
-  isSuperAdmin(): boolean {
-    return this.superAdmin;
-  }
-
-  /**
-   * 是否是我自己的账号
-   * @param accountId
-   */
-  isMySelf(accountId: string): boolean {
-    const s = this.sessionInfo;
-    return s && accountId && s.accountId === accountId;
-  }
-
-  isLogin(): boolean {
-    const s = this.sessionInfo;
-    return !!(s && s.sessionId && s.accountId && s.sessionId.length > 1);
-  }
-
-  hasAnyPermission(funcCodeList: string | string[]): boolean {
-    if (this.isSuperAdmin()) {
-      return true;
-    }
-    if (typeof funcCodeList === 'string') {
-      return this.hasPermission(funcCodeList);
-    }
-
-    if (Array.isArray(funcCodeList)) {
-      for (let i = 0; i < funcCodeList.length; i++) {
-        const funcCode = funcCodeList[i];
-        if (this.hasPermission(funcCode)) {
-          return true;
-        }
-      }
-      return false;
-    }
-  }
-
-  hasPermission(funcCode: string): boolean {
-    if (this.isSuperAdmin()) {
-      return true;
-    }
-    const ss = this.sessionInfo.functionCodes || [];
-    return ss.includes(funcCode);
-  }
-
-  hasAnyRole(roleCodeList: string | string[]): boolean {
-    if (this.isSuperAdmin()) {
-      return true;
-    }
-    if (typeof roleCodeList === 'string') {
-      return this.hasRole(roleCodeList);
-    }
-
-    if (Array.isArray(roleCodeList)) {
-      for (let i = 0; i < roleCodeList.length; i++) {
-        const funcCode = roleCodeList[i];
-        if (this.hasRole(funcCode)) {
-          return true;
-        }
-      }
-      return false;
-    }
-  }
-
-  hasRole(roleCode: string): boolean {
-    if (this.isSuperAdmin()) {
-      return true;
-    }
-    const ss = this.sessionInfo.roleCodes || [];
-    return ss.includes(roleCode);
-  }
-
-  isAuthPass(authType: KeysOfAuthType, authConfig: string | string[]): boolean {
-    if (authType === KeysOfAuthType.free || !authType) {
-      return true; // 无需校验
-    }
-    if (!this.isLogin()) {
-      return false;
-    }
-    if (this.isSuperAdmin()) {
-      return true;
-    }
-
-    if (authType === KeysOfAuthType.byFuncCode) {
-      const funcCodes = parseStringTrimArray(authConfig);
-      if (MixinUtils.isEmpty(funcCodes)) {
-        throw new CommonException(Exceptions.CFG_AUTH_CODE_EMPTY, '缺少权限编码配置');
-      }
-      return this.hasAnyPermission(funcCodes);
-    }
-
-    if (authType === KeysOfAuthType.byRoleCode) {
-      const roleCodes = parseStringTrimArray(authConfig);
-      if (MixinUtils.isEmpty(roleCodes)) {
-        throw new CommonException(Exceptions.CFG_AUTH_CODE_EMPTY, '缺少角色编码配置');
-      }
-      return this.hasAnyRole(roleCodes);
-    }
-
-    return true;
-  }
-
-  /**
-   * 批量检查权限
-   * @param params
-   */
-  batchCheckPermission(params: IBatchCheckProps): IBatchCheckRes {
-    const { permissionCodeList, roleCodeList } = params;
-    const permissionCodeResult: Record<string, boolean> = {};
-    const roleCodeResult: Record<string, boolean> = {};
-
-    if (Array.isArray(permissionCodeList)) {
-      for (let i = 0; i < permissionCodeList.length; i++) {
-        const permissionCode = permissionCodeList[i];
-        if (permissionCode && typeof permissionCode === 'string') {
-          permissionCodeResult[permissionCode] = this.hasPermission(permissionCode);
-        }
-      }
-    }
-
-    if (Array.isArray(roleCodeList)) {
-      for (let i = 0; i < roleCodeList.length; i++) {
-        const roleCode = roleCodeList[i];
-        if (roleCode && typeof roleCode === 'string') {
-          roleCodeResult[roleCode] = this.hasRole(roleCode);
-        }
-      }
-    }
-    return { permissionCodeResult, roleCodeResult };
-  }
-}
-
-export { ISessionInfo, UserSessionInfo, SESSION_ID_KEY, SYS_ACCOUNT_TYPE, IConsumerUserInfo, sessionCookieCfg };
+import * as _ from 'lodash';
+import { KeysOfAuthType } from '@/libs/crud-pro/models/keys';
+import { MixinUtils } from '@/libs/crud-pro/utils/MixinUtils';
+import { CommonException, Exceptions } from '@/libs/crud-pro/exceptions';
+import { parseStringTrimArray } from '@/libs/utils/functions';
+
+const sessionCookieCfg = {
+  signed: true,
+  encrypt: true,
+  httpOnly: true,
+  maxAge: 1000 * 3600 * 24 * 365,
+};
+
+const SESSION_ID_KEY = 'fatcmssessionid';
+const SYS_ACCOUNT_TYPE = 'fatcms';
+
+interface ISessionInfo {
+  nickName: string;
+  avatar: string;
+  roleCodes: string[];
+  functionCodes: string[];
+  loginName: string;
+  sessionId: string;
+  accountId: string;
+  workbenchCode: string;
+  accountType: string;
+  publicKey: string; // 公钥
+  privateKey: string; // 私钥
+  bizExt?: any;
+}
+
+interface IConsumerUserInfo {
+  workbenchCode: string;
+  accountType: string;
+  accountId: string;
+  nickName: string;
+  loginName: string;
+  avatar: string;
+}
+
+interface IBatchCheckProps {
+  permissionCodeList?: string[];
+  roleCodeList?: string[];
+}
+
+interface IBatchCheckRes {
+  permissionCodeResult: Record<string, boolean>;
+  roleCodeResult: Record<string, boolean>;
+}
+
+function createEmptySessionInfo(superAdmin: boolean): ISessionInfo {
+  return {
+    nickName: superAdmin ? 'SA' : '',
+    avatar: '',
+    roleCodes: [],
+    functionCodes: [],
+    loginName: '',
+    sessionId: '',
+    accountId: '',
+    workbenchCode: '',
+    accountType: '',
+    publicKey: '',
+    privateKey: '',
+  };
+}
+
+class UserSessionInfo {
+  private readonly sessionInfo: ISessionInfo;
+
+  // 是否是超级管理员；拥有一切权限。紧急情况下使用
+  private readonly superAdmin: boolean;
+  constructor(sessionInfo: ISessionInfo, superAdmin?: boolean) {
+    this.superAdmin = superAdmin === true;
+    if (sessionInfo) {
+      this.sessionInfo = sessionInfo;
+    } else {
+      this.sessionInfo = createEmptySessionInfo(superAdmin);
+    }
+  }
+
+  getSessionInfo(): ISessionInfo {
+    return this.sessionInfo;
+  }
+
+  getSessionValue(key: string): any {
+    return _.get(this.sessionInfo, key);
+  }
+
+  isSuperAdmin(): boolean {
+    return this.superAdmin;
+  }
+
+  /**
+   * 是否是我自己的账号
+   * @param accountId
+   */
+  isMySelf(accountId: string): boolean {
+    const s = this.sessionInfo;
+    return s && accountId && s.accountId === accountId;
+  }
+
+  isLogin(): boolean {
+    const s = this.sessionInfo;
+    return !!(s && s.sessionId && s.accountId && s.sessionId.length > 1);
+  }
+
+  hasAnyPermission(funcCodeList: string | string[]): boolean {
+    if (this.isSuperAdmin()) {
+      return true;
+    }
+    if (typeof funcCodeList === 'string') {
+      return this.hasPermission(funcCodeList);
+    }
+
+    if (Array.isArray(funcCodeList)) {
+      for (let i = 0; i < funcCodeList.length; i++) {
+        const funcCode = funcCodeList[i];
+        if (this.hasPermission(funcCode)) {
+          return true;
+        }
+      }
+      return false;
+    }
+  }
+
+  hasPermission(funcCode: string): boolean {
+    if (this.isSuperAdmin()) {
+      return true;
+    }
+    const ss = this.sessionInfo.functionCodes || [];
+    return ss.includes(funcCode);
+  }
+
+  hasAnyRole(roleCodeList: string | string[]): boolean {
+    if (this.isSuperAdmin()) {
+      return true;
+    }
+    if (typeof roleCodeList === 'string') {
+      return this.hasRole(roleCodeList);
+    }
+
+    if (Array.isArray(roleCodeList)) {
+      for (let i = 0; i < roleCodeList.length; i++) {
+        const funcCode = roleCodeList[i];
+        if (this.hasRole(funcCode)) {
+          return true;
+        }
+      }
+      return false;
+    }
+  }
+
+  hasRole(roleCode: string): boolean {
+    if (this.isSuperAdmin()) {
+      return true;
+    }
+    const ss = this.sessionInfo.roleCodes || [];
+    return ss.includes(roleCode);
+  }
+
+  isAuthPass(authType: KeysOfAuthType, authConfig: string | string[]): boolean {
+    if (authType === KeysOfAuthType.free || !authType) {
+      return true; // 无需校验
+    }
+    if (!this.isLogin()) {
+      return false;
+    }
+    if (this.isSuperAdmin()) {
+      return true;
+    }
+
+    if (authType === KeysOfAuthType.byFuncCode) {
+      const funcCodes = parseStringTrimArray(authConfig);
+      if (MixinUtils.isEmpty(funcCodes)) {
+        throw new CommonException(Exceptions.CFG_AUTH_CODE_EMPTY, '缺少权限编码配置');
+      }
+      return this.hasAnyPermission(funcCodes);
+    }
+
+    if (authType === KeysOfAuthType.byRoleCode) {
+      const roleCodes = parseStringTrimArray(authConfig);
+      if (MixinUtils.isEmpty(roleCodes)) {
+        throw new CommonException(Exceptions.CFG_AUTH_CODE_EMPTY, '缺少角色编码配置');
+      }
+      return this.hasAnyRole(roleCodes);
+    }
+
+    return true;
+  }
+
+  /**
+   * 批量检查权限
+   * @param params
+   */
+  batchCheckPermission(params: IBatchCheckProps): IBatchCheckRes {
+    const { permissionCodeList, roleCodeList } = params;
+    const permissionCodeResult: Record<string, boolean> = {};
+    const roleCodeResult: Record<string, boolean> = {};
+
+    if (Array.isArray(permissionCodeList)) {
+      for (let i = 0; i < permissionCodeList.length; i++) {
+        const permissionCode = permissionCodeList[i];
+        if (permissionCode && typeof permissionCode === 'string') {
+          permissionCodeResult[permissionCode] = this.hasPermission(permissionCode);
+        }
+      }
+    }
+
+    if (Array.isArray(roleCodeList)) {
+      for (let i = 0; i < roleCodeList.length; i++) {
+        const roleCode = roleCodeList[i];
+        if (roleCode && typeof roleCode === 'string') {
+          roleCodeResult[roleCode] = this.hasRole(roleCode);
+        }
+      }
+    }
+    return { permissionCodeResult, roleCodeResult };
+  }
+}
+
+export { ISessionInfo, UserSessionInfo, SESSION_ID_KEY, SYS_ACCOUNT_TYPE, IConsumerUserInfo, sessionCookieCfg };

package/src/schedule/anonymousContext.ts

@@ -0,0 +1,75 @@
+import * as koa from '@midwayjs/koa';
+import { ContextLogger } from '@/models/contextLogger';
+import { Transaction } from '@/libs/crud-pro/models/Transaction';
+import { ISessionInfo, SYS_ACCOUNT_TYPE, UserSessionInfo } from '@/models/userSession';
+import { IWorkbenchEntity } from '@/models/SystemEntities';
+
+const anonymousUserInfo: ISessionInfo = {
+  nickName: 'anonymous_user',
+  avatar: '',
+  roleCodes: [],
+  functionCodes: [],
+  loginName: 'anonymous_user',
+  sessionId: '',
+  accountId: '',
+  workbenchCode: '',
+  accountType: SYS_ACCOUNT_TYPE,
+  publicKey: '',
+  privateKey: '',
+};
+
+const anonymousWorkbenchInfo: IWorkbenchEntity = {
+  id: 0,
+  workbench_code: 'anonymous_workbench',
+  workbench_name: 'anonymous_workbench',
+  workbench_domain: 'anonymous_workbench',
+  workbench_desc: 'anonymous_workbench',
+  html_content: 'anonymous_workbench',
+  status: 1,
+  workbench_type: 0,
+};
+
+export type RunServiceAtAnonymousContextRunner = (ctx: koa.IMidwayKoaContext) => any;
+export interface RunServiceAtAnonymousContextRes {
+  result: any;
+  error: any;
+}
+
+class AnonymousContext {
+  private app: koa.Application;
+
+  setApp(app: koa.Application) {
+    this.app = app;
+  }
+
+  getApp() {
+    return this.app;
+  }
+
+  async runServiceAtAnonymousContext(runner: RunServiceAtAnonymousContextRunner): Promise<RunServiceAtAnonymousContextRes> {
+    const app = this.app;
+    const ctx: koa.IMidwayKoaContext = app.createAnonymousContext();
+
+    ctx.app = app;
+    ctx.contextLogger = new ContextLogger(ctx as any);
+    ctx.transaction = new Transaction();
+    ctx.userSession = new UserSessionInfo(anonymousUserInfo, true);
+    ctx.workbenchInfo = anonymousWorkbenchInfo;
+
+    let result: any = null;
+    let error: any = null;
+    try {
+      result = await runner(ctx);
+    } catch (e) {
+      error = e;
+      ctx.logger.error('runServiceAtAnonymousContext error ', e);
+    } finally {
+      await ctx.transaction.releaseTx();
+    }
+    return { result: result, error: error };
+  }
+}
+
+const ANONYMOUS_CONTEXT = new AnonymousContext();
+
+export { ANONYMOUS_CONTEXT };

package/src/schedule/index.ts

@@ -1,73 +1,5 @@
-import * as koa from '@midwayjs/koa';
-import { IScheduleService } from '../interface';
-import { ContextLogger } from '../models/contextLogger';
-import { Transaction } from '../libs/crud-pro/models/Transaction';
-import {ISessionInfo, SYS_ACCOUNT_TYPE, UserSessionInfo} from '../models/userSession';
-import { IWorkbenchEntity } from '../models/SystemEntities';
-
-const scheduleUserInfo:ISessionInfo = {
-  nickName: 'schedule_user',
-  avatar: '',
-  roleCodes: [],
-  functionCodes: [],
-  loginName: 'schedule_user',
-  sessionId: '',
-  accountId: '',
-  workbenchCode: '',
-  accountType: SYS_ACCOUNT_TYPE,
-};
-
-const scheduleWorkbenchInfo: IWorkbenchEntity = {
-  id: 0,
-  workbench_code: 'schedule_workbench',
-  workbench_name: 'schedule_workbench',
-  workbench_domain: 'schedule_workbench',
-  workbench_desc: 'schedule_workbench',
-  html_content: 'schedule_workbench',
-  status: 1,
-  workbench_type: 0,
-};
-
-async function runScheduleService(app: koa.Application, serviceName: string) {
-  const ctx = app.createAnonymousContext();
-
-  ctx.app = app;
-  ctx.contextLogger = new ContextLogger(ctx as any);
-  ctx.transaction = new Transaction();
-  ctx.userSession = new UserSessionInfo(scheduleUserInfo, true);
-  ctx.workbenchInfo = scheduleWorkbenchInfo;
-
-  try {
-    const serviceObject: IScheduleService = await ctx.requestContext.getAsync(serviceName);
-    if (serviceObject && serviceObject.runBySchedule) {
-      await serviceObject.runBySchedule();
-    } else {
-      ctx.logger.error(`runSchedule error , serviceName = ${serviceName}, 没有找到服务`);
-    }
-  } catch (e) {
-    ctx.logger.error(`runSchedule error , serviceName = ${serviceName}`, e);
-  } finally {
-    await ctx.transaction.releaseTx();
-  }
-}
-
-async function runSchedule(app: koa.Application, serviceList: string[]) {
-  for (let i = 0; i < serviceList.length; i++) {
-    const serviceName = serviceList[i];
-    try {
-      await runScheduleService(app, serviceName);
-    } catch (e) {
-      const ctx = app.createAnonymousContext();
-      ctx.logger.error(`runSchedule error , serviceName = ${serviceName}`, e);
-    }
-  }
-}
-
-async function startSchedule(app: koa.Application, serviceList: string[]) {
-  await runSchedule(app, serviceList);
-  setInterval(() => {
-    runSchedule(app, serviceList);
-  }, 2 * 60 * 1000);
-}
-
-export { startSchedule };
+import { SCHEDULE_QUEUE, runScheduleTaskOnce } from './runSchedule';
+import { ANONYMOUS_CONTEXT } from './anonymousContext';
+import { INNER_SCHEDULE_INTERVAL } from './scheduleNames';
+
+export { INNER_SCHEDULE_INTERVAL, ANONYMOUS_CONTEXT, SCHEDULE_QUEUE, runScheduleTaskOnce };

package/src/schedule/runSchedule.ts

@@ -0,0 +1,74 @@
+import * as koa from '@midwayjs/koa';
+import { IScheduleService } from '@/interface';
+import { ANONYMOUS_CONTEXT } from './anonymousContext';
+
+// 上次执行时间
+const lastExecuteTimeMap = {};
+
+const DEFAULT_SCHEDULE_INTERVAL = 2 * 60 * 1000;
+
+async function runScheduleTaskOnce(serviceName: string) {
+  return await ANONYMOUS_CONTEXT.runServiceAtAnonymousContext(async (ctx: koa.IMidwayKoaContext) => {
+    const serviceObject: IScheduleService = await ctx.requestContext.getAsync(serviceName);
+    if (serviceObject && serviceObject.runBySchedule) {
+      await serviceObject.runBySchedule();
+    } else {
+      ctx.logger.error(`runSchedule error , serviceName = ${serviceName}, 没有找到服务`);
+    }
+  });
+}
+
+class ScheduleQueue {
+  private scheduleIntervalMap: Map<string, number> = new Map();
+
+  public setScheduleTask(serviceName: string, intervalTime: number) {
+    this.scheduleIntervalMap.set(serviceName, intervalTime);
+  }
+
+  public removeScheduleTask(serviceName: string) {
+    this.scheduleIntervalMap.delete(serviceName);
+  }
+
+  private isAlreadyTimeToExecute(serviceName: string): boolean {
+    if (!lastExecuteTimeMap[serviceName]) {
+      return true; // 可以执行
+    }
+    const intervalTime1 = this.scheduleIntervalMap.get(serviceName);
+    const intervalTime = intervalTime1 || DEFAULT_SCHEDULE_INTERVAL; // 时间间隔
+    const lastExecuteTime = lastExecuteTimeMap[serviceName];
+    return Date.now() - lastExecuteTime >= intervalTime;
+  }
+
+  private async runScheduleTaskOnceCheckTime(serviceName: string) {
+    // 还没有到可以执行时间
+    if (!this.isAlreadyTimeToExecute(serviceName)) {
+      return;
+    }
+    lastExecuteTimeMap[serviceName] = Date.now();
+    await runScheduleTaskOnce(serviceName);
+  }
+
+  private async runScheduleServiceList() {
+    const serviceListKeys = this.scheduleIntervalMap.keys();
+    const serviceList = [...serviceListKeys];
+    for (let i = 0; i < serviceList.length; i++) {
+      const serviceName = serviceList[i];
+      try {
+        await this.runScheduleTaskOnceCheckTime(serviceName);
+      } catch (e) {
+        ANONYMOUS_CONTEXT.getApp().getCoreLogger().error(`runSchedule error , serviceName = ${serviceName}`, e);
+      }
+    }
+  }
+
+  public async startScheduleLoop() {
+    await this.runScheduleServiceList();
+    setInterval(() => {
+      this.runScheduleServiceList();
+    }, 1000); // 最少1秒执行一次
+  }
+}
+
+const SCHEDULE_QUEUE = new ScheduleQueue();
+
+export { SCHEDULE_QUEUE, runScheduleTaskOnce };