From 8a1f648a86a9c2a7f8908c9b0c888ab1aad01b45 Mon Sep 17 00:00:00 2001
From: yjz <3131811435@qq.com>
Date: Mon, 5 Aug 2024 10:29:46 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E6=96=87=E6=A1=A3=E6=8B=86=E5=88=86)?=
 =?UTF-8?q?=EF=BC=9A=E4=BC=81=E6=A0=87=E6=8B=86=E5=88=86=E8=AF=A6=E6=83=85?=
 =?UTF-8?q?=E9=A1=B5=E6=9D=83=E9=99=90=E5=88=A4=E6=96=AD=E3=80=81=E8=AF=A6?=
 =?UTF-8?q?=E6=83=85=E9=A1=B5=E6=9D=A1=E6=AC=BE=E5=86=85=E5=AE=B9=E6=90=9C?=
 =?UTF-8?q?=E7=B4=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/DocumentSplitServiceImpl.java        | 48 +++++++++++++++----
 .../mapper/xml/SarFileSplitMenuEOMapper.xml   |  8 +++-
 .../split/page/SarFileSplitMenuEOPage.java    |  9 ++++
 .../impl/SarFileSplitMenuEOServiceImpl.java   |  4 ++
 4 files changed, 59 insertions(+), 10 deletions(-)

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 48cb9497..e5809953 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
@@ -261,6 +261,13 @@ public class DocumentSplitServiceImpl implements IDocumentSplitService {
         if (horizonValid) {
             isHorizontalOverstep(parameter.get(INFO_ID).toString());
         }
+        // 企标权限校验
+        if (parameter.containsKey("serialNumber")){
+            SarFileSplitInfoEO sarFileSplitInfoEO = new SarFileSplitInfoEO();
+            sarFileSplitInfoEO.setSerialNumber((String) parameter.get("serialNumber"));
+            enterpriseCheck(sarFileSplitInfoEO);
+            parameter.remove("serialNumber");
+        }
         // 查询全部字段列表
         List<LawsTag> fieldList = lawsCommonMapper.getFieldList(TableNameEnum.DOCUMENT_SPLIT.getTableName());
         // 列表展示字段
@@ -1033,11 +1040,21 @@ public class DocumentSplitServiceImpl implements IDocumentSplitService {
     public LawsDocumentSplitView queryView(Map<String, Object> parameter) {
         LawsDocumentSplitView lawsDocumentSplitView = new LawsDocumentSplitView();
         String id = (String) parameter.get(ID);
+        boolean itemContent = false;
         // 水平越权
         if (parameter.containsKey("horizontalOverstep")){
             isHorizontalOverstep(id);
             parameter.remove("horizontalOverstep");
         }
+
+        // 企标权限校验
+        if (parameter.containsKey("serial_number")){
+            SarFileSplitInfoEO sarFileSplitInfoEO = new SarFileSplitInfoEO();
+            sarFileSplitInfoEO.setSerialNumber((String) parameter.get("serial_number"));
+            enterpriseCheck(sarFileSplitInfoEO);
+            parameter.remove("serial_number");
+        }
+
         // 文档拆分信息
         DocumentSplitInfo documentSplitInfo = new DocumentSplitInfo();
         documentSplitInfo.setId(id);
@@ -1047,20 +1064,33 @@ public class DocumentSplitServiceImpl implements IDocumentSplitService {
         List<DocumentSplitInfo> documentSplitInfos = queryByList(documentSplitInfo);
         lawsDocumentSplitView.setDocumentSplitInfo(documentSplitInfos.get(0));
 
-        // 目录
-        SarFileSplitMenuEOPage splitMenuEOPage = new SarFileSplitMenuEOPage();
-        splitMenuEOPage.setValidFlag("0");
-        splitMenuEOPage.setOrderBy("display_seq");
-        splitMenuEOPage.setInfoId(id);
-        List<SarFileSplitMenuEO> sarFileSplitMenuEOS = sarFileSplitMenuEOService.queryByList(splitMenuEOPage);
-        // 不需要总目录层级
-        lawsDocumentSplitView.setSarFileSplitMenuEO(sarFileSplitMenuEOS.get(0).getChildren());
-
+        if (parameter.containsKey("item_content")){
+            itemContent = true;
+        }
         // 文档拆分详情
         parameter.remove(ID);
         parameter.put(INFO_ID, id);
         List<Map<String, Object>> documentSplitDetailByList = queryDocumentSplitDetailByList(parameter);
         lawsDocumentSplitView.setDocumentSplitDetail(documentSplitDetailByList);
+
+        // 目录
+        SarFileSplitMenuEOPage splitMenuEOPage = new SarFileSplitMenuEOPage();
+        splitMenuEOPage.setValidFlag("0");
+        splitMenuEOPage.setOrderBy("display_seq");
+        splitMenuEOPage.setInfoId(id);
+        if (itemContent && CollectionUtils.isNotEmpty(documentSplitDetailByList)){
+            List<String> menuId = documentSplitDetailByList.stream()
+                    .map(v -> String.valueOf(v.get("menu_id"))).distinct()
+                    .collect(Collectors.toList());
+            splitMenuEOPage.setIdList(menuId);
+            List<SarFileSplitMenuEO> sarFileSplitMenuEOS = sarFileSplitMenuEOService.queryByList(splitMenuEOPage);
+            lawsDocumentSplitView.setSarFileSplitMenuEO(sarFileSplitMenuEOS);
+        }else {
+            List<SarFileSplitMenuEO> sarFileSplitMenuEOS = sarFileSplitMenuEOService.queryByList(splitMenuEOPage);
+            // 不需要总目录层级
+            lawsDocumentSplitView.setSarFileSplitMenuEO(sarFileSplitMenuEOS.get(0).getChildren());
+        }
+
         return lawsDocumentSplitView;
     }
     @Override
diff --git a/laws-modules/src/main/java/com/jero/modules/split/mapper/xml/SarFileSplitMenuEOMapper.xml b/laws-modules/src/main/java/com/jero/modules/split/mapper/xml/SarFileSplitMenuEOMapper.xml
index fa171aa4..21ef7139 100644
--- a/laws-modules/src/main/java/com/jero/modules/split/mapper/xml/SarFileSplitMenuEOMapper.xml
+++ b/laws-modules/src/main/java/com/jero/modules/split/mapper/xml/SarFileSplitMenuEOMapper.xml
@@ -71,6 +71,12 @@
       <if test="modifyUser != null" >
         and modify_user ${modifyUserOperator} #{modifyUser}
       </if>
+        <if test="idList != null and idList.size>0" >
+            and id in
+            <foreach collection="idList" item="id" index="index" open="(" close=")" separator=",">
+                #{id}
+            </foreach>
+        </if>
     </trim>
   </sql>
 
@@ -178,7 +184,7 @@
 
   <!-- 根据id查询 SAR_FILE_SPLIT_MENU -->
   <select id="selectByPrimaryKey" resultMap="BaseResultMap" parameterType="java.lang.String">
-    select <include refid="Base_Column_List" /> 
+    select <include refid="Base_Column_List" />
       from SAR_FILE_SPLIT_MENU
      where id = #{value}
 
diff --git a/laws-modules/src/main/java/com/jero/modules/split/page/SarFileSplitMenuEOPage.java b/laws-modules/src/main/java/com/jero/modules/split/page/SarFileSplitMenuEOPage.java
index 0b5edb44..73227a3f 100644
--- a/laws-modules/src/main/java/com/jero/modules/split/page/SarFileSplitMenuEOPage.java
+++ b/laws-modules/src/main/java/com/jero/modules/split/page/SarFileSplitMenuEOPage.java
@@ -1,9 +1,12 @@
 package com.jero.modules.split.page;
 
+import com.baomidou.mybatisplus.annotation.TableField;
 import com.jero.modules.split.common.BasePage;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
 
+import java.util.List;
+
 /**
  * <b>功能：</b>SAR_FILE_SPLIT_MENU SarFileSplitMenuEOPage<br>
  * <b>作者：</b>code generator<br>
@@ -142,4 +145,10 @@ public class SarFileSplitMenuEOPage extends BasePage {
      */
     @ApiModelProperty("更新人运算符")
     private String modifyUserOperator = "=";
+    /**
+     * idList
+     */
+    @ApiModelProperty("idList")
+    @TableField(exist = false)
+    private List<String> idList;
 }
diff --git a/laws-modules/src/main/java/com/jero/modules/split/service/impl/SarFileSplitMenuEOServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/split/service/impl/SarFileSplitMenuEOServiceImpl.java
index 73e3328f..b9570066 100644
--- a/laws-modules/src/main/java/com/jero/modules/split/service/impl/SarFileSplitMenuEOServiceImpl.java
+++ b/laws-modules/src/main/java/com/jero/modules/split/service/impl/SarFileSplitMenuEOServiceImpl.java
@@ -205,6 +205,10 @@ public class SarFileSplitMenuEOServiceImpl extends ServiceImpl<SarFileSplitMenuE
             sarFileSplitMenuEO.setMoveDownFlag(moveDownFlag);
         }
 
+        if (CollectionUtils.isNotEmpty(page.getIdList())){
+            return result;
+        }
+
         // 关键字查询
         if (StringUtils.isNotBlank(itemName) && !result.isEmpty()){
             List<SarFileSplitMenuEO> trees = result.stream()