diff --git a/backend/cmd/api/main.go b/backend/cmd/api/main.go index e0dfe2b..eaae184 100644 --- a/backend/cmd/api/main.go +++ b/backend/cmd/api/main.go @@ -1,11 +1,9 @@ package main import ( - "context" "log" "senlinai-agent/backend/internal/config" - "senlinai-agent/backend/internal/crontab" "senlinai-agent/backend/internal/httpx" "senlinai-agent/backend/internal/initdb" "senlinai-agent/backend/internal/logic/ai" @@ -33,7 +31,6 @@ func main() { authService := auth.NewService(cfg.AuthSecret) datasetService := dataset.NewService(models.DBService) datasetHandler := dataset.NewHandler(datasetService) - datasetScheduler := crontab.NewDatasetScheduler(datasetService, log.Default()) projectService := projects.NewService() documentService := documents.NewService(models.DBService, cfg.StorageDir) taskService := tasks.NewService(models.DBService) @@ -63,9 +60,6 @@ func main() { searchHandler, aiHandler, ) - schedulerContext, stopScheduler := context.WithCancel(context.Background()) - defer stopScheduler() - go datasetScheduler.Run(schedulerContext) if err := appRouter.Run(":" + cfg.Port); err != nil { log.Fatal(err) diff --git a/backend/cmd/scheduler/main.go b/backend/cmd/scheduler/main.go new file mode 100644 index 0000000..fcaa456 --- /dev/null +++ b/backend/cmd/scheduler/main.go @@ -0,0 +1,25 @@ +package main + +import ( + "context" + "log" + "senlinai-agent/backend/internal/config" + "senlinai-agent/backend/internal/crontab" + "senlinai-agent/backend/internal/logic/dataset" + "senlinai-agent/backend/internal/models" +) + +func main() { + cfg := config.Load() + + err := models.New(cfg.DSN) + if err != nil { + log.Fatal(err) + } + + datasetService := dataset.NewService(models.DBService) + datasetScheduler := crontab.NewDatasetScheduler(datasetService, log.Default()) + schedulerContext, stopScheduler := context.WithCancel(context.Background()) + defer stopScheduler() + go datasetScheduler.Run(schedulerContext) +}