midway-fatcms 0.0.7 → 0.0.9

package/src/libs/crud-pro/services/CrudProTableMetaService.ts

@@ -39,6 +39,24 @@ class CrudProTableMetaCache {
       expiredTime: Date.now() + cacheTime,
     };
   }
+
+  /**
+   * 清空所有缓存
+   */
+  public clearAll(): void {
+    this.cacheMap = {};
+    this.tableInfoCacheMap = {};
+  }
+
+  /**
+   * 获取缓存统计信息
+   */
+  public getCacheStats(): { metaCacheCount: number; tableInfoCacheCount: number } {
+    return {
+      metaCacheCount: Object.keys(this.cacheMap).length,
+      tableInfoCacheCount: Object.keys(this.tableInfoCacheMap).length,
+    };
+  }
 }
 
 const metaCache = new CrudProTableMetaCache();
@@ -53,6 +71,40 @@ class CrudProTableMetaService extends CrudProServiceBase {
     return obj;
   }
 
+  /**
+   * 清空所有表元数据缓存
+   *
+   * 清空后，下次请求会自动从数据库重新加载。
+   * 适用于定时刷新或手动触发缓存失效的场景。
+   */
+  public clearAllCache(): void {
+    metaCache.clearAll();
+  }
+
+  /**
+   * 获取缓存统计信息
+   */
+  public getCacheStats(): { metaCacheCount: number; tableInfoCacheCount: number } {
+    return metaCache.getCacheStats();
+  }
+
+  /**
+   * 获取表的主键列名列表
+   *
+   * 复用 getTableMeta 的缓存机制，从 columnDetails 中筛选 isPrimaryKey 的列，
+   * 避免重复查询数据库。
+   *
+   * @param query 表元数据查询参数
+   * @returns 主键列名数组
+   */
+  public async getPrimaryKeyColumns(query: ITableMetaQuery): Promise<string[]> {
+    const meta = await this.getTableMeta(query);
+    const pks = (meta.columnDetails || [])
+      .filter(col => col.isPrimaryKey)
+      .map(col => col.name);
+    return pks;
+  }
+
   // ============ 获取表和视图信息（带类型） ============
 
 
