|
@@ -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 启动任务调度器
|