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

package/dist/service/asyncTask/handler/ExportExcelByInnerHttpHandler.js

@@ -0,0 +1,141 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExportExcelByInnerHttpHandler = void 0;
+const _ = require("lodash");
+const axios_1 = require("axios");
+const AsyncTaskModel_1 = require("../../../models/AsyncTaskModel");
+const schedule_1 = require("../../../schedule");
+const functions_1 = require("../../../libs/utils/functions");
+const global_config_1 = require("../../../libs/global-config/global-config");
+function pickData(a, key) {
+    if (a && typeof a.getResModel === 'function') {
+        const s = a.getResModel();
+        return s[key];
+    }
+    if (!a) {
+        throw 'ExportExcelByInnerHttpHandler返回数据不存在';
+    }
+    return _.get(a, `data.${key}`);
+}
+/**
+ * 多实例，每个任务一个实例。
+ * 导出Excel。请求任务级别的。
+ * 使用内部HTTP接口导出
+ */
+class ExportExcelByInnerHttpHandler {
+    constructor(asyncTaskContext, inputParams, ctx) {
+        this.ctx = ctx;
+        this.inputParams = inputParams;
+        this.asyncTaskContext = asyncTaskContext;
+        this.appCode = this.inputParams.appCode;
+    }
+    async getPageSize() {
+        const maxPageSizeOfExportExcel = global_config_1.GLOBAL_STATIC_CONFIG.getConfig().maxPageSizeOfExportExcel;
+        if (typeof this.inputParams.pageSize === 'number') {
+            return Math.min(this.inputParams.pageSize, maxPageSizeOfExportExcel);
+        }
+        return 1000;
+    }
+    /**
+     * 获取Excel表的表头
+     * @returns 表头信息
+     */
+    async getExcelHeader() {
+        const columns = this.inputParams.columns;
+        if (!columns) {
+            throw new Error('导出Excel的表头信息不存在');
+        }
+        if (!Array.isArray(columns) || columns.length === 0) {
+            throw new Error('导出Excel的表头信息不是数组');
+        }
+        return columns
+            .map(cc => {
+            let { title, dataIndex } = cc;
+            if (!dataIndex) {
+                return null;
+            }
+            if (!title) {
+                title = '字段_' + dataIndex;
+            }
+            return { title, dataIndex };
+        })
+            .filter(Boolean);
+    }
+    async sendRequest(params) {
+        const port = schedule_1.ANONYMOUS_CONTEXT.getApp().getConfig().koa.port;
+        let queryApi = this.inputParams.innerHttpUrl; // 以斜杠开头
+        if (!queryApi.startsWith('/')) {
+            queryApi = '/' + queryApi;
+        }
+        const asyncTaskId = this.asyncTaskContext.task.id;
+        const createdUserSession = (0, functions_1.parseJsonObject)(this.asyncTaskContext.task.created_user_session);
+        const workbenchCode = createdUserSession.workbenchCode;
+        params[AsyncTaskModel_1.HEADER_KEY_RUN_BY_ASYNC_TASK_ID] = asyncTaskId;
+        params[AsyncTaskModel_1.HEADER_KEY_RUN_BY_ASYNC_TASK_WORKBENCH_CODE] = workbenchCode;
+        const headers = this.inputParams.headers || {};
+        headers[AsyncTaskModel_1.HEADER_KEY_RUN_BY_ASYNC_TASK_ID] = asyncTaskId;
+        headers[AsyncTaskModel_1.HEADER_KEY_RUN_BY_ASYNC_TASK_WORKBENCH_CODE] = workbenchCode;
+        const response = await (0, axios_1.default)({
+            method: 'POST',
+            url: `http://127.0.0.1:${port}${queryApi}`,
+            data: params,
+            headers: headers,
+            responseType: 'json',
+            validateStatus: () => true,
+        });
+        if (response.status !== 200) {
+            throw new Error('导出Excel请求错误失败，状态码：' + response.status);
+        }
+        return response.data;
+    }
+    /**
+     * 获取要导出的数量总数
+     * @returns 数据总量
+     */
+    async getTotalCount() {
+        var _a, _b;
+        const selectedRowKeys = ((_a = this.inputParams) === null || _a === void 0 ? void 0 : _a.selectedRowKeys) || [];
+        if (Array.isArray(selectedRowKeys) && selectedRowKeys.length > 0) {
+            return selectedRowKeys.length;
+        }
+        const lastQueryParams = (_b = this.inputParams) === null || _b === void 0 ? void 0 : _b.lastQueryParams;
+        const res = await this.sendRequest({
+            condition: (lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.condition) || {},
+            orderBy: lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.orderBy,
+            pageSize: 1,
+            pageNo: 1,
+        });
+        return pickData(res, 'total_count');
+    }
+    /**
+     * 获取要导出Excel的数据的内容
+     * @param pageNo 页号，从1开始
+     * @param pageSize 每页数量
+     * @returns
+     */
+    async getExcelDataList(pageNo, pageSize) {
+        var _a, _b;
+        const selectedRowKeys = ((_a = this.inputParams) === null || _a === void 0 ? void 0 : _a.selectedRowKeys) || [];
+        const lastQueryParams = (_b = this.inputParams) === null || _b === void 0 ? void 0 : _b.lastQueryParams;
+        const primaryKey = this.inputParams.primaryKey || 'id';
+        const params = {
+            condition: (lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.condition) || {},
+            pageSize: pageSize,
+            pageNo: pageNo,
+            orderBy: lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.orderBy,
+        };
+        if (Array.isArray(selectedRowKeys) && selectedRowKeys.length > 0) {
+            params.condition[primaryKey] = {
+                $in: selectedRowKeys,
+            };
+        }
+        const res = await this.sendRequest({
+            condition: (lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.condition) || {},
+            orderBy: lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.orderBy,
+            pageSize: pageSize,
+            pageNo: pageNo,
+        });
+        return pickData(res, 'rows');
+    }
+}
+exports.ExportExcelByInnerHttpHandler = ExportExcelByInnerHttpHandler;

