From a47111a03188499f8e5a48bcced58c3ce0174ac1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=98=B3=E5=85=89=E9=87=8C=E6=B5=B7=E7=9A=84=E5=A3=B0?=
 =?UTF-8?q?=E9=9F=B3?= <492212381@qq.com>
Date: Fri, 7 Jun 2024 00:16:54 +0800
Subject: [PATCH] =?UTF-8?q?Midjourney=20Proxy=20Plus=E6=97=A0=E5=AE=9E?=
 =?UTF-8?q?=E4=BE=8B=E8=B4=A6=E5=8F=B7=E8=87=AA=E5=8A=A8=E7=A6=81=E7=94=A8?=
 =?UTF-8?q?=E8=AF=A5=E6=B8=A0=E9=81=93?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 relay/relay-mj.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/relay/relay-mj.go b/relay/relay-mj.go
index f06956a..f4d38e7 100644
--- a/relay/relay-mj.go
+++ b/relay/relay-mj.go
@@ -538,7 +538,16 @@ func RelayMidjourneySubmit(c *gin.Context, relayMode int) *dto.MidjourneyRespons
 		ChannelId:   c.GetInt("channel_id"),
 		Quota:       quota,
 	}
-
+	if midjResponse.Code == 3 {
+		//无实例账号自动禁用渠道（No available account instance）
+		channel, err := model.GetChannelById(midjourneyTask.ChannelId, true)
+		if err != nil {
+			common.SysError("get_channel_null: " + err.Error())
+		}
+		if channel.AutoBan != nil && *channel.AutoBan == 1 {
+			model.UpdateChannelStatusById(midjourneyTask.ChannelId, 2)
+		}
+	}
 	if midjResponse.Code != 1 && midjResponse.Code != 21 && midjResponse.Code != 22 {
 		//非1-提交成功,21-任务已存在和22-排队中，则记录错误原因
 		midjourneyTask.FailReason = midjResponse.Description