refactor(backend): run dataset scheduler separately
This commit is contained in:
@@ -1,11 +1,9 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"log"
|
"log"
|
||||||
|
|
||||||
"senlinai-agent/backend/internal/config"
|
"senlinai-agent/backend/internal/config"
|
||||||
"senlinai-agent/backend/internal/crontab"
|
|
||||||
"senlinai-agent/backend/internal/httpx"
|
"senlinai-agent/backend/internal/httpx"
|
||||||
"senlinai-agent/backend/internal/initdb"
|
"senlinai-agent/backend/internal/initdb"
|
||||||
"senlinai-agent/backend/internal/logic/ai"
|
"senlinai-agent/backend/internal/logic/ai"
|
||||||
@@ -33,7 +31,6 @@ func main() {
|
|||||||
authService := auth.NewService(cfg.AuthSecret)
|
authService := auth.NewService(cfg.AuthSecret)
|
||||||
datasetService := dataset.NewService(models.DBService)
|
datasetService := dataset.NewService(models.DBService)
|
||||||
datasetHandler := dataset.NewHandler(datasetService)
|
datasetHandler := dataset.NewHandler(datasetService)
|
||||||
datasetScheduler := crontab.NewDatasetScheduler(datasetService, log.Default())
|
|
||||||
projectService := projects.NewService()
|
projectService := projects.NewService()
|
||||||
documentService := documents.NewService(models.DBService, cfg.StorageDir)
|
documentService := documents.NewService(models.DBService, cfg.StorageDir)
|
||||||
taskService := tasks.NewService(models.DBService)
|
taskService := tasks.NewService(models.DBService)
|
||||||
@@ -63,9 +60,6 @@ func main() {
|
|||||||
searchHandler,
|
searchHandler,
|
||||||
aiHandler,
|
aiHandler,
|
||||||
)
|
)
|
||||||
schedulerContext, stopScheduler := context.WithCancel(context.Background())
|
|
||||||
defer stopScheduler()
|
|
||||||
go datasetScheduler.Run(schedulerContext)
|
|
||||||
|
|
||||||
if err := appRouter.Run(":" + cfg.Port); err != nil {
|
if err := appRouter.Run(":" + cfg.Port); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
|
|||||||
25
backend/cmd/scheduler/main.go
Normal file
25
backend/cmd/scheduler/main.go
Normal 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)
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user