eta_mini_ht_api/task/task_starter.go

package task

import (
	"eta/eta_mini_ht_api/common/component/config"
	"eta/eta_mini_ht_api/common/contants"
	"eta/eta_mini_ht_api/task/base"
	_ "eta/eta_mini_ht_api/task/eta/author"
	_ "eta/eta_mini_ht_api/task/message"
	_ "eta/eta_mini_ht_api/task/report"
	_ "eta/eta_mini_ht_api/task/sms"
	"github.com/beego/beego/v2/server/web"
	"github.com/beego/beego/v2/task"
)

func AddTasksToScheduler() {
	for _, t := range base.GetTasks() {
		//if t.RunMode == web.BConfig.RunMode {
		if needAddTask(t.RunMode, web.BConfig.RunMode) {
			tk := task.NewTask(string(t.Name), t.Cron, t.Run)
			task.AddTask(string(t.Name), tk)
		}
	}
}
func needAddTask(taskRunMode string, runMode string) bool {
	switch taskRunMode {
	case base.PROD:
		return runMode == base.PROD
	case base.TEST:
		return runMode != base.PROD
	case base.DEV:
		return runMode == base.DEV
	case base.FORBIDDEN:
		return false
	default:
		return false
	}
}

// StartScheduler 启动任务调度器
func StartScheduler() {
	htConfig := config.GetConfig(contants.HT).(*config.HTBizConfig)
	if htConfig.EnableTask() {
		task.StartTask()
	}
}

func init() {
	AddTasksToScheduler()
	StartScheduler()
}