diff --git a/common/init.go b/common/init.go index 715bee99..3f3bd1df 100644 --- a/common/init.go +++ b/common/init.go @@ -131,6 +131,8 @@ func initConstantEnv() { constant.GenerateDefaultToken = GetEnvOrDefaultBool("GENERATE_DEFAULT_TOKEN", false) // 是否启用错误日志 constant.ErrorLogEnabled = GetEnvOrDefaultBool("ERROR_LOG_ENABLED", false) + // 任务轮询时查询的最大数量 + constant.TaskQueryLimit = GetEnvOrDefault("TASK_QUERY_LIMIT", 1000) soraPatchStr := GetEnvOrDefaultString("TASK_PRICE_PATCH", "") if soraPatchStr != "" { diff --git a/constant/env.go b/constant/env.go index 94c2cfcd..975bced7 100644 --- a/constant/env.go +++ b/constant/env.go @@ -15,6 +15,7 @@ var NotifyLimitCount int var NotificationLimitDurationMinute int var GenerateDefaultToken bool var ErrorLogEnabled bool +var TaskQueryLimit int // temporary variable for sora patch, will be removed in future var TaskPricePatches []string diff --git a/controller/task.go b/controller/task.go index c14d7e21..ad034d61 100644 --- a/controller/task.go +++ b/controller/task.go @@ -29,7 +29,7 @@ func UpdateTaskBulk() { time.Sleep(time.Duration(15) * time.Second) common.SysLog("任务进度轮询开始") ctx := context.TODO() - allTasks := model.GetAllUnFinishSyncTasks(500) + allTasks := model.GetAllUnFinishSyncTasks(constant.TaskQueryLimit) platformTask := make(map[constant.TaskPlatform][]*model.Task) for _, t := range allTasks { platformTask[t.Platform] = append(platformTask[t.Platform], t)