From 803b8e39bec8b0bd79b2793a1751841208a20601 Mon Sep 17 00:00:00 2001
From: yjz <3131811435@qq.com>
Date: Thu, 16 Nov 2023 14:51:55 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E6=96=87=E6=A1=A3=E6=8B=86=E5=88=86):=20?=
 =?UTF-8?q?=E9=80=9A=E8=BF=87=E6=96=87=E6=A1=A3=E6=8B=86=E5=88=86id?=
 =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E7=9B=AE=E5=BD=95=E6=A0=91=E5=BD=A2=E7=BB=93?=
 =?UTF-8?q?=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/DocumentSplitController.java       | 13 +++++++++++++
 .../service/IDocumentSplitService.java            |  8 ++++++++
 .../service/impl/DocumentSplitServiceImpl.java    | 15 +++++++++++++++
 3 files changed, 36 insertions(+)

diff --git a/laws-modules/src/main/java/com/jero/modules/laws/documenttool/controller/DocumentSplitController.java b/laws-modules/src/main/java/com/jero/modules/laws/documenttool/controller/DocumentSplitController.java
index 6ab6cd43..76e8223b 100644
--- a/laws-modules/src/main/java/com/jero/modules/laws/documenttool/controller/DocumentSplitController.java
+++ b/laws-modules/src/main/java/com/jero/modules/laws/documenttool/controller/DocumentSplitController.java
@@ -11,6 +11,7 @@ import com.jero.modules.laws.documenttool.entity.SplitFileInfo;
 import com.jero.modules.laws.documenttool.service.IDocumentSplitService;
 import com.jero.modules.split.entity.SarFileSplitInfoEO;
 import com.jero.modules.split.entity.SarFileSplitItemsValEO;
+import com.jero.modules.split.entity.SarFileSplitMenuEO;
 import com.jero.modules.split.service.ISarFileSplitInfoService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -90,6 +91,18 @@ public class DocumentSplitController {
         return Result.OK(documentSplitService.queryView(parameter));
     }
 
+    /**
+     * 文档拆分-通过文档拆分id查询目录树形结构
+     *
+     * @return
+     */
+    @AutoLog(value = "文档拆分-通过文档拆分id查询目录树形结构")
+    @ApiOperation(value="文档拆分-通过文档拆分id查询目录树形结构", notes="文档拆分-通过文档拆分id查询目录树形结构")
+    @GetMapping(value = "/queryTreeMenuByInfoId")
+    public Result<List<SarFileSplitMenuEO>> queryTreeMenuByInfoId(String infoId) {
+        return Result.OK(documentSplitService.queryTreeMenuByInfoId(infoId));
+    }
+
     /**
      *  更改权限
      *
diff --git a/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/IDocumentSplitService.java b/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/IDocumentSplitService.java
index 815bf0a8..a8719567 100644
--- a/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/IDocumentSplitService.java
+++ b/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/IDocumentSplitService.java
@@ -6,6 +6,7 @@ import com.jero.modules.laws.documenttool.entity.LawsDocumentSplitView;
 import com.jero.modules.laws.documenttool.entity.SplitFileInfo;
 import com.jero.modules.split.entity.SarFileSplitInfoEO;
 import com.jero.modules.split.entity.SarFileSplitItemsValEO;
+import com.jero.modules.split.entity.SarFileSplitMenuEO;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
@@ -183,4 +184,11 @@ public interface IDocumentSplitService {
      * @return
      */
     IPage<SplitFileInfo> querySplitFileInfoByPage(SplitFileInfo splitFileInfo, Integer pageNo, Integer pageSize);
+
+    /**
+     * 文档拆分-通过文档拆分id查询目录树形结构
+     * @param infoId
+     * @return
+     */
+    List<SarFileSplitMenuEO> queryTreeMenuByInfoId(String infoId);
 }
diff --git a/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/impl/DocumentSplitServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/impl/DocumentSplitServiceImpl.java
index debacf1d..cdbbe01f 100644
--- a/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/impl/DocumentSplitServiceImpl.java
+++ b/laws-modules/src/main/java/com/jero/modules/laws/documenttool/service/impl/DocumentSplitServiceImpl.java
@@ -1058,6 +1058,21 @@ public class DocumentSplitServiceImpl implements IDocumentSplitService {
         return documentSplitMapper.querySplitFileInfoByPage(queryWrapper, new Page<>(pageNo, pageSize));
     }
 
+    @Override
+    public List<SarFileSplitMenuEO> queryTreeMenuByInfoId(String infoId) {
+        // 目录
+        SarFileSplitMenuEOPage splitMenuEOPage = new SarFileSplitMenuEOPage();
+        splitMenuEOPage.setValidFlag("0");
+        splitMenuEOPage.setOrderBy("display_seq");
+        splitMenuEOPage.setInfoId(infoId);
+        List<SarFileSplitMenuEO> sarFileSplitMenuEOS = sarFileSplitMenuEOService.queryByList(splitMenuEOPage);
+
+        if (!sarFileSplitMenuEOS.isEmpty() && !Objects.isNull(sarFileSplitMenuEOS.get(0).getChildren())){
+            return sarFileSplitMenuEOS.get(0).getChildren();
+        }
+        return new ArrayList<>();
+    }
+
     private void manageData(Map<String, Object> resultMap, StringBuilder fieldsBuilder, StringBuilder valuesBuilder) {
         // 所有key(字段名)
         resultMap.forEach((k, v) -> {