xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-09-20 10:16:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: deducating the user's img call quota after stable diffusion callback

Browse Source
This commit is contained in:
RockYang 2023-11-14 08:59:39 +08:00
parent 00de18be9a
commit 576adc9036
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/service/sd/service.go
View File

@ -288,7 +288,7 @@ func (s *Service) callback(data CBReq) {
jobVo.ImgURL = data.ImageData jobVo.ImgURL = data.ImageData
} }
// 扣减绘图次数 // 扣减绘图次数
s.db.Where("id = ?", jobVo.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1)) s.db.Model(&model.User{}).Where("id = ?", jobVo.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
// 推送任务到前端 // 推送任务到前端
if client != nil { if client != nil {
utils.ReplyChunkMessage(client, jobVo) utils.ReplyChunkMessage(client, jobVo)