fix: restore user img_calls quota when image task run failed

This commit is contained in:
RockYang
2023-12-29 10:41:29 +08:00
parent 816e4a096e
commit 70c46d098b
4 changed files with 6 additions and 0 deletions

View File

@@ -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
}