midway-fatcms 0.0.1-beta.3 → 0.0.1-beta.31

package/dist/service/asyncTask/AsyncTaskRunnerService.d.ts

@@ -0,0 +1,50 @@
+import { Context } from '@midwayjs/koa';
+import { BaseService } from '../../service/base/BaseService';
+import { IScheduleService } from '../../interface';
+import { ISysAsyncTaskHandler, SysAsyncTaskEntity } from '../../models/AsyncTaskModel';
+declare class AsyncTaskRunner {
+    isBusy: boolean;
+    taskHandlerMap: Map<string, ISysAsyncTaskHandler>;
+    executeTaskList(taskList: SysAsyncTaskEntity[]): Promise<void>;
+    private executeTask;
+    /**
+     * 更新任务状态或任务进度
+     * @param taskElement
+     * @param updatePartials
+     * @private
+     */
+    private updateTaskStatus;
+    /**
+     * 获取当前进程可以处理的任务类型。
+     */
+    getHandlerTaskTypeList(): string[];
+}
+/**
+ * 业务可以扩展
+ */
+export declare const ASYNC_TASK_RUNNER: AsyncTaskRunner;
+export declare class AsyncTaskRunnerService extends BaseService implements IScheduleService {
+    protected ctx: Context;
+    private curdProService;
+    /**
+     * 定时任务执行
+     */
+    runBySchedule(): Promise<void>;
+    /**
+     * 更新超时的任务状态
+     * @returns
+     */
+    private runByScheduleForTimeoutTasks;
+    /**
+     * 检查是否有新任务。如果有，就从数据库中获取一批任务，放到自己的任务队列里。
+     * @returns
+     */
+    private runByScheduleForPendingTasks;
+    fetchPendingTasks(): Promise<void>;
+    /**
+     * 是否存在新任务
+     * @private
+     */
+    private isExistNewTask;
+}
+export {};

package/dist/service/asyncTask/AsyncTaskRunnerService.js

