github/new-api
1
0
Fork 0
mirror of https://github.com/QuantumNous/new-api.git synced 2026-04-19 11:48:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 将任务查询数量改为可配置环境变量 TASK_QUERY_LIMIT

Browse Source
This commit is contained in:
CaIon
2025-12-03 19:27:15 +08:00
parent 7d1bad1b37
commit 0b9f6a58bc
3 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
common/init.go
View File

@@ -131,6 +131,8 @@ func initConstantEnv() {
constant.GenerateDefaultToken = GetEnvOrDefaultBool("GENERATE_DEFAULT_TOKEN", false) constant.GenerateDefaultToken = GetEnvOrDefaultBool("GENERATE_DEFAULT_TOKEN", false)
// 是否启用错误日志 // 是否启用错误日志
constant.ErrorLogEnabled = GetEnvOrDefaultBool("ERROR_LOG_ENABLED", false) constant.ErrorLogEnabled = GetEnvOrDefaultBool("ERROR_LOG_ENABLED", false)
// 任务轮询时查询的最大数量
constant.TaskQueryLimit = GetEnvOrDefault("TASK_QUERY_LIMIT", 1000)
soraPatchStr := GetEnvOrDefaultString("TASK_PRICE_PATCH", "") soraPatchStr := GetEnvOrDefaultString("TASK_PRICE_PATCH", "")
if soraPatchStr != "" { if soraPatchStr != "" {

1
constant/env.go
View File

@@ -15,6 +15,7 @@ var NotifyLimitCount int
var NotificationLimitDurationMinute int var NotificationLimitDurationMinute int
var GenerateDefaultToken bool var GenerateDefaultToken bool
var ErrorLogEnabled bool var ErrorLogEnabled bool
var TaskQueryLimit int
// temporary variable for sora patch, will be removed in future // temporary variable for sora patch, will be removed in future
var TaskPricePatches []string var TaskPricePatches []string

2
controller/task.go
View File

@@ -29,7 +29,7 @@ func UpdateTaskBulk() {
time.Sleep(time.Duration(15) * time.Second) time.Sleep(time.Duration(15) * time.Second)
common.SysLog("任务进度轮询开始") common.SysLog("任务进度轮询开始")
ctx := context.TODO() ctx := context.TODO()
allTasks := model.GetAllUnFinishSyncTasks(500) allTasks := model.GetAllUnFinishSyncTasks(constant.TaskQueryLimit)
platformTask := make(map[constant.TaskPlatform][]*model.Task) platformTask := make(map[constant.TaskPlatform][]*model.Task)
for _, t := range allTasks { for _, t := range allTasks {
platformTask[t.Platform] = append(platformTask[t.Platform], t) platformTask[t.Platform] = append(platformTask[t.Platform], t)