@@ -84,8 +136,7 @@ class CrudProTableMetaService extends CrudProServiceBase {
 
     // 3. 更新缓存
     if (!skipCache || refreshCache) {
-      const { tableMetaCacheTime } = this.getContextCfg();
-      metaCache.setTableInfos(cacheKey, result, tableMetaCacheTime || 3600000);
+      metaCache.setTableInfos(cacheKey, result,  1000 * 3600 * 24 * 365);
     }
 
     return result;
@@ -245,12 +296,16 @@ class CrudProTableMetaService extends CrudProServiceBase {
     // 获取字段注释
     const commentMap = await this.loadPostgreSQLColumnComments(baseInfo, schemaname);
 
+    // 获取主键列
+    const pkSet = await this.loadPostgreSQLPrimaryKeys(baseInfo, schemaname);
+
     return columnArray.map((columnObj: any) => {
       const { column_name, data_type, is_nullable, column_default, character_maximum_length } = columnObj;
       return {
         name: column_name,
         type: data_type,
         isNullable: is_nullable === 'YES',
+        isPrimaryKey: pkSet.has(column_name),
         defaultValue: column_default,
         maxLength: character_maximum_length,
         comment: commentMap[column_name],
@@ -288,6 +343,27 @@ class CrudProTableMetaService extends CrudProServiceBase {
     return map;
   }
 
+  private async loadPostgreSQLPrimaryKeys(
+    baseInfo: IExecuteUnsafeQueryCtx,
+    schemaname: string
+  ): Promise<Set<string>> {
+    const pkSql = `
+      SELECT a.attname
+      FROM pg_index i
+      JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
+      JOIN pg_class c ON c.oid = i.indrelid
+      JOIN pg_namespace n ON n.oid = c.relnamespace
+      WHERE n.nspname = '${schemaname}'
+        AND c.relname = '${baseInfo.sqlTable}'
+        AND i.indisprimary
+      ORDER BY array_position(i.indkey, a.attnum);
+    `.trim();
+
+    const queryRes = await this.executeUnsafeQuery(baseInfo, pkSql);
+    const pkRows = queryRes.rows || [];
+    return new Set(pkRows.map((row: any) => row.attname));
+  }
+
   private async loadSQLServerColumnDetails(baseInfo: IExecuteUnsafeQueryCtx): Promise<ITableColumn[]> {
     const columnArraySql = `
       SELECT
@@ -311,19 +387,50 @@ class CrudProTableMetaService extends CrudProServiceBase {
     const queryRes = await this.executeUnsafeQuery(baseInfo, columnArraySql);
     const columnArray = queryRes.rows || [];
 
+    // 获取主键列
+    const pkSet = await this.loadSQLServerPrimaryKeys(baseInfo);
+
     return columnArray.map((columnObj: any) => {
       const { column_name, data_type, is_nullable, column_default, column_comment, max_length, is_identity } = columnObj;
       return {
         name: column_name,
         type: data_type,
         isNullable: is_nullable,
+        isPrimaryKey: pkSet.has(column_name),
         defaultValue: column_default,
         comment: column_comment,
         maxLength: max_length,
-        isPrimaryKey: is_identity,
+        isIdentity: is_identity,
       };
     });
   }
+
+  private async loadSQLServerPrimaryKeys(baseInfo: IExecuteUnsafeQueryCtx): Promise<Set<string>> {
+    const pkSql = `
+      SELECT COL_NAME(ic.object_id, ic.column_id) AS column_name
+      FROM sys.indexes i
+      INNER JOIN sys.index_columns ic ON i.object_id = ic.object_id AND i.index_id = ic.index_id
+      WHERE i.object_id = OBJECT_ID('${baseInfo.sqlTable}')
+        AND i.is_primary_key = 1
+      ORDER BY ic.key_ordinal;
+    `.trim();
+
+    const queryRes = await this.executeUnsafeQuery(baseInfo, pkSql);
+    const pkRows = queryRes.rows || [];
+    return new Set(pkRows.map((row: any) => row.column_name));
+  }
 }
 
 export { CrudProTableMetaService };
+
+/**
+ * 清空所有表元数据缓存（独立函数，无需实例化 Service 即可调用）
+ *
+ * 适用于定时任务等无法通过 Service 实例访问的场景。
+ * 清空后，下次请求会自动从数据库重新加载（懒加载刷新）。
+ */
+export function clearTableMetaCache(): { metaCacheCount: number; tableInfoCacheCount: number } {
+  const stats = metaCache.getCacheStats();
+  metaCache.clearAll();
+  return stats;
+}

package/src/libs/crud-pro/services/CurdProServiceHub.ts

@@ -98,6 +98,10 @@ class CurdProServiceHub implements ICurdProServiceHub {
     return await this.tableMetaService.getTableMeta(query);
   }
 
+  async getPrimaryKeyColumns(query: ITableMetaQuery): Promise<string[]> {
+    return await this.tableMetaService.getPrimaryKeyColumns(query);
+  }
+
   async getAllTableInfos(query: ITableNamesQuery, options?: ITableNamesOptions): Promise<ITableListResult> {
     return await this.tableMetaService.getAllTableInfos(query, options);
   }
@@ -109,6 +113,10 @@ class CurdProServiceHub implements ICurdProServiceHub {
   async convertDataTypeByTableMeta(reqModel: RequestModel, cfgModel: RequestCfgModel): Promise<void> {
     return await this.dataTypeConvertService.convertDataTypeByTableMeta(reqModel, cfgModel);
   }
+
+  async convertConditionTypeByTableMeta(reqModel: RequestModel, cfgModel: RequestCfgModel): Promise<void> {
+    return await this.dataTypeConvertService.convertConditionTypeByTableMeta(reqModel, cfgModel);
+  }
 }
 
 export { CurdProServiceHub };