From 6bbff95334defadbbebd1c25f7217ae23fdaf052 Mon Sep 17 00:00:00 2001
From: wanjia <553039491@qq.com>
Date: Tue, 25 Apr 2023 08:56:58 +0800
Subject: [PATCH] correct number of installations

---
 .gitignore                                             |  1 +
 .../gitlink/softbot/service/user/impl/UserService.java | 10 ++++++++++
 2 files changed, 11 insertions(+)

diff --git a/.gitignore b/.gitignore
index ea8c4bf..3a8cabc 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 /target
+.idea
diff --git a/src/main/java/com/gitlink/softbot/service/user/impl/UserService.java b/src/main/java/com/gitlink/softbot/service/user/impl/UserService.java
index dc38de6..96f65cb 100644
--- a/src/main/java/com/gitlink/softbot/service/user/impl/UserService.java
+++ b/src/main/java/com/gitlink/softbot/service/user/impl/UserService.java
@@ -3,6 +3,7 @@ package com.gitlink.softbot.service.user.impl;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.gitlink.softbot.dao.db.*;
@@ -730,6 +731,7 @@ public class UserService extends AbstractUserBot implements IUserService {
             for (InstallBot installBot : installBotList) {
                 //插入表
                 installMapper.insert(installBot);
+                botInstallNumIncrement(installBot.getBotId());
                 asyncAddWebhookService.asyncAddWebhookAndRetry(installMarketBotRequest.getUserId(), botLimitEvents, installBot);
             }
 
@@ -885,11 +887,19 @@ public class UserService extends AbstractUserBot implements IUserService {
         for (Integer store : storeList){
             InstallBot installBot = getInstallBot(updateInstallBotRequest,store);
             installMapper.insert(installBot);
+            botInstallNumIncrement(installBot.getBotId());
             asyncAddWebhookService.asyncAddWebhookAndRetry(updateInstallBotRequest.getUserId(), botLimitEvents, installBot);
         }
 
     }
 
+    private void botInstallNumIncrement(Integer botId) {
+        UpdateWrapper<Bot> updateWrapper = new UpdateWrapper<>();
+        updateWrapper.lambda().eq(Bot::getId, botId)
+                        .setSql("install_num = install_num + 1");
+        botMapper.update(null, updateWrapper);
+    }
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void deleteInstallBot(DeleteInstallBotRequest deleteInstallBotRequest) throws BotException{