From 4e94b05b765acb54053123580c570468d41371e1 Mon Sep 17 00:00:00 2001
From: whywhyo <1511349576@qq.com>
Date: 星期五, 14 七月 2023 11:18:19 +0800
Subject: [PATCH] 123456

---
 zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java |   29 ++++++++++++++++++++---------
 1 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java b/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
index 6e8139c..7332469 100644
--- a/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
+++ b/zhang-content/src/main/java/com/ruoyi/service/impl/ZInfoUserServiceImpl.java
@@ -264,17 +264,23 @@
         List<ZInfoUser> allPeopleList = list();
         List<ZInfoUser> result = null;
         try {
-            result = allPeopleList.stream().filter(people -> people.getUserId()!=1&&(people.getFatherId() == 0||people.getMomId()==0))
-                    .map(people -> {
-                        if(depth>1){
-                            people.setIdentity(1);
-                            people.setChildList(fillChildren(people, allPeopleList, depth-1));
-                        }
-                        return people;
-                    }).collect(Collectors.toList());
+              result = allPeopleList.stream().filter(people -> people.getUserId()!=1&&(people.getFatherId() == 0||people.getMomId()==0))
+                .map(people -> {
+                    if(depth>1){
+                        people.setIdentity(1);
+                        //鏍规嵁閰嶅伓id鏌ヨ閰嶅伓鐨勬暟鎹紝灏佽濂戒竴璧疯繑鍥�
+                        Long spouseId = people.getSpouseId();
+                        ZInfoUser spouse = zInfoUserService.getById(spouseId);
+                        people.setSpouse(spouse);
+                        people.setChildList(fillChildren(people, allPeopleList, depth-1));
+                    }
+                    return people;
+                }).collect(Collectors.toList());
         } catch (NullPointerException e) {
             throw new RuntimeException("鎮ㄥ湪鍔犲叆鎴愬憳鐨勬椂鍊欐病鏈夋寚瀹氳鎴愬憳鐨勭埗浜叉垨鑰呮瘝浜�");
         }
+
+
         return AjaxResult.success(result);
 
     }
@@ -475,13 +481,18 @@
        // layer = layer + 1;
         System.out.println(depth);
         List<ZInfoUser> collect = allPeopleList.stream().filter(
-                one -> one.getFatherId() == people.getUserId() || one.getMomId() == people.getUserId()
+                one -> (one.getFatherId() == people.getUserId() || one.getMomId() == people.getUserId()) && (people.getIsMyFamily() == 1)
         ).map(
                 one -> {
                     if(depth==1)
                         return one;
                     else {
                         one.setIdentity(people.getIdentity()+1);
+
+                        //鏍规嵁閰嶅伓id鏌ヨ閰嶅伓鐨勬暟鎹紝灏佽濂戒竴璧疯繑鍥�
+                        Long spouseId = one.getSpouseId();
+                        ZInfoUser spouse = zInfoUserService.getById(spouseId);
+                        one.setSpouse(spouse);
                         List<ZInfoUser> zinfo = fillChildren(one, allPeopleList, depth - 1);
                         one.setChildList(zinfo);
                         System.out.println(one);

--
Gitblit v1.9.1