fix: harden generation task coordination

2026-06-08 15:00:19 +08:00
parent 5367e8c2fd
commit ea91155f9e
2 changed files with 309 additions and 5 deletions
@@ -144,7 +144,9 @@ async function main() {
  startSettlementWorker()
  startProviderHealthMonitor()

-  const { startStaleTaskCleanup } = require('./aiTaskWorker')
+  const { startStaleTaskCleanup, startTaskEventListener, startPollerRecovery } = require('./aiTaskWorker')
+  await startTaskEventListener()
+  startPollerRecovery()
  startStaleTaskCleanup()

  server = app.listen(PORT, HOST, () => {
@@ -183,6 +185,9 @@ function gracefulShutdown(signal) {
      console.log('[shutdown] Server closed, cleaning up...')
      const { stopProviderHealthMonitor } = require('./providerHealthMonitor')
      stopProviderHealthMonitor()
+      const { stopTaskEventListener, stopPollerRecovery } = require('./aiTaskWorker')
+      stopPollerRecovery()
+      void stopTaskEventListener()
      const { pool } = require('./db')
      pool.end().then(() => {
        console.log('[shutdown] Database pool closed')