@@ -0,0 +1,287 @@
+"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;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AsyncTaskRunnerService = exports.ASYNC_TASK_RUNNER = void 0;
+const moment = require("moment");
+const core_1 = require("@midwayjs/core");
+const BaseService_1 = require("../../service/base/BaseService");
+const CurdProService_1 = require("../../service/curd/CurdProService");
+const AsyncTaskModel_1 = require("../../models/AsyncTaskModel");
+const global_config_1 = require("../../libs/global-config/global-config");
+const SystemTables_1 = require("../../models/SystemTables");
+const keys_1 = require("../../libs/crud-pro/models/keys");
+const errorToString_1 = require("../../libs/utils/errorToString");
+const schedule_1 = require("../../schedule");
+const RedisKeys_1 = require("../../models/RedisKeys");
+const functions_1 = require("../../libs/utils/functions");
+const ExportExcelAsyncTaskHandler_1 = require("./handler/ExportExcelAsyncTaskHandler");
+const END_STATUS_LIST = [AsyncTaskModel_1.SysAsyncTaskStatus.PART_SUCCEEDED, AsyncTaskModel_1.SysAsyncTaskStatus.SUCCEEDED, AsyncTaskModel_1.SysAsyncTaskStatus.FAILED, AsyncTaskModel_1.SysAsyncTaskStatus.CANCELLED, AsyncTaskModel_1.SysAsyncTaskStatus.PAUSED];
+class AsyncTaskRunner {
+    constructor() {
+        this.isBusy = false;
+        this.taskHandlerMap = new Map();
+    }
+    async executeTaskList(taskList) {
+        if (!taskList || taskList.length === 0) {
+            return;
+        }
+        this.isBusy = true;
+        for (let i = 0; i < taskList.length; i++) {
+            const taskElement = taskList[i];
+            taskElement.task_status = AsyncTaskModel_1.SysAsyncTaskStatus.RUNNING;
+            try {
+                // 1. 更新当前这个任务的状态
+                await this.updateTaskStatus(taskElement, {
+                    task_status: taskElement.task_status,
+                    started_at: (0, functions_1.getCurrentFullMoment)(),
+                    updated_at: (0, functions_1.getCurrentFullMoment)(),
+                });
+                // 任务状态的设置在自定义的执行器里面。
+                await this.executeTask(taskElement);
+                // 当前不是终止状态，并且没有异常就设置成成功。
+                if (!END_STATUS_LIST.includes(taskElement.task_status)) {
+                    taskElement.task_status = AsyncTaskModel_1.SysAsyncTaskStatus.SUCCEEDED;
+                }
+            }
+            catch (error) {
+                taskElement.task_status = AsyncTaskModel_1.SysAsyncTaskStatus.FAILED;
+                taskElement.error_message = (0, errorToString_1.errorToString)(error);
+            }
+            try {
+                // 2. 更新当前这个任务的状态
+                await this.updateTaskStatus(taskElement, {
+                    task_status: taskElement.task_status,
+                    error_message: taskElement.error_message,
+                    updated_at: (0, functions_1.getCurrentFullMoment)(),
+                    completed_at: (0, functions_1.getCurrentFullMoment)(),
+                });
+            }
+            catch (error) {
+                schedule_1.ANONYMOUS_CONTEXT.getApp().getCoreLogger().error('[AsyncTaskRunner] executeTaskList error', error);
+            }
+        }
+        this.isBusy = false;
+    }
+    async executeTask(taskElement) {
+        const taskType = taskElement.task_type;
+        const taskHandler = this.taskHandlerMap.get(taskType);
+        if (!taskHandler) {
+            throw new Error('TaskHandler not found , taskType = ' + taskType);
+        }
+        const updateTaskStatus = (updatePartials) => {
+            return this.updateTaskStatus(taskElement, updatePartials);
+        };
+        await taskHandler.execute({ task: taskElement, updateTaskStatus });
+    }
+    /**
+     * 更新任务状态或任务进度
+     * @param taskElement
+     * @param updatePartials
+     * @private
+     */
+    async updateTaskStatus(taskElement, updatePartials) {
+        if (!updatePartials) {
+            throw new Error('updatePartials not found');
+        }
+        return await schedule_1.ANONYMOUS_CONTEXT.runServiceAtAnonymousContext(async (ctx) => {
+            const curdProService = await ctx.requestContext.getAsync('curdProService');
+            const { SystemDbName, SystemDbType } = global_config_1.GLOBAL_STATIC_CONFIG.getConfig();
+            // 一定要有一个更新时间字段
+            updatePartials.updated_at = (0, functions_1.getCurrentFullMoment)();
+            const res = await curdProService.executeCrudByCfg({
+                condition: {
+                    id: taskElement.id,
+                },
+                data: updatePartials,
+            }, {
+                sqlTable: SystemTables_1.SystemTables.sys_async_tasks,
+                sqlSimpleName: keys_1.KeysOfSimpleSQL.SIMPLE_UPDATE,
+                sqlDatabase: SystemDbName,
+                sqlDbType: SystemDbType,
+            });
+            return res.getResModel().affected;
+        });
+    }
+    /**
+     * 获取当前进程可以处理的任务类型。
+     */
+    getHandlerTaskTypeList() {
+        const keys = this.taskHandlerMap.keys();
+        return [...keys];
+    }
+}
+/**
+ * 业务可以扩展
+ */
+exports.ASYNC_TASK_RUNNER = new AsyncTaskRunner();
+//内置的异步任务处理器：EXCEL导出
+exports.ASYNC_TASK_RUNNER.taskHandlerMap.set('EXCEL_EXPORT', new ExportExcelAsyncTaskHandler_1.ExportExcelAsyncTaskHandler());
+/**
+ * Redis锁
+ */
+const ASYNC_TASK_LOCK = RedisKeys_1.RedisKeys.ASYNC_TASK_LOCK;
+const ASYNC_TASK_TIMEOUT_LOCK = RedisKeys_1.RedisKeys.ASYNC_TASK_TIMEOUT_LOCK;
+/**
+ * 上次检查异步任务的时间点
+ */
+const ASYNC_TASK_RUNTIME_OBJ = {
+    LAST_CHECK_ASYNC_TASK_UPDATE_TIME: 0,
+};
+let AsyncTaskRunnerService = class AsyncTaskRunnerService extends BaseService_1.BaseService {
+    /**
+     * 定时任务执行
+     */
+    async runBySchedule() {
+        await this.runByScheduleForPendingTasks();
+        await this.runByScheduleForTimeoutTasks();
+    }
+    /**
+     * 更新超时的任务状态
+     * @returns
+     */
+    async runByScheduleForTimeoutTasks() {
+        const { SystemDbName, SystemDbType } = global_config_1.GLOBAL_STATIC_CONFIG.getConfig();
+        // 每隔一小时执行一次，即可。
+        const lock = await this.redisService.set(ASYNC_TASK_TIMEOUT_LOCK, 1, 'EX', 60 * 60, 'NX');
+        if (lock !== 'OK') {
+            return Promise.resolve();
+        }
+        const timeoutTime = moment().subtract(24, 'hours').format("YYYY-MM-DD HH:mm:ss.SSS");
+        this.curdProService.executeCrudByCfg({
+            condition: {
+                task_status: AsyncTaskModel_1.SysAsyncTaskStatus.RUNNING,
+                updated_at: {
+                    $lt: timeoutTime,
+                }
+            },
+            data: {
+                task_status: AsyncTaskModel_1.SysAsyncTaskStatus.FAILED,
+                error_message: '任务超时',
+            }
+        }, {
+            sqlTable: SystemTables_1.SystemTables.sys_async_tasks,
+            sqlSimpleName: keys_1.KeysOfSimpleSQL.SIMPLE_UPDATE,
+            sqlDatabase: SystemDbName,
+            sqlDbType: SystemDbType,
+        });
+    }
+    /**
+     * 检查是否有新任务。如果有，就从数据库中获取一批任务，放到自己的任务队列里。
+     * @returns
+     */
+    async runByScheduleForPendingTasks() {
+        // 1. 当前很忙
+        if (exports.ASYNC_TASK_RUNNER.isBusy) {
+            return Promise.resolve();
+        }
+        // 2. 发现了新任务
+        const isExistNewTask = await this.isExistNewTask();
+        if (!isExistNewTask) {
+            return Promise.resolve();
+        }
+        // 这里的过期时间1分钟即可。fetchPendingTasks函数不可能超过一分钟。
+        // 因为这里只是从数据库中获取一批任务，放到自己的任务队列里。还没触发执行。
+        const lock = await this.redisService.set(ASYNC_TASK_LOCK, 1, 'EX', 60, 'NX');
+        if (lock !== 'OK') {
+            return Promise.resolve();
+        }
+        try {
+            await this.fetchPendingTasks();
+        }
+        catch (e) {
+            console.error('[AsyncTaskRunnerService] fetchPendingTasks error', (0, errorToString_1.errorToString)(e));
+        }
+        await this.redisService.del(ASYNC_TASK_LOCK);
+        return Promise.resolve();
+    }
+    async fetchPendingTasks() {
+        const { SystemDbName, SystemDbType } = global_config_1.GLOBAL_STATIC_CONFIG.getConfig();
+        // 只获取本进程能够处理的任务类型。
+        const taskTypeList = exports.ASYNC_TASK_RUNNER.getHandlerTaskTypeList();
+        if (taskTypeList.length === 0) {
+            return Promise.resolve();
+        }
+        // 查询等待处理的任务。
+        const queryRes = await this.curdProService.executeCrudByCfg({
+            condition: {
+                task_status: AsyncTaskModel_1.SysAsyncTaskStatus.PENDING,
+                task_type: {
+                    $in: taskTypeList,
+                },
+            },
+            orderBy: 'id+',
+            limit: 10,
+        }, {
+            sqlTable: SystemTables_1.SystemTables.sys_async_tasks,
+            sqlSimpleName: keys_1.KeysOfSimpleSQL.SIMPLE_QUERY_PAGE,
+            sqlDatabase: SystemDbName,
+            sqlDbType: SystemDbType,
+        });
+        const taskList = queryRes.getResRows();
+        if (taskList.length === 0) {
+            return Promise.resolve();
+        }
+        const taskIds = taskList.map(elem => elem.id).filter(Boolean);
+        // 将状态更新为处理中。防止其它进程重复处理。
+        await this.curdProService.executeCrudByCfg({
+            condition: {
+                id: {
+                    $in: taskIds,
+                },
+            },
+            data: {
+                task_status: AsyncTaskModel_1.SysAsyncTaskStatus.RUNNING,
+            },
+        }, {
+            sqlTable: SystemTables_1.SystemTables.sys_async_tasks,
+            sqlSimpleName: keys_1.KeysOfSimpleSQL.SIMPLE_UPDATE,
+            sqlDatabase: SystemDbName,
+            sqlDbType: SystemDbType,
+        });
+        // 开始执行。
+        exports.ASYNC_TASK_RUNNER.executeTaskList(taskList).then(() => {
+            console.log('[AsyncTaskRunnerService] ASYNC_TASK_RUNNER finished taskIds ==> ' + JSON.stringify(taskIds));
+        });
+    }
+    /**
+     * 是否存在新任务
+     * @private
+     */
+    async isExistNewTask() {
+        // 刚启动，没有检查过。
+        if (!ASYNC_TASK_RUNTIME_OBJ.LAST_CHECK_ASYNC_TASK_UPDATE_TIME) {
+            ASYNC_TASK_RUNTIME_OBJ.LAST_CHECK_ASYNC_TASK_UPDATE_TIME = Date.now();
+            return true;
+        }
+        //Redis没有任何内容
+        const updateTime = await this.redisService.get(RedisKeys_1.RedisKeys.ASYNC_TASK_UPDATE_TIME);
+        if (!updateTime) {
+            ASYNC_TASK_RUNTIME_OBJ.LAST_CHECK_ASYNC_TASK_UPDATE_TIME = Date.now();
+            return false;
+        }
+        const updateTimeNumber = Number.parseInt(updateTime, 10);
+        const isExistNewTask = updateTimeNumber > ASYNC_TASK_RUNTIME_OBJ.LAST_CHECK_ASYNC_TASK_UPDATE_TIME;
+        ASYNC_TASK_RUNTIME_OBJ.LAST_CHECK_ASYNC_TASK_UPDATE_TIME = updateTimeNumber;
+        return isExistNewTask;
+    }
+};
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", Object)
+], AsyncTaskRunnerService.prototype, "ctx", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", CurdProService_1.CurdProService)
+], AsyncTaskRunnerService.prototype, "curdProService", void 0);
+AsyncTaskRunnerService = __decorate([
+    (0, core_1.Provide)()
+], AsyncTaskRunnerService);
+exports.AsyncTaskRunnerService = AsyncTaskRunnerService;

