xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-18 00:16:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: only update task on master node

Browse Source
This commit is contained in:
CalciumIon 2024-06-26 17:23:03 +08:00
parent 11be36dafd
commit cadd8aa622
3 changed files with 1798 additions and 1506 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
controller/midjourney.go
View File

@ -19,9 +19,6 @@ import (
) )
func UpdateMidjourneyTaskBulk() { func UpdateMidjourneyTaskBulk() {
if !common.IsMasterNode {
return
}
//imageModel := "midjourney" //imageModel := "midjourney"
ctx := context.TODO() ctx := context.TODO()
for { for {

2
main.go
View File

@ -89,12 +89,14 @@ func main() {
} }
go controller.AutomaticallyTestChannels(frequency) go controller.AutomaticallyTestChannels(frequency)
} }
if common.IsMasterNode {
common.SafeGoroutine(func() { common.SafeGoroutine(func() {
controller.UpdateMidjourneyTaskBulk() controller.UpdateMidjourneyTaskBulk()
}) })
common.SafeGoroutine(func() { common.SafeGoroutine(func() {
controller.UpdateTaskBulk() controller.UpdateTaskBulk()
}) })
}
if os.Getenv("BATCH_UPDATE_ENABLED") == "true" { if os.Getenv("BATCH_UPDATE_ENABLED") == "true" {
common.BatchUpdateEnabled = true common.BatchUpdateEnabled = true
common.SysLog("batch update enabled with interval " + strconv.Itoa(common.BatchUpdateInterval) + "s") common.SysLog("batch update enabled with interval " + strconv.Itoa(common.BatchUpdateInterval) + "s")

3141
web/pnpm-lock.yaml
View File

File diff suppressed because it is too large Load Diff