From 2525a22d78274a7b91bfdf2dc51cad504b0548cf Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Sat, 11 Nov 2023 11:00:26 +0800
Subject: [PATCH] fix: fix bug for issue 49, stable diffusion service not
 decrease img_calls

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

diff --git a/api/service/sd/service.go b/api/service/sd/service.go
index e1f88920..a0ddadf3 100644
--- a/api/service/sd/service.go
+++ b/api/service/sd/service.go
@@ -287,7 +287,8 @@ func (s *Service) callback(data CBReq) {
 		if data.Progress < 100 && data.ImageData != "" {
 			jobVo.ImgURL = data.ImageData
 		}
-
+		// 扣减绘图次数
+		s.db.Where("id = ?", jobVo.UserId).UpdateColumn("img_calls", gorm.Expr("img_calls - ?", 1))
 		// 推送任务到前端
 		if client != nil {
 			utils.ReplyChunkMessage(client, jobVo)