mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-11-07 09:43:43 +08:00
fix: restore user img_calls quota when image task run failed
This commit is contained in:
@@ -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
|
||||
}
|
||||
// 正在运行中任务使用代理访问图片
|
||||
|
||||
Reference in New Issue
Block a user