昔念
d71b1dd169
- 在 base 模块初始化时添加重置所有序列的函数
- 修改多个模块中的查询语句，使用 Where(key, value) 替代 Where("key = ?", value)
- 优化部分代码结构，提高可读性和维护性
49 lines
1.0 KiB
Go
49 lines
1.0 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 TaskAddTask struct {
|
|
}
|
|
|
|
func (t *TaskAddTask) Func(ctx g.Ctx, id string) error {
|
|
taskInfo := model.NewTaskInfo()
|
|
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 *TaskAddTask) IsSingleton() bool {
|
|
return false
|
|
}
|
|
|
|
func (t *TaskAddTask) IsAllWorker() bool {
|
|
return true
|
|
}
|
|
|
|
func init() {
|
|
cool.RegisterFunc("TaskAddTask", &TaskAddTask{})
|
|
}
|