111
whywhyo
2023-05-05 e3e5d863f7da346e8beaed4fb6ed4a4fca0cb22b
zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
@@ -296,7 +296,7 @@
     * @return
     */
    @Override
    public AjaxResult listWithTree() {
    public AjaxResult listWithTree(Integer depth) {
        List<ZInfoUser> allPeopleList = list();
        List<ZInfoUser> result = null;
@@ -341,6 +341,24 @@
    }
    @Override
    public AjaxResult listMySelfAndSpouse(Long id) {
        LambdaQueryWrapper<ZInfoUser> lqw1 = new LambdaQueryWrapper<>();
        lqw1.eq(ZInfoUser::getUserId,id);
        ZInfoUser myself = getOne(lqw1);
        Long spouseId = myself.getSpouseId();
        LambdaQueryWrapper<ZInfoUser> lqw2 = new LambdaQueryWrapper<>();
        lqw2.eq(ZInfoUser::getUserId,spouseId);
        ZInfoUser spouse = getOne(lqw2);
        ArrayList<ZInfoUser> zInfoUsers = new ArrayList<>();
        zInfoUsers.add(myself);
        zInfoUsers.add(spouse);
        return AjaxResult.success(zInfoUsers);
    }
    /**
     * 递归算法
     * @param people
@@ -348,7 +366,10 @@
     * @return
     */
    private List<ZInfoUser> fillChildren(ZInfoUser people, List<ZInfoUser> allPeopleList) {
        return allPeopleList.stream().filter(
        // TODO: 2023-05-05 控制递归的次数
        List<ZInfoUser> collect = allPeopleList.stream().filter(
                one -> one.getFatherId() == people.getUserId() || one.getMomId() == people.getUserId()
        ).map(
                one -> {
@@ -357,6 +378,8 @@
                }
        ).collect(Collectors.toList());
        return collect;
    }