feat: schedule dataset collection

This commit is contained in:
2026-07-23 20:55:38 +08:00
parent 5da343cc71
commit 3749cf4543
5 changed files with 309 additions and 1 deletions

View File

@@ -1,9 +1,11 @@
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"
@@ -29,7 +31,9 @@ func main() {
}
authService := auth.NewService(cfg.AuthSecret)
datasetHandler := dataset.NewHandler(dataset.NewService(models.DBService))
datasetService := dataset.NewService(models.DBService)
datasetHandler := dataset.NewHandler(datasetService)
datasetScheduler := crontab.NewDatasetScheduler(datasetService, log.Default())
projectService := projects.NewService()
fileService := files.NewService(cfg.StorageDir, models.DBService)
taskService := tasks.NewService(models.DBService)
@@ -57,6 +61,10 @@ 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)
}