增加上游构图失败判断

175ac5b8 · 余生一个白恩 · 130d2751 · 175ac5b8
Commit 175ac5b8 authored Mar 30, 2024 by 余生一个白恩
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

controller/midjourney.go
+3 -2

No files found.
--- a/controller/midjourney.go
+++ b/controller/midjourney.go
@@ -5,7 +5,6 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"github.com/gin-gonic/gin"
 	"io"
 	"log"
 	"net/http"
@@ -16,6 +15,8 @@ import (
 	"strconv"
 	"strings"
 	"time"
+
+	"github.com/gin-gonic/gin"
 )

 func UpdateMidjourneyTaskBulk() {
@@ -147,7 +148,7 @@ func UpdateMidjourneyTaskBulk() {
 					task.Buttons = string(buttonStr)
 				}

-				if task.Progress != "100%" && responseItem.FailReason != "" {
+				if task.Progress != "100%" && responseItem.FailReason != "" || task.Progress == "100%" && task.Status == "FAILURE" {
 					common.LogInfo(ctx, task.MjId+" 构建失败，"+task.FailReason)
 					task.Progress = "100%"
 					err = model.CacheUpdateUserQuota(task.UserId)