midway-fatcms 0.0.5 → 0.0.7

package/src/service/curd/CurdProService.ts

@@ -2,7 +2,7 @@ import * as _ from 'lodash';
 import { Config, Inject, Provide } from '@midwayjs/core';
 import { CrudPro } from '@/libs/crud-pro/CrudPro';
 import { Context } from '@midwayjs/koa';
-import { IConnectionPool, ICrudProCfg, IRequestCfgModel, IRequestModel, ISqlCfgModel } from '@/libs/crud-pro/interfaces';
+import { IConnectionPool, ICrudProCfg, IRequestCfgModel, IRequestModel, ISqlCfgModel, ITableListResult, ITableNamesOptions, ITableNamesQuery } from '@/libs/crud-pro/interfaces';
 import { SqlCfgModel } from '@/libs/crud-pro/models/SqlCfgModel';
 import { getConnectionPool as getMySQLConnectionPool } from '../../libs/crud-pro/utils/pool/MySQLUtils';
 import { getConnectionPool as getPostgresConnectionPool } from '../../libs/crud-pro/utils/pool/PostgresUtils';
@@ -16,7 +16,8 @@ import { RelatedType } from './CurdMixUtils';
 import { BaseService } from '../base/BaseService';
 import { IRequestCfgModel2, IVisitorExt } from '@/models/bizmodels';
 import { IWorkbenchEntity } from '@/models/SystemEntities';
-import { CrudProQuick } from '@/service/curd/CrudProQuick';
+import { CrudProQuick } from './CrudProQuick';
+import { ShardingCrudPro, IShardingConfig } from '@/libs/crud-sharding';
 import { fixCfgModel } from './fixCfgModel';
 import { GLOBAL_STATIC_CONFIG } from '@/libs/global-config/global-config';
 import { fixSoftDelete } from "@/service/curd/fixSoftDelete";
@@ -220,9 +221,62 @@ export class CurdProService extends BaseService {
     return await curdPro.executeCrudByCfg(reqJson, cfgModel);
   }
 
+  /**
+   * 获取 CrudProQuick 工具类（已废弃）
+   *
+   * @deprecated 请使用 getQuickCrud 替代
+   * @param sqlDatabase 数据库名
+   * @param sqlDbType 数据库类型
+   * @param sqlTable 表名（可选）
+   * @returns CrudProQuick 实例
+   */
   public getBbUtil(sqlDatabase: string, sqlDbType: SqlDbType, sqlTable?: string): CrudProQuick {
+    return this.getQuickCrud(sqlDatabase, sqlDbType, sqlTable);
+  }
+
+  /**
+   * 获取 CrudProQuick 工具类
+   *
+   * 提供快速 CRUD 操作的便捷方法，封装了常用的数据库操作。
+   *
+   * @param sqlDatabase 数据库名
+   * @param sqlDbType 数据库类型
+   * @param sqlTable 表名（可选）
+   * @returns CrudProQuick 实例
+   *
+   * @example
+   * const quick = curdProService.getQuickCrud('mydb', SqlDbType.mysql, 't_user');
+   * const users = await quick.query({ status: 'active' });
+   */
+  public getQuickCrud(sqlDatabase: string, sqlDbType: SqlDbType, sqlTable?: string): CrudProQuick {
+    const curdPro = this.getCrudPro();
+    return new CrudProQuick({ curdPro, sqlDatabase, sqlDbType, sqlTable, ctx: this.ctx });
+  }
+
+  /**
+   * 获取分表 CRUD 操作器
+   *
+   * @param sqlDatabase 数据库名
+   * @param sqlDbType 数据库类型
+   * @param shardingConfig 分表配置
+   * @returns ShardingCrudPro 实例
+   *
+   * @example
+   * const sharding = curdProService.getShardingCrud('mydb', SqlDbType.mysql, {
+   *   type: ShardingType.MONTH,
+   *   baseTable: 't_order',
+   *   timeColumn: 'created_at',
+   * });
+   */
+  public getShardingCrud(
+    sqlDatabase: string,
+    sqlDbType: SqlDbType,
+    shardingConfig: IShardingConfig
+  ): ShardingCrudPro {
     const curdPro = this.getCrudPro();
-    return new CrudProQuick({curdPro, sqlDatabase, sqlDbType, sqlTable, ctx: this.ctx});
+    const sharding = new ShardingCrudPro(curdPro, shardingConfig);
+    sharding.setBaseCfg({ sqlDatabase, sqlDbType });
+    return sharding;
   }
 
   public async getCachedCfgByMethod(method: string): Promise<IRequestCfgModel> {
@@ -230,4 +284,9 @@ export class CurdProService extends BaseService {
     const isEnableCache = !this.isEnableDebug(); // 开发环境不使用缓存
     return curdPro.getCachedCfgByMethod(method, isEnableCache);
   }
+
+  public async getAllTableInfos(query: ITableNamesQuery, options?: ITableNamesOptions): Promise<ITableListResult> {
+    const curdPro = this.getCrudPro();
+    return curdPro.getAllTableInfos(query, options);
+  }
 }