package cron
|
|
import (
|
"time"
|
|
"github.com/go-co-op/gocron"
|
|
"model-engine/models"
|
"model-engine/pkg/logger"
|
"model-engine/pkg/safe"
|
"model-engine/service"
|
)
|
|
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
|
}
|
|
t := task
|
safe.Go(func() {
|
if err := model.Init(t); err != nil {
|
logger.Warnf("Init task %s. %s", t.Name, err.Error())
|
return
|
}
|
|
model.KeepAlive()
|
|
if err := model.Run(); err != nil {
|
logger.Warnf("Run task %s. %s", t.Name, err.Error())
|
return
|
}
|
if err := model.Shutdown(); err != nil {
|
return
|
}
|
})
|
}
|
}
|
|
func Run() (err error) {
|
_, err = s.Every(10).Minute().StartImmediately().Do(Dispatch)
|
if err != nil {
|
return err
|
}
|
|
s.StartAsync()
|
return nil
|
}
|