package/dist/service/asyncTask/handler/ExportExcelByStdCrudHandler.d.ts

@@ -0,0 +1,46 @@
+import { IMidwayKoaContext } from '@midwayjs/koa';
+import { IExcelAsyncTaskHandler, IExcelHeaderInfo } from './ExcelInfoModel';
+import { SysAsyncTaskContext } from '../../../models/AsyncTaskModel';
+import { ICrudStdAppInfo, IStdCrudExportInputParams } from '../../../models/bizmodels';
+/**
+ * 多实例，每个任务一个实例。
+ * 导出Excel。请求任务级别的。
+ * 使用标准CtdCrudApp导出
+ */
+declare class ExportExcelByStdCrudHandler implements IExcelAsyncTaskHandler {
+    protected asyncTaskContext: SysAsyncTaskContext;
+    protected ctx: IMidwayKoaContext;
+    protected inputParams: IStdCrudExportInputParams;
+    protected appCode: string;
+    protected crudStdAppInfo: ICrudStdAppInfo;
+    constructor(asyncTaskContext: SysAsyncTaskContext, inputParams: IStdCrudExportInputParams, ctx: IMidwayKoaContext);
+    getPageSize(): Promise<number>;
+    /**
+     * 获取应用信息
+     * @returns CrudStdService服务实例
+     */
+    private getCrudStdService;
+    /**
+     * 获取应用的配置信息
+     * @returns 应用的配置信息
+     */
+    private getCrudStdAppInfo;
+    /**
+     * 获取Excel表的表头
+     * @returns 表头信息
+     */
+    getExcelHeader(): Promise<IExcelHeaderInfo[]>;
+    /**
+     * 获取要导出的数量总数
+     * @returns 数据总量
+     */
+    getTotalCount(): Promise<number>;
+    /**
+     * 获取要导出Excel的数据的内容
+     * @param pageNo 页号，从1开始
+     * @param pageSize 每页数量
+     * @returns
+     */
+    getExcelDataList(pageNo: number, pageSize: number): Promise<any[]>;
+}
+export { ExportExcelByStdCrudHandler };

