Update src/index.ts

Co-authored-by: Shigma <[email protected]>
koishijs · Aug 15, 2024 · 7b2aa62 · 7b2aa62
1 parent a6bd059
commit 7b2aa62
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -309,9 +309,11 @@ export function apply(ctx: Context, config: Config) {
         if (globalTasks.size >= config.globalConcurrency) {
           const pendingId = container.pop()
           globalPending.add(pendingId)
-          await new Promise<void>((resolve) => ctx.once('novelai/finish', (id) => {
-            if (id === pendingId) {
+          await new Promise<void>((resolve) => {
+            const dispose = ctx.on('novelai/finish', (id) => {
+              if (id !== pendingId) return
               resolve()
+              dispose()
             }
           }))
         }