graceful shutdown

bluesky-social · Nov 8, 2023 · aa7a337 · aa7a337
1 parent dc6b7f6
commit aa7a337
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/pds/src/migrate-script/migrate-all.ts b/packages/pds/src/migrate-script/migrate-all.ts
@@ -37,8 +37,17 @@ export const runScript = async () => {
   let pdsCounter = 0
   let completed = 0
   let failed = 0
+
   console.log('migrating: ', todo.length)
+
   const migrateQueue = new PQueue({ concurrency: 40 })
+  process.on('SIGINT', async () => {
+    migrateQueue.clear()
+    console.log(`waiting on ${migrateQueue.pending} to finish`)
+    await migrateQueue.onIdle()
+    process.exit(0)
+  })
+
   for (const status of todo) {
     if (!status.pdsId) {
       status.pdsId = pdsInfos[pdsCounter % pdsInfos.length].id