Browse Source

任务修改为生产运行

kobe6258 7 months ago
parent
commit
a8c799a289
1 changed files with 16 additions and 9 deletions
  1. 16 9
      task/task_starter.go

+ 16 - 9
task/task_starter.go

@@ -22,18 +22,25 @@ func AddTasksToScheduler() {
 	}
 }
 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:
+	if taskRunMode == base.PROD {
+		return true
+	}
+	if taskRunMode == base.TEST {
+		if runMode == base.PROD {
+			return false
+		}
+		return true
+	}
+	if taskRunMode == base.DEV {
+		if runMode == base.DEV {
+			return true
+		}
 		return false
-	default:
+	}
+	if taskRunMode == base.FORBIDDEN {
 		return false
 	}
+	return false
 }
 
 // StartScheduler 启动任务调度器