提交

modules/task/funcs/task_add_task.go

@@ -0,0 +1,46 @@
+package funcs
+
+import (
+	"github.com/cool-team-official/cool-admin-go/cool"
+	"github.com/cool-team-official/cool-admin-go/modules/task/model"
+	"github.com/cool-team-official/cool-admin-go/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{})
+}

modules/task/funcs/task_stop_func.go

@@ -0,0 +1,33 @@
+package funcs
+
+import (
+	"github.com/cool-team-official/cool-admin-go/cool"
+	"github.com/cool-team-official/cool-admin-go/modules/task/model"
+	"github.com/cool-team-official/cool-admin-go/modules/task/service"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type TaskStopFunc struct {
+}
+
+func (t *TaskStopFunc) Func(ctx g.Ctx, id string) error {
+	taskInfo := model.NewTaskInfo()
+	_, err := cool.DBM(taskInfo).Where("id = ?", id).Update(g.Map{"status": 0})
+	if err != nil {
+		return err
+	}
+
+	return service.DisableTask(ctx, id)
+}
+
+func (t *TaskStopFunc) IsSingleton() bool {
+	return false
+}
+
+func (t *TaskStopFunc) IsAllWorker() bool {
+	return true
+}
+
+func init() {
+	cool.RegisterFunc("TaskStopFunc", &TaskStopFunc{})
+}

modules/task/funcs/task_testfunc.go

@@ -0,0 +1,24 @@
+package funcs
+
+import (
+	"github.com/cool-team-official/cool-admin-go/cool"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type TaskTest struct {
+}
+
+func (t *TaskTest) Func(ctx g.Ctx, param string) error {
+	g.Log().Info(ctx, "TaskTest Run ~~~~~~~~~~~~~~~~", param)
+	return nil
+}
+func (t *TaskTest) IsSingleton() bool {
+	return false
+}
+func (t *TaskTest) IsAllWorker() bool {
+	return true
+}
+
+func init() {
+	cool.RegisterFunc("TaskTest", &TaskTest{})
+}

modules/task/funcs/tast_start_func.go

@@ -0,0 +1,49 @@
+package funcs
+
+import (
+	"github.com/cool-team-official/cool-admin-go/cool"
+	"github.com/cool-team-official/cool-admin-go/modules/task/model"
+	"github.com/cool-team-official/cool-admin-go/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{})
+}