From e1fbb067f7e933d9f4add1254e89a07db0cb96e2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E6=80=9D=E8=B1=AA?= <1437200870@qq.com>
Date: Tue, 27 Aug 2024 11:47:42 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=AE=9E=E4=BD=93=E7=B1=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/main/java/com/muyu/domain/TableInfo.java       |  6 ++++++
 .../com/muyu/service/impl/TableInfoServiceImpl.java    | 10 +++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/cloud-task-common/src/main/java/com/muyu/domain/TableInfo.java b/cloud-task-common/src/main/java/com/muyu/domain/TableInfo.java
index 6213f2e..465a790 100644
--- a/cloud-task-common/src/main/java/com/muyu/domain/TableInfo.java
+++ b/cloud-task-common/src/main/java/com/muyu/domain/TableInfo.java
@@ -1,5 +1,6 @@
 package com.muyu.domain;
 
+import com.baomidou.mybatisplus.annotation.TableField;
 import com.muyu.common.core.web.domain.BaseEntity;
 import lombok.AllArgsConstructor;
 import lombok.Data;
@@ -7,6 +8,8 @@ import lombok.EqualsAndHashCode;
 import lombok.NoArgsConstructor;
 import lombok.experimental.SuperBuilder;
 
+import java.util.List;
+
 /**
  * @PackageName:com.muyu.domain
  * @ClassName:TableInfo
@@ -28,4 +31,7 @@ public class TableInfo extends BaseEntity {
     private String dataNum;
     private String center;
     private String parentId;
+
+    @TableField(exist = false)
+    private List<TableInfo> tableInfoList;
 }
diff --git a/cloud-task-server/src/main/java/com/muyu/service/impl/TableInfoServiceImpl.java b/cloud-task-server/src/main/java/com/muyu/service/impl/TableInfoServiceImpl.java
index 9f4ef25..7a93455 100644
--- a/cloud-task-server/src/main/java/com/muyu/service/impl/TableInfoServiceImpl.java
+++ b/cloud-task-server/src/main/java/com/muyu/service/impl/TableInfoServiceImpl.java
@@ -26,7 +26,15 @@ public class TableInfoServiceImpl extends ServiceImpl<TableInfoMapper, TableInfo
     @Override
     public List<TableInfo> selectList() {
         LambdaQueryWrapper<TableInfo> queryWrapper = new LambdaQueryWrapper<>();
-        return this.list(queryWrapper);
+        queryWrapper.eq(TableInfo::getParentId,0);
+        List<TableInfo> list = this.list(queryWrapper);
+        list.forEach(tableInfo -> {
+            LambdaQueryWrapper<TableInfo> queryWrapperC = new LambdaQueryWrapper<>();
+            queryWrapperC.eq(TableInfo::getParentId,tableInfo.getId());
+            List<TableInfo> list1 = this.list(queryWrapperC);
+            tableInfo.setTableInfoList(list1);
+        });
+        return list;
     }
 
     @Override