From 65ece4e9e659ba7d0be699a64a4ca016d37a637b Mon Sep 17 00:00:00 2001 From: 无关风月 <443237572@qq.com> Date: 星期三, 15 一月 2025 14:43:14 +0800 Subject: [PATCH] Merge branch 'master' of http://120.76.84.145:10101/gitblit/r/java/qijisheng --- ruoyi-service/ruoyi-account/src/test/java/com/ruoyi/order/account/RuoYiAccountApplicationTests.java | 55 ++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 54 insertions(+), 1 deletions(-) diff --git a/ruoyi-service/ruoyi-account/src/test/java/com/ruoyi/order/account/RuoYiAccountApplicationTests.java b/ruoyi-service/ruoyi-account/src/test/java/com/ruoyi/order/account/RuoYiAccountApplicationTests.java index 5d28046..acaa797 100644 --- a/ruoyi-service/ruoyi-account/src/test/java/com/ruoyi/order/account/RuoYiAccountApplicationTests.java +++ b/ruoyi-service/ruoyi-account/src/test/java/com/ruoyi/order/account/RuoYiAccountApplicationTests.java @@ -1,8 +1,61 @@ -package com.ruoyi.account; +package com.ruoyi.order.account; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.ruoyi.account.RuoYiAccountApplication; +import com.ruoyi.account.api.model.AppUser; +import com.ruoyi.account.service.AppUserService; +import org.junit.jupiter.api.Test; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; +import org.springframework.util.CollectionUtils; + +import java.util.ArrayList; +import java.util.List; +import java.util.stream.Collectors; @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT, classes = RuoYiAccountApplication.class) public class RuoYiAccountApplicationTests { + @Autowired + private AppUserService appUserService; + + @Test + public void contextLoads() { + List<AppUser> appUserList = appUserService.list(new LambdaQueryWrapper<AppUser>() + .ne(AppUser::getStatus, 3)); + ArrayList<Long> userIds = new ArrayList<>(); + userIds.add(1879046499735789569L); + getUserAncestorList(userIds, appUserList); + System.err.println(count1); + System.err.println(count2); + System.err.println(count3); + System.err.println(count4); + System.err.println(count5); + System.err.println(count6); + System.err.println(count7); + } + + 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); + } + } + + } -- Gitblit v1.7.1