fix: restore user img_calls quota when image task run failed

api/handler/mj_handler.go

@@ -333,6 +333,8 @@ func (h *MidJourneyHandler) JobList(c *gin.Context) {
 			// 10 分钟还没完成的任务直接删除
 			if time.Now().Sub(item.CreatedAt) > time.Minute*10 {
 				h.db.Delete(&item)
+				// 退回绘图次数
+				h.db.Model(&model.User{}).Where("id = ?", item.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 				continue
 			}
 

api/handler/sd_handler.go

@@ -183,6 +183,8 @@ func (h *SdJobHandler) JobList(c *gin.Context) {
 			// 5 分钟还没完成的任务直接删除
 			if time.Now().Sub(item.CreatedAt) > time.Minute*5 {
 				h.db.Delete(&item)
+				// 退回绘图次数
+				h.db.Model(&model.User{}).Where("id = ?", item.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls + ?", 1))
 				continue
 			}
 			// 正在运行中任务使用代理访问图片

web/src/views/ImageMj.vue

@@ -617,6 +617,7 @@ const fetchRunningJobs = (userId) => {
           message: "任务ID：" + jobs[i]['task_id'],
           type: 'error',
         })
+        imgCalls.value += 1
         continue
       }
       _jobs.push(jobs[i])

web/src/views/ImageSd.vue

@@ -596,6 +596,7 @@ onMounted(() => {
             message: "任务ID：" + jobs[i]['task_id'],
             type: 'error',
           })
+          imgCalls.value += 1
           continue
         }
         _jobs.push(jobs[i])