feat(server): blob data migration (#5461)

2026-07-19 11:06:25 +08:00 · 2024-01-03 10:56:57 +00:00
parent 0d34805375
commit 760d900f99
11 changed files with 109 additions and 39 deletions
@@ -0,0 +1,38 @@
+import { ModuleRef } from '@nestjs/core';
+
+import { WorkspaceBlobStorage } from '../../modules/storage';
+import { PrismaService } from '../../prisma';
+
+export class WorkspaceBlobs1703828796699 {
+  // do the migration
+  static async up(db: PrismaService, injector: ModuleRef) {
+    const blobStorage = injector.get(WorkspaceBlobStorage, { strict: false });
+    let hasMore = true;
+    let turn = 0;
+    const eachTurnCount = 50;
+
+    while (hasMore) {
+      const blobs = await db.blob.findMany({
+        skip: turn * eachTurnCount,
+        take: eachTurnCount,
+        orderBy: {
+          createdAt: 'asc',
+        },
+      });
+
+      hasMore = blobs.length === eachTurnCount;
+      turn += 1;
+
+      await Promise.all(
+        blobs.map(async ({ workspaceId, hash, blob }) =>
+          blobStorage.put(workspaceId, hash, blob)
+        )
+      );
+    }
+  }
+
+  // revert the migration
+  static async down(_db: PrismaService) {
+    // old data kept, no need to downgrade the migration
+  }
+}