add a retry on migration

bluesky-social · Nov 8, 2023 · 375d44d · 375d44d
1 parent fd4c745
commit 375d44d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/packages/pds/src/migrate-script/migrate-all.ts b/packages/pds/src/migrate-script/migrate-all.ts
@@ -15,6 +15,7 @@ import {
   makeAdminHeaders,
   repairBlob,
   repairFailedPrefs,
+  retryOnce,
   transferPreferences,
 } from './util'
 
@@ -66,7 +67,9 @@ export const runScript = async () => {
     }
     migrateQueue.add(async () => {
       try {
-        await migrateRepo(ctx, db, pdsInfo, status, adminHeaders)
+        await retryOnce(() =>
+          migrateRepo(ctx, db, pdsInfo, status, adminHeaders),
+        )
         await db
           .updateTable('status')
           .set({ failed: 0, err: null })

diff --git a/packages/pds/src/migrate-script/util.ts b/packages/pds/src/migrate-script/util.ts
@@ -27,6 +27,14 @@ export const makeAdminHeaders = (secrets: ServerSecrets): AdminHeaders => {
   }
 }
 
+export const retryOnce = async (fn: () => Promise<unknown>) => {
+  try {
+    await fn()
+  } catch {
+    await fn()
+  }
+}
+
 export const repairFailedPrefs = async (
   ctx: AppContext,
   db: MigrateDb,