fix: import table add projection field option

teableio · Jan 6, 2025 · aeb61ca · aeb61ca
1 parent b3b3caa
commit aeb61ca
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/apps/nestjs-backend/src/features/import/open-api/import-open-api.service.ts b/apps/nestjs-backend/src/features/import/open-api/import-open-api.service.ts
@@ -128,7 +128,8 @@ export class ImportOpenApiService {
     baseId: string,
     tableId: string,
     inplaceImportRo: IInplaceImportOptionRo,
-    maxRowCount?: number
+    maxRowCount?: number,
+    projection?: string[]
   ) {
     const userId = this.cls.get('user.id');
     const { attachmentUrl, fileType, insertConfig, notification = false } = inplaceImportRo;
@@ -144,14 +145,18 @@ export class ImportOpenApiService {
         throw new BadRequestException('table is not found');
       });
 
-    const fieldRaws = await this.prismaService.field.findMany({
+    let fieldRaws = await this.prismaService.field.findMany({
       where: { tableId, deletedTime: null, hasError: null },
       select: {
         id: true,
         type: true,
       },
     });
 
+    if (projection?.length) {
+      fieldRaws = fieldRaws.filter(({ id }) => projection.includes(id));
+    }
+
     if (!tableRaw || !fieldRaws) {
       return;
     }