package/dist/service/asyncTask/handler/ExportExcelByStdCrudHandler.js

@@ -0,0 +1,135 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExportExcelByStdCrudHandler = void 0;
+const _ = require("lodash");
+const CrudStdService_1 = require("../../../service/crudstd/CrudStdService");
+const keys_1 = require("../../../libs/crud-pro/models/keys");
+const global_config_1 = require("../../../libs/global-config/global-config");
+function pickData(a, key) {
+    if (a && typeof a.getResModel === 'function') {
+        const s = a.getResModel();
+        return s[key];
+    }
+    if (!a) {
+        throw 'executeStdQuery返回数据不存在';
+    }
+    return _.get(a, `data.${key}`);
+}
+/**
+ * 多实例，每个任务一个实例。
+ * 导出Excel。请求任务级别的。
+ * 使用标准CtdCrudApp导出
+ */
+class ExportExcelByStdCrudHandler {
+    constructor(asyncTaskContext, inputParams, ctx) {
+        this.ctx = ctx;
+        this.inputParams = inputParams;
+        this.asyncTaskContext = asyncTaskContext;
+        this.appCode = this.inputParams.appCode;
+    }
+    async getPageSize() {
+        const maxPageSizeOfExportExcel = global_config_1.GLOBAL_STATIC_CONFIG.getConfig().maxPageSizeOfExportExcel;
+        if (typeof this.inputParams.pageSize === 'number') {
+            return Math.min(this.inputParams.pageSize, maxPageSizeOfExportExcel);
+        }
+        return 1000;
+    }
+    /**
+     * 获取应用信息
+     * @returns CrudStdService服务实例
+     */
+    async getCrudStdService() {
+        return await this.ctx.requestContext.getAsync(CrudStdService_1.CrudStdService);
+    }
+    /**
+     * 获取应用的配置信息
+     * @returns 应用的配置信息
+     */
+    async getCrudStdAppInfo() {
+        if (!this.crudStdAppInfo) {
+            const appCode = this.inputParams.appCode;
+            if (!appCode) {
+                throw new Error('inputParams appCode is required');
+            }
+            const service = await this.getCrudStdService();
+            this.crudStdAppInfo = await service.getParsedCrudStdAppInfo(appCode);
+        }
+        return this.crudStdAppInfo;
+    }
+    /**
+     * 获取Excel表的表头
+     * @returns 表头信息
+     */
+    async getExcelHeader() {
+        const crudStdAppInfo = await this.getCrudStdAppInfo();
+        const stdCrudCfgObj = crudStdAppInfo.stdCrudCfgObj;
+        const tableColumns = stdCrudCfgObj.tableColumns || [];
+        return tableColumns.map(cc => {
+            return cc;
+        });
+    }
+    /**
+     * 获取要导出的数量总数
+     * @returns 数据总量
+     */
+    async getTotalCount() {
+        var _a, _b;
+        const selectedRowKeys = ((_a = this.inputParams) === null || _a === void 0 ? void 0 : _a.selectedRowKeys) || [];
+        if (Array.isArray(selectedRowKeys) && selectedRowKeys.length > 0) {
+            return selectedRowKeys.length;
+        }
+        const lastQueryParams = (_b = this.inputParams) === null || _b === void 0 ? void 0 : _b.lastQueryParams;
+        const filterValue = _.get(lastQueryParams, 'filterValue');
+        const crudStdService = await this.getCrudStdService();
+        const appCode = this.appCode;
+        const params = {
+            condition: (lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.condition) || {},
+            filterValue,
+            pageSize: 1,
+            pageNo: 1,
+        };
+        if (!this.ctx.innerHeaders) {
+            this.ctx.innerHeaders = {};
+        }
+        Object.assign(this.ctx.innerHeaders, this.inputParams.headers || {});
+        const stdAction = {
+            appCode: appCode,
+            settingKey: CrudStdService_1.SPECIAL_SETTING_KEY.QUERY_LIST,
+        };
+        const a = await crudStdService.executeStdQuery(stdAction, params, keys_1.KeysOfSimpleSQL.SIMPLE_QUERY_PAGE);
+        return pickData(a, 'total_count');
+    }
+    /**
+     * 获取要导出Excel的数据的内容
+     * @param pageNo 页号，从1开始
+     * @param pageSize 每页数量
+     * @returns
+     */
+    async getExcelDataList(pageNo, pageSize) {
+        const selectedRowKeys = this.inputParams.selectedRowKeys || [];
+        const primaryKey = this.inputParams.primaryKey || 'id';
+        const lastQueryParams = this.inputParams.lastQueryParams;
+        const filterValue = _.get(lastQueryParams, 'filterValue');
+        const crudStdService = await this.getCrudStdService();
+        const appCode = this.appCode;
+        const params = {
+            condition: (lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.condition) || {},
+            filterValue: filterValue,
+            pageSize: pageSize,
+            pageNo: pageNo,
+            orderBy: lastQueryParams === null || lastQueryParams === void 0 ? void 0 : lastQueryParams.orderBy,
+        };
+        if (Array.isArray(selectedRowKeys) && selectedRowKeys.length > 0) {
+            params.condition[primaryKey] = {
+                $in: selectedRowKeys,
+            };
+        }
+        const stdAction = {
+            appCode: appCode,
+            settingKey: CrudStdService_1.SPECIAL_SETTING_KEY.QUERY_LIST,
+        };
+        const a = await crudStdService.executeStdQuery(stdAction, params, keys_1.KeysOfSimpleSQL.SIMPLE_QUERY_PAGE);
+        return pickData(a, 'rows');
+    }
+}
+exports.ExportExcelByStdCrudHandler = ExportExcelByStdCrudHandler;

