From 7887d14e61121ffcc197eee0bb661b60779e2564 Mon Sep 17 00:00:00 2001
From: GYJ <1157756119@qq.com>
Date: Wed, 26 Feb 2025 17:37:29 +0800
Subject: [PATCH] bug fix

---
 .../modules/invite/service/impl/InviteServiceImpl.java   | 3 ++-
 src/test/java/com/sqx/Tets.java                          | 9 ++++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

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 b0c5f1b3..2c6e634c 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
@@ -214,9 +214,10 @@ public class InviteServiceImpl extends ServiceImpl<InviteDao, Invite> implements
         user.setUserId(userId);
         user.setInviterCode(userEntity.getInvitationCode());
         user.setInviterUserId(userEntity.getUserId());
-        user.setInviteCount((userEntity.getInviteCount() == null ? 0 : userEntity.getInviteCount()) + 1);
         userService.updateById(user);
 
+        userEntity.setInviteCount((user.getInviteCount() == null ? 0 : user.getInviteCount()) + 1);
+
         // 金币
         int money = Integer.parseInt(commonInfoService.findOne(911).getValue());
         if (money > 0 && userEntity.getUserId() != 1) {
diff --git a/src/test/java/com/sqx/Tets.java b/src/test/java/com/sqx/Tets.java
index 1bff9c0d..c4d974a7 100644
--- a/src/test/java/com/sqx/Tets.java
+++ b/src/test/java/com/sqx/Tets.java
@@ -77,7 +77,11 @@ public class Tets {
     @Test
     public void testUserInviteCount() {
         List<UserEntity> userList = userService.list(Wrappers.emptyWrapper());
-        System.out.println(userList.size());
+
+        int totalSize = userList.size();
+        System.out.println(totalSize);
+
+        AtomicInteger hasExecuteCount = new AtomicInteger();
 
         userList.parallelStream().forEach(item -> {
             try {
@@ -94,6 +98,9 @@ public class Tets {
                 item.setInviteCount(count);
                 userService.updateById(item);
                 log.info("更新邀请人数: {}, {}", item.getUserId(), count);
+
+                hasExecuteCount.getAndIncrement();
+                log.info("已执行: {}/{}", hasExecuteCount, totalSize);
             } catch (Exception e) {
                 log.error("", e);
             }