From 14bf865034aa17bf1c3a8c778acac7d83492cb4b Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Thu, 18 Jul 2024 17:26:21 +0800
Subject: [PATCH] feat: add UPDATE_TASK env

---
 README.md       | 1 +
 constant/env.go | 2 ++
 main.go         | 3 ++-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index ffc0e5a..d2c805c 100644
--- a/README.md
+++ b/README.md
@@ -76,6 +76,7 @@
 - `FORCE_STREAM_OPTION`：是否覆盖客户端stream_options参数，请求上游返回流模式usage，默认为 `true`
 - `GET_MEDIA_TOKEN`：是统计图片token，默认为 `true`，关闭后将不再在本地计算图片token，可能会导致和上游计费不同，此项覆盖 `GET_MEDIA_TOKEN_NOT_STREAM` 选项作用，
 - `GET_MEDIA_TOKEN_NOT_STREAM`：是否在非流（`stream=false`）情况下统计图片token，默认为 `true`
+- `UPDATE_TASK`：是否更新异步任务（Midjourney、Suno），默认为 `true`，关闭后将不会更新任务进度
 ## 部署
 ### 部署要求
 - 本地数据库（默认）：SQLite（Docker 部署默认使用 SQLite，必须挂载 `/data` 目录到宿主机）
diff --git a/constant/env.go b/constant/env.go
index a18b875..76146ca 100644
--- a/constant/env.go
+++ b/constant/env.go
@@ -13,3 +13,5 @@ var ForceStreamOption = common.GetEnvOrDefaultBool("FORCE_STREAM_OPTION", true)
 var GetMediaToken = common.GetEnvOrDefaultBool("GET_MEDIA_TOKEN", true)
 
 var GetMediaTokenNotStream = common.GetEnvOrDefaultBool("GET_MEDIA_TOKEN_NOT_STREAM", true)
+
+var UpdateTask = common.GetEnvOrDefaultBool("UPDATE_TASK", true)
diff --git a/main.go b/main.go
index e929e0c..ed2ab2e 100644
--- a/main.go
+++ b/main.go
@@ -9,6 +9,7 @@ import (
 	"log"
 	"net/http"
 	"one-api/common"
+	"one-api/constant"
 	"one-api/controller"
 	"one-api/middleware"
 	"one-api/model"
@@ -89,7 +90,7 @@ func main() {
 		}
 		go controller.AutomaticallyTestChannels(frequency)
 	}
-	if common.IsMasterNode {
+	if common.IsMasterNode && constant.UpdateTask {
 		common.SafeGoroutine(func() {
 			controller.UpdateMidjourneyTaskBulk()
 		})