浏览代码

任务修改为生产运行

kobe6258 7 月之前
父节点
当前提交
8ad27401e3
共有 1 个文件被更改,包括 16 次插入1 次删除
  1. 16 1
      task/task_starter.go

+ 16 - 1
task/task_starter.go

@@ -14,12 +14,27 @@ import (
 
 func AddTasksToScheduler() {
 	for _, t := range base.GetTasks() {
-		if t.RunMode == web.BConfig.RunMode {
+		//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() {