package cron
|
|
import (
|
"github.com/go-co-op/gocron"
|
"model-engine/models"
|
"model-engine/pkg/logger"
|
"model-engine/pkg/safe"
|
"model-engine/service"
|
"time"
|
)
|
|
var s *gocron.Scheduler
|
|
func init() {
|
s = gocron.NewScheduler(time.Local)
|
}
|
|
func Dispatch() {
|
tasks, err := service.GetTasks()
|
if err != nil {
|
panic(err)
|
}
|
for _, task := range tasks {
|
model, err := models.GetModel(task.ModelID)
|
if err != nil {
|
logger.Errorf("can not find model for id:%v", task.ModelID)
|
continue
|
}
|
task := task
|
safe.Go(func() {
|
if err := model.Init(task); err != nil {
|
return
|
}
|
if err := model.Run(); err != nil {
|
return
|
}
|
if err := model.Shutdown(); err != nil {
|
return
|
}
|
})
|
}
|
}
|
|
func Run() (err error) {
|
_, err = s.Every(1).Minute().StartImmediately().Do(Dispatch)
|
if err != nil {
|
return err
|
}
|
|
s.StartAsync()
|
return nil
|
}
|