package/dist/service/base/ApiBaseService.d.ts

@@ -1,15 +1,15 @@
-import { BaseService } from './BaseService';
-import { IApiBaseEntity } from '../../models/SystemEntities';
-export declare const API_BASE_TYPE: {
-    PROXY_API: string;
-    ANY_API: string;
-    CRUD_METHOD: string;
-};
-export declare class ApiBaseService extends BaseService {
-    /**
-     * 前置校验
-     * @param apiBaseType
-     * @param entity
-     */
-    beforeCheckApiAccessibility(apiBaseType: string, entity: IApiBaseEntity): Promise<void>;
-}
+import { BaseService } from './BaseService';
+import { IApiBaseEntity } from '../../models/SystemEntities';
+export declare const API_BASE_TYPE: {
+    PROXY_API: string;
+    ANY_API: string;
+    CRUD_METHOD: string;
+};
+export declare class ApiBaseService extends BaseService {
+    /**
+     * 前置校验
+     * @param apiBaseType
+     * @param entity
+     */
+    beforeCheckApiAccessibility(apiBaseType: string, entity: IApiBaseEntity): Promise<void>;
+}

package/dist/service/base/ApiBaseService.js

@@ -1,50 +1,50 @@
-"use strict";
-var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
-    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
-    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
-    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
-    return c > 3 && r && Object.defineProperty(target, key, r), r;
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ApiBaseService = exports.API_BASE_TYPE = void 0;
-const core_1 = require("@midwayjs/core");
-const BaseService_1 = require("./BaseService");
-const devops_1 = require("../../models/devops");
-const exceptions_1 = require("../../libs/crud-pro/exceptions");
-const ApiRateLimiter_1 = require("./ApiRateLimiter");
-exports.API_BASE_TYPE = {
-    PROXY_API: 'PROXY_API',
-    ANY_API: 'ANY_API',
-    CRUD_METHOD: 'CRUD_METHOD',
-};
-let ApiBaseService = class ApiBaseService extends BaseService_1.BaseService {
-    /**
-     * 前置校验
-     * @param apiBaseType
-     * @param entity
-     */
-    async beforeCheckApiAccessibility(apiBaseType, entity) {
-        // 权限校验
-        const isPermSuccess = this.ctx.userSession.isAuthPass(entity.auth_type, entity.auth_config);
-        if (!isPermSuccess) {
-            throw new devops_1.BizException('权限校验失败', exceptions_1.Exceptions.NO_AUTH);
-        }
-        //服务降级
-        if (entity.fallback_enable === 1) {
-            throw new devops_1.BizException(entity.fallback_message || '此服务暂时不支持访问', 'FALLBACK_ENABLE');
-        }
-        // 限流令牌桶
-        if (entity.limit_enabled === 1) {
-            const accountId = this.ctx.userSession.getSessionInfo().accountId;
-            const apiId = apiBaseType + '_' + entity.id;
-            const isAcquire = (0, ApiRateLimiter_1.tryAcquireToken)(apiId, accountId, entity.limit_max, entity.limit_rate);
-            if (!isAcquire) {
-                throw new devops_1.BizException('频繁访问此接口，被限流', 'LIMIT_ENABLED');
-            }
-        }
-    }
-};
-ApiBaseService = __decorate([
-    (0, core_1.Provide)()
-], ApiBaseService);
-exports.ApiBaseService = ApiBaseService;
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ApiBaseService = exports.API_BASE_TYPE = void 0;
+const core_1 = require("@midwayjs/core");
+const BaseService_1 = require("./BaseService");
+const devops_1 = require("../../models/devops");
+const exceptions_1 = require("../../libs/crud-pro/exceptions");
+const ApiRateLimiter_1 = require("./ApiRateLimiter");
+exports.API_BASE_TYPE = {
+    PROXY_API: 'PROXY_API',
+    ANY_API: 'ANY_API',
+    CRUD_METHOD: 'CRUD_METHOD',
+};
+let ApiBaseService = class ApiBaseService extends BaseService_1.BaseService {
+    /**
+     * 前置校验
+     * @param apiBaseType
+     * @param entity
+     */
+    async beforeCheckApiAccessibility(apiBaseType, entity) {
+        // 权限校验
+        const isPermSuccess = this.ctx.userSession.isAuthPass(entity.auth_type, entity.auth_config);
+        if (!isPermSuccess) {
+            throw new devops_1.BizException('权限校验失败', exceptions_1.Exceptions.NO_AUTH);
+        }
+        //服务降级
+        if (entity.fallback_enable === 1) {
+            throw new devops_1.BizException(entity.fallback_message || '此服务暂时不支持访问', 'FALLBACK_ENABLE');
+        }
+        // 限流令牌桶
+        if (entity.limit_enabled === 1) {
+            const accountId = this.ctx.userSession.getSessionInfo().accountId;
+            const apiId = apiBaseType + '_' + entity.id;
+            const isAcquire = (0, ApiRateLimiter_1.tryAcquireToken)(apiId, accountId, entity.limit_max, entity.limit_rate);
+            if (!isAcquire) {
+                throw new devops_1.BizException('频繁访问此接口，被限流', 'LIMIT_ENABLED');
+            }
+        }
+    }
+};
+ApiBaseService = __decorate([
+    (0, core_1.Provide)()
+], ApiBaseService);
+exports.ApiBaseService = ApiBaseService;