package/dist/service/asyncTask/AsyncTaskService.d.ts

@@ -0,0 +1,7 @@
+import { Context } from '@midwayjs/koa';
+import { BaseService } from '../../service/base/BaseService';
+export declare class AsyncTaskService extends BaseService {
+    protected ctx: Context;
+    startTask(): Promise<void>;
+    cancelTask(id: number): Promise<void>;
+}

package/dist/service/asyncTask/AsyncTaskService.js

@@ -0,0 +1,34 @@
+"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;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AsyncTaskService = void 0;
+const core_1 = require("@midwayjs/core");
+const BaseService_1 = require("../../service/base/BaseService");
+const schedule_1 = require("../../schedule");
+const scheduleNames_1 = require("../../schedule/scheduleNames");
+const RedisKeys_1 = require("../../models/RedisKeys");
+let AsyncTaskService = class AsyncTaskService extends BaseService_1.BaseService {
+    async startTask() {
+        await this.redisService.set(RedisKeys_1.RedisKeys.ASYNC_TASK_UPDATE_TIME, `${Date.now()}`);
+        (0, schedule_1.runScheduleTaskOnce)(scheduleNames_1.INNER_SCHEDULE_NAMES.asyncTaskRunnerService).then(schedule => {
+            console.log(schedule);
+        });
+    }
+    async cancelTask(id) { }
+};
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", Object)
+], AsyncTaskService.prototype, "ctx", void 0);
+AsyncTaskService = __decorate([
+    (0, core_1.Provide)()
+], AsyncTaskService);
+exports.AsyncTaskService = AsyncTaskService;

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

