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{})
}