From 70404d83e893ef2b562e7d2d84d7606b57be7a0c Mon Sep 17 00:00:00 2001
From: GYJ <1157756119@qq.com>
Date: Sat, 28 Dec 2024 14:37:20 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E9=82=80=E8=AF=B7=E4=BA=BA?=
 =?UTF-8?q?=E6=95=B0=20bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/sqx/modules/invite/dao/InviteDao.java      | 1 +
 .../java/com/sqx/modules/invite/service/InviteService.java   | 1 +
 .../sqx/modules/invite/service/impl/InviteServiceImpl.java   | 5 +++++
 src/main/resources/mapper/invite/InviteDao.xml               | 5 +++++
 4 files changed, 12 insertions(+)

diff --git a/src/main/java/com/sqx/modules/invite/dao/InviteDao.java b/src/main/java/com/sqx/modules/invite/dao/InviteDao.java
index 6fcd88a6..71544fe3 100644
--- a/src/main/java/com/sqx/modules/invite/dao/InviteDao.java
+++ b/src/main/java/com/sqx/modules/invite/dao/InviteDao.java
@@ -20,6 +20,7 @@ public interface InviteDao extends BaseMapper<Invite> {
     IPage<Invite> selectInviteList(Page<Map<String,Object>> page, @Param("state") Integer state, @Param("userId") Long userId);
 
     Integer selectInviteCount(@Param("state") Integer state, @Param("userId") Long userId);
+    Integer selectInviteCount2(@Param("code") String code);
     Integer selectInviteSignCount(@Param("code") String code);
 
     Double selectInviteSum(@Param("state") Integer state, @Param("userId") Long userId);
diff --git a/src/main/java/com/sqx/modules/invite/service/InviteService.java b/src/main/java/com/sqx/modules/invite/service/InviteService.java
index ed4780bb..46b31db0 100644
--- a/src/main/java/com/sqx/modules/invite/service/InviteService.java
+++ b/src/main/java/com/sqx/modules/invite/service/InviteService.java
@@ -15,6 +15,7 @@ public interface InviteService {
     PageUtils selectInviteList(int page, int limit, Integer state, Long userId);
 
     Integer selectInviteCount(Integer state,Long userId);
+    Integer selectInviteCount2(String code);
     Integer selectInviteSignCount(String code);
 
     Double selectInviteSum(Integer state,Long userId);
diff --git a/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java b/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java
index 48bfac8e..9f21bd37 100644
--- a/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java
+++ b/src/main/java/com/sqx/modules/invite/service/impl/InviteServiceImpl.java
@@ -112,6 +112,11 @@ public class InviteServiceImpl extends ServiceImpl<InviteDao, Invite> implements
         return inviteDao.selectInviteCount(state, userId);
     }
 
+    @Override
+    public Integer selectInviteCount2(String code) {
+        return inviteDao.selectInviteCount2(code);
+    }
+
     @Override
     public Integer selectInviteSignCount(String code) {
         return inviteDao.selectInviteSignCount(code);
diff --git a/src/main/resources/mapper/invite/InviteDao.xml b/src/main/resources/mapper/invite/InviteDao.xml
index d62025da..ae7d9162 100644
--- a/src/main/resources/mapper/invite/InviteDao.xml
+++ b/src/main/resources/mapper/invite/InviteDao.xml
@@ -51,6 +51,11 @@
         </if>
     </select>
 
+    <select id="selectInviteCount2" resultType="Integer">
+        select count(*) from tb_user tu
+        where tu.inviter_code = #{code}
+    </select>
+
     <select id="selectInviteSignCount" resultType="Integer">
         SELECT
             count(*)