@@ -0,0 +1,10 @@
+export interface IExcelHeaderInfo {
+    title: string;
+    dataIndex: string;
+}
+export interface IExcelAsyncTaskHandler {
+    getExcelHeader(): Promise<IExcelHeaderInfo[]>;
+    getPageSize(): Promise<number>;
+    getTotalCount(): Promise<number>;
+    getExcelDataList(pageNo: number, pageSize: number): Promise<any[]>;
+}

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

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

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

@@ -0,0 +1,7 @@
+import { ISysAsyncTaskHandler, SysAsyncTaskContext } from '../../../models/AsyncTaskModel';
+/**
+ * 单例
+ */
+export declare class ExportExcelAsyncTaskHandler implements ISysAsyncTaskHandler {
+    execute(asyncTaskContext: SysAsyncTaskContext): Promise<any>;
+}

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

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExportExcelAsyncTaskHandler = void 0;
+const AsyncTaskModel_1 = require("../../../models/AsyncTaskModel");
+const schedule_1 = require("../../../schedule");
+const AsyncTaskModel_2 = require("../../../models/AsyncTaskModel");
+const ExportExcelByStdCrudHandler_1 = require("./ExportExcelByStdCrudHandler");
+const ExportExcelByInnerHttpHandler_1 = require("./ExportExcelByInnerHttpHandler");
+const bizmodels_1 = require("../../../models/bizmodels");
+const _ = require("lodash");
+const fastcsv = require("fast-csv");
+const fs = require("node:fs");
+const fs2 = require("node:fs/promises");
+const path = require("node:path");
+const functions_1 = require("../../../libs/utils/functions");
+const userSession_1 = require("../../../models/userSession");
+const errorToString_1 = require("../../../libs/utils/errorToString");
+const indexRef = { current: 0 };
+async function createFileFolder(distFileFolder) {
+    try {
+        await fs2.access(distFileFolder);
+    }
+    catch (e) {
+        await fs2.mkdir(distFileFolder, { recursive: true });
+    }
+}
+function getExcelAsyncTaskHandler(asyncTaskContext, ctx) {
+    var _a;
+    const inputParams = (0, functions_1.parseJsonObject)((_a = asyncTaskContext === null || asyncTaskContext === void 0 ? void 0 : asyncTaskContext.task) === null || _a === void 0 ? void 0 : _a.input_params);
+    if (inputParams && inputParams.appType === bizmodels_1.StdCrudExportInputParamsAppType.STD_CRUD) {
+        return new ExportExcelByStdCrudHandler_1.ExportExcelByStdCrudHandler(asyncTaskContext, inputParams, ctx);
+    }
+    if (inputParams && inputParams.appType === bizmodels_1.StdCrudExportInputParamsAppType.INNER_HTTP) {
+        return new ExportExcelByInnerHttpHandler_1.ExportExcelByInnerHttpHandler(asyncTaskContext, inputParams, ctx);
+    }
+    throw new Error('[getExcelAsyncTaskHandler]appType不支持；appType = ' + inputParams.appType);
+}
+/**
+ * 多实例，每个任务一个实例。
+ * 导出EXCEL异步任务:
+ */
+class ExportExcelAsyncTask {
+    constructor(asyncTaskContext, ctx) {
+        this.asyncTaskContext = asyncTaskContext;
+        this.ctx = ctx;
+        this.excelAsyncTaskHandler = getExcelAsyncTaskHandler(asyncTaskContext, ctx);
+    }
+    getDistFileFolder() {
+        const app = schedule_1.ANONYMOUS_CONTEXT.getApp();
+        const config = app.getConfig();
+        const fatcmsExportExcelTmpFilePath = config.fatcmsExportExcelTmpFilePath;
+        if (!fatcmsExportExcelTmpFilePath) {
+            return path.join(__dirname, '../../tmp');
+        }
+        return fatcmsExportExcelTmpFilePath;
+    }
+    async executeWithContext() {
+        const taskElement = this.asyncTaskContext.task;
+        const taskId = _.get(this.asyncTaskContext, 'task.id');
+        this.ctx.logger.info('[ExportExcelAsyncTask] executeWithContext start, taskId=' + taskId);
+        const pageSize = await this.excelAsyncTaskHandler.getPageSize();
+        const fileName = await this.getExcelFileName(); // 文件名
+        const distFileFolder = this.getDistFileFolder();
+        await createFileFolder(distFileFolder);
+        const diskFilePath = path.resolve(distFileFolder, fileName); // 磁盘文件路径
+        const totalCount = await this.getTotalCount();
+        await this.asyncTaskContext.updateTaskStatus({
+            output_total_records: totalCount,
+            output_file_format: AsyncTaskModel_2.SysAsyncFileFormat.CSV,
+            output_file_path: diskFilePath,
+        });
+        this.headerColumns = await this.getExcelHeader();
+        const ws = fs.createWriteStream(diskFilePath, {
+            encoding: 'utf-8',
+        });
+        const csvStream = fastcsv.format({ writeHeaders: false, headers: false, delimiter: ',' });
+        csvStream.pipe(ws);
+        csvStream.write('\ufeff'); // 写入BOM头，解决中文乱码问题
+        csvStream.write(this.formatToCsvHeader(this.headerColumns));
+        let processed_records = 0;
+        try {
+            if (typeof totalCount === 'number' && totalCount > 0) {
+                const batchCount = Math.ceil(totalCount / pageSize);
+                for (let i = 0; i < batchCount; i++) {
+                    const num = await this.writeExcelBatchData(csvStream, i + 1, pageSize);
+                    processed_records = processed_records + num;
+                    const progress = Math.round((processed_records / totalCount) * 100);
+                    await this.asyncTaskContext.updateTaskStatus({
+                        processed_records,
+                        progress,
+                    });
+                }
+            }
+            taskElement.task_status = AsyncTaskModel_1.SysAsyncTaskStatus.SUCCEEDED;
+        }
+        catch (error) {
+            taskElement.task_status = AsyncTaskModel_1.SysAsyncTaskStatus.FAILED;
+            taskElement.error_message = (0, errorToString_1.errorToString)(error);
+            const progress = Math.round((processed_records / totalCount) * 100);
+            await this.asyncTaskContext.updateTaskStatus({
+                processed_records,
+                progress,
+                task_status: taskElement.task_status,
+                error_message: taskElement.error_message,
+            });
+        }
+        csvStream.end();
+        ws.on('finish', async () => {
+            this.ctx.logger.info('[ExportExcelAsyncTask] executeWithContext finish, taskId=' + taskId);
+            const fsStat = await fs2.stat(diskFilePath);
+            const output_file_size = fsStat.size;
+            this.asyncTaskContext.updateTaskStatus({
+                output_file_size,
+            });
+        });
+        this.ctx.logger.info('[ExportExcelAsyncTask] executeWithContext end, taskId=' + taskId);
+    }
+    formatToCsvHeader(headerColumns) {
+        return headerColumns.map(item => {
+            return item.title;
+        });
+    }
+    async writeExcelBatchData(csvStream, pageNo, pageSize) {
+        const dataList = await this.getExcelDataList(pageNo, pageSize);
+        for (let i = 0; i < dataList.length; i++) {
+            const csvRow = this.formatToCsvRow(dataList[i]);
+            csvStream.write(csvRow);
+        }
+        return dataList.length;
+    }
+    formatToCsvRow(rowObj) {
+        return this.headerColumns.map(obj => {
+            const dataIndex = obj.dataIndex;
+            const value = _.get(rowObj, dataIndex);
+            if (typeof value === 'undefined' || value === null) {
+                return '';
+            }
+            if (typeof value !== 'string') {
+                // boolean , number
+                return '' + value;
+            }
+            return value;
+        });
+    }
+    /**
+     * 获取文件名
+     * @returns 文件名
+     */
+    async getExcelFileName() {
+        indexRef.current = indexRef.current + 1;
+        if (indexRef.current > 100000) {
+            indexRef.current = 1;
+        }
+        const num = `${indexRef.current}`.padStart(10, '0');
+        return `export_${Date.now()}_${num}.csv`;
+    }
+    /**
+     * 获取Excel的表头
+     * @returns
+     */
+    async getExcelHeader() {
+        return this.excelAsyncTaskHandler.getExcelHeader();
+        // const inputParams = parseJsonObject(this.asyncTaskContext?.task?.input_params);
+        // if (inputParams && inputParams.appType === 'STD_CRUD') {
+        //     // const appCode = inputParams.appCode;
+        //     console.log(this)
+        // }
+        // return [
+        //     { title: 'ID', dataIndex: 'id' },
+        //     { title: '姓名', dataIndex: 'name' },
+        //     { title: '邮箱', dataIndex: 'email' }
+        // ]
+    }
+    /**
+     * 获取文件总数
+     */
+    async getTotalCount() {
+        const totalCount = await this.excelAsyncTaskHandler.getTotalCount();
+        // 最多导出10万条数据
+        return Math.min(totalCount, 10 * 10000);
+    }
+    /**
+     * 获取数据内容数组
+     * @param pageNo 当前页号，从1开始
+     * @param pageSize 每页获取的数量。
+     */
+    async getExcelDataList(pageNo, pageSize) {
+        return this.excelAsyncTaskHandler.getExcelDataList(pageNo, pageSize);
+        // const dataList = [];
+        // for (var i = 0; i < pageSize; i++) {
+        //     dataList.push({ id: i, name: 'nmae' + i, email: 'www.dss@dd.dom' })
+        // }
+        // return dataList;
+    }
+}
+/**
+ * 单例
+ */
+class ExportExcelAsyncTaskHandler {
+    async execute(asyncTaskContext) {
+        const taskId = _.get(asyncTaskContext, 'task.id');
+        console.log('[ExportExcelAsyncTaskHandler] execute, taskId=>' + taskId);
+        const res = await schedule_1.ANONYMOUS_CONTEXT.runServiceAtAnonymousContext(async (ctx) => {
+            var _a;
+            const createdUserSession = (0, functions_1.parseJsonObject)((_a = asyncTaskContext === null || asyncTaskContext === void 0 ? void 0 : asyncTaskContext.task) === null || _a === void 0 ? void 0 : _a.created_user_session);
+            ctx.userSession = new userSession_1.UserSessionInfo(createdUserSession, false);
+            const task = new ExportExcelAsyncTask(asyncTaskContext, ctx);
+            return task.executeWithContext();
+        });
+        if (res.error) {
+            throw res.error;
+        }
+        return res.result;
+    }
+}
+exports.ExportExcelAsyncTaskHandler = ExportExcelAsyncTaskHandler;

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

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