refactor(backend): run dataset scheduler separately

This commit is contained in:
2026-07-25 20:59:24 +08:00
parent b2c22901ba
commit 7534fadfca
2 changed files with 25 additions and 6 deletions

View File

@@ -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)

View File

@@ -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)
}