package/dist/service/base/ApiRateLimiter.d.ts

@@ -1,10 +1,10 @@
-/**
- * 对接口ID ， 每个用户ID 进行限流。
- * @param apiId 接口ID
- * @param userId 用户ID
- * @param limit_max 限流令牌桶最大值，最大容量
- * @param limit_rate 每秒产生的令牌数量
- * @return {Boolean} 返回token是否可以用。不能被处理就返回false
- */
-declare function tryAcquireToken(apiId: string, userId: string, limit_max: number, limit_rate: number): boolean;
-export { tryAcquireToken };
+/**
+ * 对接口ID ， 每个用户ID 进行限流。
+ * @param apiId 接口ID
+ * @param userId 用户ID
+ * @param limit_max 限流令牌桶最大值，最大容量
+ * @param limit_rate 每秒产生的令牌数量
+ * @return {Boolean} 返回token是否可以用。不能被处理就返回false
+ */
+declare function tryAcquireToken(apiId: string, userId: string, limit_max: number, limit_rate: number): boolean;
+export { tryAcquireToken };

package/dist/service/base/ApiRateLimiter.js

@@ -1,51 +1,51 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.tryAcquireToken = void 0;
-const lru_cache_1 = require("lru-cache");
-// 定义 TokenBucket 类
-class TokenBucket {
-    constructor(capacity, fillRate) {
-        this.capacity = capacity;
-        this.fillRate = fillRate;
-        this.tokens = capacity;
-        this.lastRefillTime = Date.now();
-    }
-    refill() {
-        const now = Date.now();
-        const elapsedTime = (now - this.lastRefillTime) / 1000;
-        const newTokens = elapsedTime * this.fillRate;
-        this.tokens = Math.min(this.capacity, this.tokens + newTokens);
-        this.lastRefillTime = now;
-    }
-    tryAcquire(tokensNeeded = 1) {
-        this.refill();
-        if (this.tokens >= tokensNeeded) {
-            this.tokens -= tokensNeeded;
-            return true;
-        }
-        return false;
-    }
-}
-// 定义一个全局的 Map 来存储不同接口和用户组合的令牌桶
-const apiUserBuckets = new lru_cache_1.LRUCache({
-    max: 500,
-    ttl: 1000 * 60 * 5,
-    ttlAutopurge: true,
-});
-/**
- * 对接口ID ， 每个用户ID 进行限流。
- * @param apiId 接口ID
- * @param userId 用户ID
- * @param limit_max 限流令牌桶最大值，最大容量
- * @param limit_rate 每秒产生的令牌数量
- * @return {Boolean} 返回token是否可以用。不能被处理就返回false
- */
-function tryAcquireToken(apiId, userId, limit_max, limit_rate) {
-    const key = `${apiId}-${userId}`;
-    if (!apiUserBuckets.has(key)) {
-        apiUserBuckets.set(key, new TokenBucket(limit_max, limit_rate));
-    }
-    const bucket = apiUserBuckets.get(key);
-    return bucket.tryAcquire();
-}
-exports.tryAcquireToken = tryAcquireToken;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tryAcquireToken = void 0;
+const lru_cache_1 = require("lru-cache");
+// 定义 TokenBucket 类
+class TokenBucket {
+    constructor(capacity, fillRate) {
+        this.capacity = capacity;
+        this.fillRate = fillRate;
+        this.tokens = capacity;
+        this.lastRefillTime = Date.now();
+    }
+    refill() {
+        const now = Date.now();
+        const elapsedTime = (now - this.lastRefillTime) / 1000;
+        const newTokens = elapsedTime * this.fillRate;
+        this.tokens = Math.min(this.capacity, this.tokens + newTokens);
+        this.lastRefillTime = now;
+    }
+    tryAcquire(tokensNeeded = 1) {
+        this.refill();
+        if (this.tokens >= tokensNeeded) {
+            this.tokens -= tokensNeeded;
+            return true;
+        }
+        return false;
+    }
+}
+// 定义一个全局的 Map 来存储不同接口和用户组合的令牌桶
+const apiUserBuckets = new lru_cache_1.LRUCache({
+    max: 500,
+    ttl: 1000 * 60 * 5,
+    ttlAutopurge: true,
+});
+/**
+ * 对接口ID ， 每个用户ID 进行限流。
+ * @param apiId 接口ID
+ * @param userId 用户ID
+ * @param limit_max 限流令牌桶最大值，最大容量
+ * @param limit_rate 每秒产生的令牌数量
+ * @return {Boolean} 返回token是否可以用。不能被处理就返回false
+ */
+function tryAcquireToken(apiId, userId, limit_max, limit_rate) {
+    const key = `${apiId}-${userId}`;
+    if (!apiUserBuckets.has(key)) {
+        apiUserBuckets.set(key, new TokenBucket(limit_max, limit_rate));
+    }
+    const bucket = apiUserBuckets.get(key);
+    return bucket.tryAcquire();
+}
+exports.tryAcquireToken = tryAcquireToken;

