diff --git a/api/handler/mj_handler.go b/api/handler/mj_handler.go index ad946e14..06f92b35 100644 --- a/api/handler/mj_handler.go +++ b/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 } diff --git a/api/handler/sd_handler.go b/api/handler/sd_handler.go index b5515f9f..4e22ba56 100644 --- a/api/handler/sd_handler.go +++ b/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 } // 正在运行中任务使用代理访问图片 diff --git a/web/src/views/ImageMj.vue b/web/src/views/ImageMj.vue index d7993788..0475fa7a 100644 --- a/web/src/views/ImageMj.vue +++ b/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]) diff --git a/web/src/views/ImageSd.vue b/web/src/views/ImageSd.vue index e1a46f61..2d57ce91 100644 --- a/web/src/views/ImageSd.vue +++ b/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])