From 576adc90362c1025af22cc449b6682e8c5a0368b Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Tue, 14 Nov 2023 08:59:39 +0800
Subject: [PATCH] fix: deducating the user's img call quota after stable
 diffusion callback

---
 api/service/sd/service.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/service/sd/service.go b/api/service/sd/service.go
index a0ddadf3..a6133a55 100644
--- a/api/service/sd/service.go
+++ b/api/service/sd/service.go
@@ -288,7 +288,7 @@ func (s *Service) callback(data CBReq) {
 			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 {
 			utils.ReplyChunkMessage(client, jobVo)