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

package/src/service/UserSessionService.ts

@@ -1,10 +1,9 @@
-import {Inject, Provide} from '@midwayjs/core';
-import {Context} from '@midwayjs/koa';
-import {parseJsonObject} from '@/libs/utils/functions';
-import {ISessionInfo, SESSION_ID_KEY, sessionCookieCfg} from '@/models/userSession';
-import {RedisCacheService} from "./base/RedisCacheService";
-import {RedisKeys} from "@/models/RedisKeys";
-
+import { Inject, Provide } from '@midwayjs/core';
+import { Context } from '@midwayjs/koa';
+import { parseJsonObject } from '@/libs/utils/functions';
+import { ISessionInfo, SESSION_ID_KEY, sessionCookieCfg } from '@/models/userSession';
+import { RedisCacheService } from './base/RedisCacheService';
+import { RedisKeys } from '@/models/RedisKeys';
 
 function pickUserAvatar(avatar: any): string {
   if (!avatar) {
@@ -33,12 +32,11 @@ function pickUserAvatar(avatar: any): string {
   return null;
 }
 
-
 const SESSION_KEEP_TIME_SECOND: number = 3600 * 24 * 30;
 
-const toCacheKey = (sessionId: string): string =>{
+const toCacheKey = (sessionId: string): string => {
   return `${RedisKeys.USER_SESSION_PREFIX}${sessionId}`;
-}
+};
 
 @Provide()
 export class UserSessionService {
@@ -46,13 +44,12 @@ export class UserSessionService {
   private ctx: Context;
 
   @Inject()
-  private redisCacheService: RedisCacheService
-
+  private redisCacheService: RedisCacheService;
 
   public async saveUserSession(sessionInfo: ISessionInfo) {
     sessionInfo.avatar = pickUserAvatar(sessionInfo.avatar);
     const sessionId = sessionInfo.sessionId;
-    await this.redisCacheService.setJsonObject(toCacheKey(sessionId), sessionInfo, SESSION_KEEP_TIME_SECOND)
+    await this.redisCacheService.setJsonObject(toCacheKey(sessionId), sessionInfo, SESSION_KEEP_TIME_SECOND);
   }
 
   /**
@@ -60,14 +57,14 @@ export class UserSessionService {
    * @param sessionId
    */
   public async removeUserSession(sessionId: string) {
-    return await this.redisCacheService.removeItem(toCacheKey(sessionId))
+    return await this.redisCacheService.removeItem(toCacheKey(sessionId));
   }
 
   private async getCurrentUserSessionForMiddleware1(sessionId: string): Promise<ISessionInfo> {
     if (!sessionId) {
       return null;
     }
-    return await this.redisCacheService.getJsonObject(toCacheKey(sessionId))
+    return await this.redisCacheService.getJsonObject(toCacheKey(sessionId));
   }
 
   /**
@@ -78,5 +75,4 @@ export class UserSessionService {
     const sessionId = cookies.get(SESSION_ID_KEY, sessionCookieCfg);
     return this.getCurrentUserSessionForMiddleware1(sessionId);
   }
-
 }

package/src/service/WorkbenchService.ts

@@ -1,16 +1,16 @@
 import { Inject, Provide } from '@midwayjs/core';
 import { Context } from '@midwayjs/koa';
-import { KeysOfSimpleSQL } from '../libs/crud-pro/models/keys';
-import { SystemTables} from '../models/SystemTables';
+import { KeysOfSimpleSQL } from '@/libs/crud-pro/models/keys';
+import { SystemTables } from '@/models/SystemTables';
 import { BaseService } from './base/BaseService';
 // import { LRUCache } from 'lru-cache';
 import { CurdProService } from './curd/CurdProService';
-import { CommonException } from '../libs/crud-pro/exceptions';
-import { parseJsonObject } from '../libs/utils/functions';
-import { IWorkbenchEntity } from '../models/SystemEntities';
-import { IScheduleService } from '../interface';
-import { getDebugWorkbenchCode } from '../libs/utils/fatcms-request';
-import { MixinUtils } from '../libs/crud-pro/utils/MixinUtils';
+import { CommonException } from '@/libs/crud-pro/exceptions';
+import { parseJsonObject } from '@/libs/utils/functions';
+import { IWorkbenchEntity } from '@/models/SystemEntities';
+import { IScheduleService } from '@/interface';
+import { getDebugWorkbenchCode } from '@/libs/utils/fatcms-request';
+import { MixinUtils } from '@/libs/crud-pro/utils/MixinUtils';
 import { GLOBAL_STATIC_CONFIG } from '@/libs/global-config/global-config';
 //
 // const lruCache = new LRUCache<string, any>({
@@ -24,7 +24,6 @@ const lruCache = new Map<string, any>();
 const CACHE_KEY_WORKBENCH_LIST = 'CACHE_KEY_WORKBENCH_LIST';
 const CACHE_KEY_DOMAIN_MAP = 'CACHE_KEY_DOMAIN_MAP';
 
-
 @Provide()
 export class WorkbenchService extends BaseService implements IScheduleService {
   @Inject()
@@ -43,10 +42,8 @@ export class WorkbenchService extends BaseService implements IScheduleService {
   }
 
   public async getAllWorkbenchInfoList(isForceRefresh?: boolean): Promise<IWorkbenchEntity[]> {
-
     const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
 
-
     let listData = lruCache.get(CACHE_KEY_WORKBENCH_LIST);
     if (!listData || isForceRefresh === true) {
       const reqJson = { condition: {}, pageSize: 1000, limit: 1000 };
@@ -70,7 +67,7 @@ export class WorkbenchService extends BaseService implements IScheduleService {
       const domainMap1 = MixinUtils.toMap(listData, (e: any) => e.workbench_domain); // 主域名。其他两个域名。
       const domainMap2 = MixinUtils.toMap(listData, (e: any) => e.workbench_domain2);
       const domainMap3 = MixinUtils.toMap(listData, (e: any) => e.workbench_domain3);
-      domainMap = {...domainMap1, ...domainMap2, ...domainMap3};
+      domainMap = { ...domainMap1, ...domainMap2, ...domainMap3 };
       lruCache.set(CACHE_KEY_DOMAIN_MAP, domainMap);
     }
     return domainMap;
@@ -110,18 +107,21 @@ export class WorkbenchService extends BaseService implements IScheduleService {
     //正式环境： 根据域名获取站点配置
     const host = this.getCurrentHost();
     const domainMap = await this.getAllWorkbenchInfoMap();
-    return domainMap[host];
+    const workbenchInfo = domainMap[host];
+    if (workbenchInfo) {
+      return workbenchInfo;
+    }
+    return domainMap['*']; // 最后兜底
   }
 
   private getCurrentHost(): string {
     const host = this.ctx.request.host;
     if (!host) {
-      throw new CommonException('GET_CURRENT_HOST_ERROR','获取当前域名失败');
+      throw new CommonException('GET_CURRENT_HOST_ERROR', '获取当前域名失败');
     }
     return host;
   }
 
-
   public async isSupportCurrentWorkbench(workbench_code_array: string | string[]): Promise<boolean> {
     // 本地调试模式
     if (this.isEnableDebug() && this.isLocalEnv()) {

package/src/service/anyapi/AnyApiService.ts

@@ -1,7 +1,7 @@
 import { Inject, Provide } from '@midwayjs/core';
 import { Context } from '@midwayjs/koa';
 import { KeysOfSimpleSQL } from '@/libs/crud-pro/models/keys';
-import { SystemTables} from '@/models/SystemTables';
+import { SystemTables } from '@/models/SystemTables';
 import { LRUCache } from 'lru-cache';
 import { BizException } from '@/models/devops';
 import { CurdMixService } from '../curd/CurdMixService';
@@ -65,7 +65,6 @@ export class AnyApiService extends ApiBaseService {
   }
 
   private async getAnyApiMethod(methodCode: string): Promise<ISysAnyApiEntity> {
-
     // 开发环境，不使用缓存
     if (this.isEnableDebug()) {
       return this._getAnyApiMethod(methodCode);

package/src/service/asyncTask/AsyncTaskRunnerService.ts

@@ -10,14 +10,10 @@ import { KeysOfSimpleSQL } from '@/libs/crud-pro/models/keys';
 import { errorToString } from '@/libs/utils/errorToString';
 import { ANONYMOUS_CONTEXT } from '@/schedule';
 import { RedisKeys } from '@/models/RedisKeys';
+import { getCurrentFullMoment } from '@/libs/utils/functions';
+import { ExportExcelAsyncTaskHandler } from './handler/ExportExcelAsyncTaskHandler';
 
-const END_STATUS_LIST = [
-  SysAsyncTaskStatus.PART_SUCCEEDED,
-  SysAsyncTaskStatus.SUCCEEDED,
-  SysAsyncTaskStatus.FAILED,
-  SysAsyncTaskStatus.CANCELLED,
-  SysAsyncTaskStatus.PAUSED,
-];
+const END_STATUS_LIST = [SysAsyncTaskStatus.PART_SUCCEEDED, SysAsyncTaskStatus.SUCCEEDED, SysAsyncTaskStatus.FAILED, SysAsyncTaskStatus.CANCELLED, SysAsyncTaskStatus.PAUSED];
 
 class AsyncTaskRunner {
   isBusy = false;
@@ -35,6 +31,13 @@ class AsyncTaskRunner {
       taskElement.task_status = SysAsyncTaskStatus.RUNNING;
 
       try {
+        // 1. 更新当前这个任务的状态
+        await this.updateTaskStatus(taskElement, {
+          task_status: taskElement.task_status,
+          started_at: getCurrentFullMoment(),
+          updated_at: getCurrentFullMoment(),
+        });
+
         // 任务状态的设置在自定义的执行器里面。
         await this.executeTask(taskElement);
 
@@ -42,19 +45,21 @@ class AsyncTaskRunner {
         if (!END_STATUS_LIST.includes(taskElement.task_status)) {
           taskElement.task_status = SysAsyncTaskStatus.SUCCEEDED;
         }
-
       } catch (error) {
         taskElement.task_status = SysAsyncTaskStatus.FAILED;
         taskElement.error_message = errorToString(error);
       }
 
       try {
+        // 2. 更新当前这个任务的状态
         await this.updateTaskStatus(taskElement, {
           task_status: taskElement.task_status,
           error_message: taskElement.error_message,
+          updated_at: getCurrentFullMoment(),
+          completed_at: getCurrentFullMoment(),
         });
       } catch (error) {
-        ANONYMOUS_CONTEXT.getApp().getCoreLogger().error('executeTaskList error', error);
+        ANONYMOUS_CONTEXT.getApp().getCoreLogger().error('[AsyncTaskRunner] executeTaskList error', error);
       }
     }
 
@@ -119,11 +124,17 @@ class AsyncTaskRunner {
  */
 export const ASYNC_TASK_RUNNER = new AsyncTaskRunner();
 
+//内置的异步任务处理器：EXCEL导出
+ASYNC_TASK_RUNNER.taskHandlerMap.set('EXCEL_EXPORT', new ExportExcelAsyncTaskHandler());
+
 /**
  * Redis锁
  */
 const ASYNC_TASK_LOCK = RedisKeys.ASYNC_TASK_LOCK;
 
+/**
+ * 上次检查异步任务的时间点
+ */
 const ASYNC_TASK_RUNTIME_OBJ = {
   LAST_CHECK_ASYNC_TASK_UPDATE_TIME: 0,
 };
@@ -194,7 +205,7 @@ export class AsyncTaskRunnerService extends BaseService implements IScheduleServ
 
     // 开始执行。
     ASYNC_TASK_RUNNER.executeTaskList(taskList).then(() => {
-      console.log('ASYNC_TASK_RUNNER finished taskIds ==> ' + JSON.stringify(taskIds));
+      console.log('[AsyncTaskRunnerService] ASYNC_TASK_RUNNER finished taskIds ==> ' + JSON.stringify(taskIds));
     });
   }
 
@@ -220,7 +231,7 @@ export class AsyncTaskRunnerService extends BaseService implements IScheduleServ
     try {
       await this.fetchPendingTasks();
     } catch (e) {
-      console.error('fetchPendingTasks error', errorToString(e));
+      console.error('[AsyncTaskRunnerService] fetchPendingTasks error', errorToString(e));
     }
 
     await this.redisService.del(ASYNC_TASK_LOCK);

package/src/service/asyncTask/handler/ExcelInfoModel.ts

@@ -0,0 +1,11 @@
+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/src/service/asyncTask/handler/ExportExcelAsyncTaskHandler.ts

@@ -0,0 +1,245 @@
+import { IMidwayKoaContext } from '@midwayjs/koa';
+import { ISysAsyncTaskHandler, SysAsyncTaskContext, SysAsyncTaskStatus } from '@/models/AsyncTaskModel';
+import { ANONYMOUS_CONTEXT } from '@/schedule';
+import { IExcelHeaderInfo, IExcelAsyncTaskHandler } from './ExcelInfoModel';
+import { SysAsyncFileFormat } from '@/models/AsyncTaskModel';
+import { ExportExcelByStdCrudHandler } from './ExportExcelByStdCrudHandler';
+import { ExportExcelByInnerHttpHandler } from './ExportExcelByInnerHttpHandler';
+import { IStdCrudExportInputParams, StdCrudExportInputParamsAppType } from '@/models/bizmodels';
+
+import * as _ from 'lodash';
+import * as fastcsv from 'fast-csv';
+import * as fs from 'node:fs';
+import * as fs2 from 'node:fs/promises';
+import * as path from 'node:path';
+import { parseJsonObject } from '@/libs/utils/functions';
+import { UserSessionInfo } from '@/models/userSession';
+import { errorToString } from '@/libs/utils/errorToString';
+
+const indexRef = { current: 0 };
+
+async function createFileFolder(distFileFolder: string) {
+  try {
+    await fs2.access(distFileFolder);
+  } catch (e) {
+    await fs2.mkdir(distFileFolder, { recursive: true });
+  }
+}
+
+function getExcelAsyncTaskHandler(asyncTaskContext: SysAsyncTaskContext, ctx: IMidwayKoaContext): IExcelAsyncTaskHandler {
+  const inputParams: IStdCrudExportInputParams = parseJsonObject(asyncTaskContext?.task?.input_params);
+  if (inputParams && inputParams.appType === StdCrudExportInputParamsAppType.STD_CRUD) {
+    return new ExportExcelByStdCrudHandler(asyncTaskContext, inputParams, ctx);
+  }
+  if (inputParams && inputParams.appType === StdCrudExportInputParamsAppType.INNER_HTTP) {
+    return new ExportExcelByInnerHttpHandler(asyncTaskContext, inputParams, ctx);
+  }
+  throw new Error('[getExcelAsyncTaskHandler]appType不支持；appType = ' + inputParams.appType);
+}
+
+/**
+ * 多实例，每个任务一个实例。
+ * 导出EXCEL异步任务:
+ */
+class ExportExcelAsyncTask {
+  protected asyncTaskContext: SysAsyncTaskContext;
+  protected ctx: IMidwayKoaContext;
+  protected headerColumns: IExcelHeaderInfo[];
+  protected excelAsyncTaskHandler: IExcelAsyncTaskHandler;
+
+  constructor(asyncTaskContext: SysAsyncTaskContext, ctx: IMidwayKoaContext) {
+    this.asyncTaskContext = asyncTaskContext;
+    this.ctx = ctx;
+    this.excelAsyncTaskHandler = getExcelAsyncTaskHandler(asyncTaskContext, ctx);
+  }
+
+  private getDistFileFolder() {
+    const app = 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: 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 = SysAsyncTaskStatus.SUCCEEDED;
+    } catch (error) {
+      taskElement.task_status = SysAsyncTaskStatus.FAILED;
+      taskElement.error_message = 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);
+  }
+
+  private formatToCsvHeader(headerColumns: IExcelHeaderInfo[]): string[] {
+    return headerColumns.map(item => {
+      return item.title;
+    });
+  }
+
+  private async writeExcelBatchData(csvStream: fastcsv.CsvFormatterStream<any, any>, pageNo: number, pageSize: number): Promise<number> {
+    const dataList: any[] = 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;
+  }
+
+  private formatToCsvRow(rowObj: any): string[] {
+    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 文件名
+   */
+  private async getExcelFileName() {
+    indexRef.current = indexRef.current + 1;
+    if (indexRef.current > 100000) {
+      indexRef.current = 1;
+    }
+    const num = `${indexRef.current}`.padStart(10);
+    return `export_${Date.now()}_${num}.csv`;
+  }
+
+  /**
+   * 获取Excel的表头
+   * @returns
+   */
+  private async getExcelHeader(): Promise<IExcelHeaderInfo[]> {
+    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' }
+    // ]
+  }
+
+  /**
+   * 获取文件总数
+   */
+  private async getTotalCount(): Promise<number> {
+    const totalCount = await this.excelAsyncTaskHandler.getTotalCount();
+    // 最多导出10万条数据
+    return Math.min(totalCount, 10 * 10000);
+  }
+
+  /**
+   * 获取数据内容数组
+   * @param pageNo 当前页号，从1开始
+   * @param pageSize 每页获取的数量。
+   */
+  private async getExcelDataList(pageNo: number, pageSize: number): Promise<any[]> {
+    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;
+  }
+}
+
+/**
+ * 单例
+ */
+export class ExportExcelAsyncTaskHandler implements ISysAsyncTaskHandler {
+  public async execute(asyncTaskContext: SysAsyncTaskContext): Promise<any> {
+    const taskId = _.get(asyncTaskContext, 'task.id');
+
+    console.log('[ExportExcelAsyncTaskHandler] execute, taskId=>' + taskId);
+
+    const res = await ANONYMOUS_CONTEXT.runServiceAtAnonymousContext(async (ctx: IMidwayKoaContext) => {
+      const createdUserSession = parseJsonObject(asyncTaskContext?.task?.created_user_session);
+      ctx.userSession = new UserSessionInfo(createdUserSession, false);
+      const task = new ExportExcelAsyncTask(asyncTaskContext, ctx);
+      return task.executeWithContext();
+    });
+    if (res.error) {
+      throw res.error;
+    }
+    return res.result;
+  }
+}

package/src/service/asyncTask/handler/ExportExcelByInnerHttpHandler.ts

@@ -0,0 +1,147 @@
+import * as _ from 'lodash';
+import axios from 'axios';
+import { IMidwayKoaContext } from '@midwayjs/koa';
+import { IExcelAsyncTaskHandler, IExcelHeaderInfo } from './ExcelInfoModel';
+import { SysAsyncTaskContext } from '@/models/AsyncTaskModel';
+import { IStdCrudExportInputParams, ITableProQueryParams } from '@/models/bizmodels';
+import { IRequestModel } from '@/libs/crud-pro/interfaces';
+import { ANONYMOUS_CONTEXT } from '@/schedule';
+
+function pickData(a: any, key: string) {
+  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 implements IExcelAsyncTaskHandler {
+  protected asyncTaskContext: SysAsyncTaskContext;
+  protected ctx: IMidwayKoaContext;
+  protected inputParams: IStdCrudExportInputParams;
+  protected appCode: string;
+  public constructor(asyncTaskContext: SysAsyncTaskContext, inputParams: IStdCrudExportInputParams, ctx: IMidwayKoaContext) {
+    this.ctx = ctx;
+    this.inputParams = inputParams;
+    this.asyncTaskContext = asyncTaskContext;
+    this.appCode = this.inputParams.appCode;
+  }
+
+  async getPageSize(): Promise<number> {
+    if (typeof this.inputParams.pageSize === 'number') {
+      return Math.min(this.inputParams.pageSize, 1000);
+    }
+    return 1000;
+  }
+
+  /**
+   * 获取Excel表的表头
+   * @returns 表头信息
+   */
+  async getExcelHeader(): Promise<IExcelHeaderInfo[]> {
+    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);
+  }
+
+  private async sendRequest(params: any): Promise<any> {
+    const port = ANONYMOUS_CONTEXT.getApp().getConfig().koa.port;
+    let queryApi = this.inputParams.innerHttpUrl; // 以斜杠开头
+    if (!queryApi.startsWith('/')) {
+      queryApi = '/' + queryApi;
+    }
+
+    params.runByExportExcel = true;
+
+    const response = await axios({
+      method: 'POST',
+      url: `http://127.0.0.1:${port}${queryApi}`,
+      data: params,
+      headers: this.inputParams.headers || {},
+      responseType: 'json',
+      validateStatus: () => true,
+    });
+    if (response.status !== 200) {
+      throw new Error('导出Excel请求错误失败，状态码：' + response.status);
+    }
+    return response.data;
+  }
+
+  /**
+   * 获取要导出的数量总数
+   * @returns 数据总量
+   */
+  async getTotalCount(): Promise<number> {
+    const selectedRowKeys: string[] | number[] = this.inputParams?.selectedRowKeys || [];
+    if (Array.isArray(selectedRowKeys) && selectedRowKeys.length > 0) {
+      return selectedRowKeys.length;
+    }
+
+    const lastQueryParams: ITableProQueryParams = this.inputParams?.lastQueryParams;
+    const res = await this.sendRequest({
+      condition: lastQueryParams?.condition || {},
+      orderBy: lastQueryParams?.orderBy,
+      pageSize: 1,
+      pageNo: 1,
+    });
+    return pickData(res, 'total_count');
+  }
+
+  /**
+   * 获取要导出Excel的数据的内容
+   * @param pageNo 页号，从1开始
+   * @param pageSize 每页数量
+   * @returns
+   */
+  async getExcelDataList(pageNo: number, pageSize: number): Promise<any[]> {
+    const selectedRowKeys: string[] | number[] = this.inputParams?.selectedRowKeys || [];
+    const lastQueryParams: ITableProQueryParams = this.inputParams?.lastQueryParams;
+    const primaryKey: string = this.inputParams.primaryKey || 'id';
+
+    const params: IRequestModel = {
+      condition: lastQueryParams?.condition || {},
+      pageSize: pageSize,
+      pageNo: pageNo,
+      orderBy: lastQueryParams?.orderBy,
+    };
+
+    if (Array.isArray(selectedRowKeys) && selectedRowKeys.length > 0) {
+      params.condition[primaryKey] = {
+        $in: selectedRowKeys,
+      };
+    }
+    const res = await this.sendRequest({
+      condition: lastQueryParams?.condition || {},
+      orderBy: lastQueryParams?.orderBy,
+      pageSize: pageSize,
+      pageNo: pageNo,
+    });
+    return pickData(res, 'rows');
+  }
+}
+
+export { ExportExcelByInnerHttpHandler };