feat: add river-backed async task queue

2026-05-12 10:11:54 +08:00
parent d69aaed444
commit 7e220b7477
30 changed files with 1342 additions and 200 deletions
@@ -33,18 +33,19 @@ func main() {
 	if recovery, err := db.RecoverInterruptedRuntimeState(ctx); err != nil {
 		logger.Error("recover interrupted runtime state failed", "error", err)
 		os.Exit(1)
-	} else if recovery.ReleasedConcurrencyLeases > 0 || recovery.ReleasedRateReservations > 0 || recovery.FailedAttempts > 0 || recovery.FailedTasks > 0 {
+	} else if recovery.ReleasedConcurrencyLeases > 0 || recovery.ReleasedRateReservations > 0 || recovery.FailedAttempts > 0 || recovery.FailedTasks > 0 || recovery.RequeuedAsyncTasks > 0 {
 		logger.Warn("interrupted runtime state recovered",
 			"releasedConcurrencyLeases", recovery.ReleasedConcurrencyLeases,
 			"releasedRateReservations", recovery.ReleasedRateReservations,
 			"failedAttempts", recovery.FailedAttempts,
 			"failedTasks", recovery.FailedTasks,
+			"requeuedAsyncTasks", recovery.RequeuedAsyncTasks,
 		)
 	}

 	server := &http.Server{
 		Addr:              cfg.HTTPAddr,
-		Handler:           httpapi.NewServer(cfg, db, logger),
+		Handler:           httpapi.NewServerWithContext(ctx, cfg, db, logger),
 		ReadHeaderTimeout: 10 * time.Second,
 	}