package/dist/service/base/BaseService.d.ts

@@ -1,29 +1,41 @@
-import { Context } from '@midwayjs/koa';
-import { RedisService } from '@midwayjs/redis';
-import { OSSService, OSSServiceFactory } from '@midwayjs/oss';
-import * as koa from '@midwayjs/koa';
-import { ILoggerContext } from '../../models/contextLogger';
-export declare class BaseService {
-    protected ctx: Context;
-    protected app: koa.Application;
-    protected redisService: RedisService;
-    protected ossServiceFactory: OSSServiceFactory;
-    /**
-     * 用户自己上传的文件，使用私有的bucket。
-     * @protected
-     */
-    protected getPrivateOSSService(): OSSService;
-    /**
-     * 公共读的bucket，如js文件等使用它
-     * @protected
-     */
-    protected getPublicOSSService(): OSSService;
-    protected getContextLogger(): ILoggerContext;
-    protected getPrefixContextLogger(prefix: string): ILoggerContext;
-    protected logInfo(msg: string, ...args: any): void;
-    protected logError(msg: string, ...args: any): void;
-    protected logWarn(msg: string, ...args: any): void;
-    protected logDebug(msg: string, ...args: any): void;
-    protected isLocalEnv(): boolean;
-    protected isEnableDebug(): boolean;
-}
+import { Context } from '@midwayjs/koa';
+import { RedisService } from '@midwayjs/redis';
+import { OSSService, OSSServiceFactory } from '@midwayjs/oss';
+import * as koa from '@midwayjs/koa';
+import { ILoggerContext } from '../../models/contextLogger';
+export declare class BaseService {
+    protected ctx: Context;
+    protected app: koa.Application;
+    protected redisService: RedisService;
+    protected ossServiceFactory: OSSServiceFactory;
+    /**
+     * 请求级别的内存缓存
+     * @param key
+     * @returns
+     */
+    protected getFromCtxCache(key: string): any;
+    /**
+     * 请求级别的内存缓存
+     * @param key
+     * @param obj
+     */
+    protected setToCtxCache(key: string, obj: any): void;
+    /**
+     * 用户自己上传的文件，使用私有的bucket。
+     * @protected
+     */
+    protected getPrivateOSSService(): OSSService;
+    /**
+     * 公共读的bucket，如js文件等使用它
+     * @protected
+     */
+    protected getPublicOSSService(): OSSService;
+    protected getContextLogger(): ILoggerContext;
+    protected getPrefixContextLogger(prefix: string): ILoggerContext;
+    protected logInfo(msg: string, ...args: any): void;
+    protected logError(msg: string, ...args: any): void;
+    protected logWarn(msg: string, ...args: any): void;
+    protected logDebug(msg: string, ...args: any): void;
+    protected isLocalEnv(): boolean;
+    protected isEnableDebug(): boolean;
+}