From 70d66864a3f686cd52b7eba173aa5b03dfb4ea7c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=9D=BE?=
 <8605635+zhang3064194730@user.noreply.gitee.com>
Date: Tue, 31 Dec 2024 11:29:21 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AD=BE=E5=88=B0=E4=BA=BA=E6=95=B0=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../invite/controller/app/AppInviteController.java   | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/sqx/modules/invite/controller/app/AppInviteController.java b/src/main/java/com/sqx/modules/invite/controller/app/AppInviteController.java
index 61b81a77..e6ccf197 100644
--- a/src/main/java/com/sqx/modules/invite/controller/app/AppInviteController.java
+++ b/src/main/java/com/sqx/modules/invite/controller/app/AppInviteController.java
@@ -4,13 +4,16 @@ package com.sqx.modules.invite.controller.app;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.extra.qrcode.QrCodeUtil;
 import cn.hutool.extra.qrcode.QrConfig;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.sqx.common.utils.PageUtils;
 import com.sqx.common.utils.Result;
 import com.sqx.modules.app.annotation.Login;
 import com.sqx.modules.app.annotation.LoginUser;
+import com.sqx.modules.app.entity.InviteAchievement;
 import com.sqx.modules.app.entity.UserEntity;
+import com.sqx.modules.app.service.InviteAchievementService;
 import com.sqx.modules.app.service.UserMoneyDetailsService;
 import com.sqx.modules.app.service.UserMoneyService;
 import com.sqx.modules.app.service.UserService;
@@ -65,6 +68,11 @@ public class AppInviteController {
     private QrConfig qrconig;
     @Autowired
     private UrlAddressService urlAddressService;
+    private final InviteAchievementService inviteAchievementService;
+
+    public AppInviteController(InviteAchievementService inviteAchievementService) {
+        this.inviteAchievementService = inviteAchievementService;
+    }
 
     @RequestMapping(value = "/selectInviteCount", method = RequestMethod.GET)
     @ApiOperation("查看我邀请的人员数量")
@@ -105,7 +113,9 @@ public class AppInviteController {
         }
         InviteMoney inviteMoney = inviteMoneyService.selectInviteMoneyByUserId(userId);
         Integer inviteCount = inviteService.selectInviteCount2(userEntity.getInvitationCode());
-        Integer inviteSignCount = inviteService.selectInviteSignCount(userEntity.getInvitationCode());
+//        Integer inviteSignCount = inviteService.selectInviteSignCount(userEntity.getInvitationCode());
+        int inviteSignCount = inviteAchievementService.count(new LambdaQueryWrapper<InviteAchievement>().eq(InviteAchievement::getState, 1)
+                .eq(InviteAchievement::getSourceUserId, userId));
         Map<String, Double> earningMap = userMoneyDetailsService.queryUserTotalEarning(userId);
         Map<String, Object> result = new HashMap<>();
         result.put("inviteMoney", inviteMoney);