52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package funcs
|
|
|
|
import (
|
|
"blazing/cool"
|
|
|
|
"blazing/modules/task/model"
|
|
"blazing/modules/task/service"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
type TaskStartFunc struct {
|
|
}
|
|
|
|
func (t *TaskStartFunc) Func(ctx g.Ctx, id string) error {
|
|
taskInfo := model.NewTaskInfo()
|
|
_, err := cool.DBM(taskInfo).Where("id = ?", id).Update(g.Map{"status": 1})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
result, err := cool.DBM(taskInfo).Where("id = ?", id).One()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if result["taskType"].Int() == 1 {
|
|
every := result["every"].Uint() / 1000
|
|
cron := "@every " + gconv.String(every) + "s"
|
|
funcstring := result["service"].String()
|
|
startDate := result["startDate"].String()
|
|
err = service.EnableTask(ctx, id, funcstring, cron, startDate)
|
|
|
|
} else {
|
|
cron := result["cron"].String()
|
|
funcstring := result["service"].String()
|
|
startDate := result["startDate"].String()
|
|
err = service.EnableTask(ctx, id, funcstring, cron, startDate)
|
|
}
|
|
return err
|
|
|
|
}
|
|
func (t *TaskStartFunc) IsSingleton() bool {
|
|
return false
|
|
}
|
|
func (t *TaskStartFunc) IsAllWorker() bool {
|
|
return true
|
|
}
|
|
|
|
func init() {
|
|
cool.RegisterFunc("TaskStartFunc", &TaskStartFunc{})
|
|
}
|