task_starter.go 536 B

12345678910111213141516171819202122232425262728
  1. package task
  2. import (
  3. "eta_mini_ht_api/task/base"
  4. _ "eta_mini_ht_api/task/eta"
  5. _ "eta_mini_ht_api/task/sms"
  6. "github.com/beego/beego/v2/server/web"
  7. "github.com/beego/beego/v2/task"
  8. )
  9. func AddTasksToScheduler() {
  10. for _, t := range base.GetTasks() {
  11. if t.RunMode == web.BConfig.RunMode {
  12. tk := task.NewTask(string(t.Name), t.Cron, t.Run)
  13. task.AddTask(string(t.Name), tk)
  14. }
  15. }
  16. }
  17. // StartScheduler 启动任务调度器
  18. func StartScheduler() {
  19. task.StartTask()
  20. }
  21. func init() {
  22. AddTasksToScheduler()
  23. StartScheduler()
  24. }