From 1d599e2e60de9b7e8d849f0f90c780f9fdfbe48f Mon Sep 17 00:00:00 2001
From: luodangjia <luodangjia>
Date: 星期三, 15 一月 2025 15:37:05 +0800
Subject: [PATCH] 12.18

---
 ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java |  101 +++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 83 insertions(+), 18 deletions(-)

diff --git a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java
index d7472d4..834ecc6 100644
--- a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java
+++ b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/controller/AppUserController.java
@@ -272,15 +272,54 @@
             user.setTopUser(byId);
         }
         //获取绑定下级列表
-        List<AppUser> list = appUserService.lambdaQuery().eq(AppUser::getInviteUserId, user.getId()).list();
+        List<AppUser> list = appUserService.lambdaQuery()
+                .eq(AppUser::getInviteUserId, user.getId())
+                .eq(AppUser::getDelFlag, 0)
+                .ne(AppUser::getStatus, 3)
+                .list();
         for (AppUser appUser : list) {
-            Long count1 = appUserService.lambdaQuery().eq(AppUser::getVipId, 1).eq(AppUser::getInviteUserId, appUser.getId()).count();
-            Long count2 = appUserService.lambdaQuery().eq(AppUser::getVipId, 2).eq(AppUser::getInviteUserId, appUser.getId()).count();
-            Long count3 = appUserService.lambdaQuery().eq(AppUser::getVipId, 3).eq(AppUser::getInviteUserId, appUser.getId()).count();
-            Long count4 = appUserService.lambdaQuery().eq(AppUser::getVipId, 4).eq(AppUser::getInviteUserId, appUser.getId()).count();
-            Long count5 = appUserService.lambdaQuery().eq(AppUser::getVipId, 5).eq(AppUser::getInviteUserId, appUser.getId()).count();
-            Long count6 = appUserService.lambdaQuery().eq(AppUser::getVipId, 6).eq(AppUser::getInviteUserId, appUser.getId()).count();
-            Long count7 = appUserService.lambdaQuery().eq(AppUser::getVipId, 7).eq(AppUser::getInviteUserId, appUser.getId()).count();
+            Long count1 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 1)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
+            Long count2 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 2)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
+            Long count3 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 3)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
+            Long count4 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 4)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
+            Long count5 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 5)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
+            Long count6 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 6)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
+            Long count7 = appUserService.lambdaQuery()
+                    .eq(AppUser::getVipId, 7)
+                    .eq(AppUser::getInviteUserId, appUser.getId())
+                    .eq(AppUser::getDelFlag, 0)
+                    .ne(AppUser::getStatus, 3)
+                    .count();
             appUser.setCount1(count1);
             appUser.setCount2(count2);
             appUser.setCount3(count3);
@@ -330,13 +369,19 @@
                 user.setShopName(shop.getName());
             }
         }
-        Long count1 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 1).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
-        Long count2 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 2).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
-        Long count3 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 3).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
-        Long count4 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 4).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
-        Long count5 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 5).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
-        Long count6 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 6).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
-        Long count7 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 7).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+        List<AppUser> appUserList = appUserService.list(new LambdaQueryWrapper<AppUser>()
+                .ne(AppUser::getStatus, 3)
+                .eq(AppUser::getDelFlag, 0));
+        ArrayList<Long> userIds = new ArrayList<>();
+        userIds.add(userId);
+        getUserAncestorList(userIds, appUserList);
+//        Long count1 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 1).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+//        Long count2 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 2).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+//        Long count3 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 3).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+//        Long count4 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 4).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+//        Long count5 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 5).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+//        Long count6 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 6).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
+//        Long count7 = appUserService.lambdaQuery().ne(AppUser::getStatus,3).eq(AppUser::getVipId, 7).eq(AppUser::getInviteUserId, userId).or().eq(AppUser::getTopInviteId,userId).count();
         user.setCount1(count1);
         user.setCount2(count2);
         user.setCount3(count3);
@@ -345,15 +390,35 @@
         user.setCount6(count6);
         user.setCount7(count7);
 
-        List<UserSignRecord> list = userSignRecordService.lambdaQuery().eq(UserSignRecord::getSignDay, LocalDate.now()).list();
+        List<UserSignRecord> list = userSignRecordService.lambdaQuery().eq(UserSignRecord::getSignDay, LocalDate.now())
+                .eq(UserSignRecord::getAppUserId,userId).list();
         if (!list.isEmpty()) {
             user.setIsSign(1);
         } else {
             user.setIsSign(0);
         }
         return R.ok(user);
-
-
+    }
+    private Long count1 = 0L;
+    private Long count2 = 0L;
+    private Long count3 = 0L;
+    private Long count4 = 0L;
+    private Long count5 = 0L;
+    private Long count6 = 0L;
+    private Long count7 = 0L;
+    public void getUserAncestorList(List<Long> userIds, List<AppUser> list) {
+        List<AppUser> children = list.stream().filter(u -> userIds.contains(u.getInviteUserId()) || userIds.contains(u.getTopInviteId())).collect(Collectors.toList());
+        if(!CollectionUtils.isEmpty(children)){
+            count1 = count1 + children.stream().filter(e->e.getVipId() == 1).count();
+            count2 = count2 + children.stream().filter(e->e.getVipId() == 2).count();
+            count3 = count3 + children.stream().filter(e->e.getVipId() == 3).count();
+            count4 = count4 + children.stream().filter(e->e.getVipId() == 4).count();
+            count5 = count5 + children.stream().filter(e->e.getVipId() == 5).count();
+            count6 = count6 + children.stream().filter(e->e.getVipId() == 6).count();
+            count7 = count7 + children.stream().filter(e->e.getVipId() == 7).count();
+            List<Long> userIdList = children.stream().map(AppUser::getId).collect(Collectors.toList());
+            getUserAncestorList(userIdList, children);
+        }
     }
 
     @GetMapping("/index/change")

--
Gitblit v1.7.1