gatway

main.go

@@ -5,10 +5,11 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
 	"model-asynch/controller"
+	"model-asynch/service"
 
-	_ "gitea.com/red-future/common/config"
 	"gitea.com/red-future/common/http"
 	"gitea.com/red-future/common/jaeger"
 	_ "gitea.com/red-future/common/swagger"
@@ -25,11 +26,13 @@ func main() {
 	// 注册路由
 	http.RouteRegister([]interface{}{
 		controller.Model,
-		controller.ModelType,
 		controller.Task,
 		controller.Stat,
 	})
 
+	// 本地调试：可选自动触发 worker/cleaner（由配置文件控制）
+	startAutoRunner(ctx)
+
 	// 监听退出信号，确保 Ctrl+C 能完整退出（停止 worker/cleaner 并关闭 http server）
 	quit := make(chan os.Signal, 1)
 	signal.Notify(quit, os.Interrupt, syscall.SIGTERM)
@@ -40,3 +43,49 @@ func main() {
 	// 关闭 http server（RouteRegister 内部是 go Httpserver.Run() 启动的）
 	_ = http.Httpserver.Shutdown()
 }
+
+func startAutoRunner(ctx context.Context) {
+	// worker
+	if g.Cfg().MustGet(ctx, "asynch.worker.enabled").Bool() {
+		interval := g.Cfg().MustGet(ctx, "asynch.worker.intervalSeconds").Int()
+		if interval <= 0 {
+			interval = 5
+		}
+		batchSize := g.Cfg().MustGet(ctx, "asynch.worker.batchSize").Int()
+		goroutines := g.Cfg().MustGet(ctx, "asynch.worker.goroutines").Int()
+		ticker := time.NewTicker(time.Duration(interval) * time.Second)
+		go func() {
+			defer ticker.Stop()
+			for {
+				select {
+				case <-ctx.Done():
+					return
+				case <-ticker.C:
+					if _, err := service.AsyncWorker.RunOnce(ctx, batchSize, goroutines); err != nil {
+						g.Log().Warningf(ctx, "[auto-worker] run once failed: %v", err)
+					}
+				}
+			}
+		}()
+	}
+
+	// cleaner
+	if g.Cfg().MustGet(ctx, "asynch.cleaner.enabled").Bool() {
+		interval := g.Cfg().MustGet(ctx, "asynch.cleaner.intervalSeconds").Int()
+		if interval <= 0 {
+			interval = 30
+		}
+		ticker := time.NewTicker(time.Duration(interval) * time.Second)
+		go func() {
+			defer ticker.Stop()
+			for {
+				select {
+				case <-ctx.Done():
+					return
+				case <-ticker.C:
+					service.Cleaner.RunOnce(ctx)
+				}
+			}
+		}()
+	}
+}