From 5662b27adbeb876b6947b56ed4459f666a960f6b Mon Sep 17 00:00:00 2001
From: "liang.chao" <1360241448@qq.com>
Date: Tue, 16 Sep 2025 13:33:06 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=87=E4=BB=B6=E5=88=86=E7=B1=BB=E6=A0=87?=
 =?UTF-8?q?=E8=AE=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/bonus/web/mapper/ArchiveMapper.java     |  2 ++
 .../bonus/web/service/impl/ArchiveServiceImpl.java   | 12 ++++++++++++
 .../src/main/resources/mapper/ArchiveMapper.xml      |  3 +++
 3 files changed, 17 insertions(+)

diff --git a/bonus-admin/src/main/java/com/bonus/web/mapper/ArchiveMapper.java b/bonus-admin/src/main/java/com/bonus/web/mapper/ArchiveMapper.java
index 2cf3e44..b241c37 100644
--- a/bonus-admin/src/main/java/com/bonus/web/mapper/ArchiveMapper.java
+++ b/bonus-admin/src/main/java/com/bonus/web/mapper/ArchiveMapper.java
@@ -39,4 +39,6 @@ public interface ArchiveMapper {
     Integer getSortById(Integer id);
 
     String getClassifyMark(ArchivalCatalogueDto dto);
+
+    Integer getLevelById(Integer id);
 }
diff --git a/bonus-admin/src/main/java/com/bonus/web/service/impl/ArchiveServiceImpl.java b/bonus-admin/src/main/java/com/bonus/web/service/impl/ArchiveServiceImpl.java
index 4cd14b3..f4e5d2b 100644
--- a/bonus-admin/src/main/java/com/bonus/web/service/impl/ArchiveServiceImpl.java
+++ b/bonus-admin/src/main/java/com/bonus/web/service/impl/ArchiveServiceImpl.java
@@ -12,6 +12,7 @@ import com.bonus.web.service.ArchiveService;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
+import java.util.Iterator;
 import java.util.List;
 
 /**
@@ -27,6 +28,15 @@ public class ArchiveServiceImpl implements ArchiveService {
     @Override
     public List<ArchivalCatalogueDto> getArchivalCatalogueTree(ArchivalCatalogueDto dto) {
         List<ArchivalCatalogueDto> archivalCatalogueDto = archiveMapper.getArchivalCatalogueTree(dto);
+        if (dto.getId() != null) {
+            Integer level = archiveMapper.getLevelById(dto.getId());
+            for (Iterator<ArchivalCatalogueDto> it = archivalCatalogueDto.iterator(); it.hasNext(); ) {
+                ArchivalCatalogueDto item = it.next();
+                if (level <= item.getLevel()) {
+                    it.remove();
+                }
+            }
+        }
         // 构建树
         List<ArchivalCatalogueDto> tree = TreeBuilder.buildTree(archivalCatalogueDto);
         return tree;
@@ -51,6 +61,8 @@ public class ArchiveServiceImpl implements ArchiveService {
         }
         vo.setCreateUserId(SecurityUtils.getLoginUser().getUserId().toString());
         vo.setCreateUserName(SecurityUtils.getLoginUser().getUsername());
+        vo.setUpdateUserId(SecurityUtils.getLoginUser().getUserId().toString());
+        vo.setUpdateUserName(SecurityUtils.getLoginUser().getUsername());
         return R.ok(archiveMapper.saveArchivalCatalogue(vo));
     }
 
diff --git a/bonus-admin/src/main/resources/mapper/ArchiveMapper.xml b/bonus-admin/src/main/resources/mapper/ArchiveMapper.xml
index 8dcf587..77d731e 100644
--- a/bonus-admin/src/main/resources/mapper/ArchiveMapper.xml
+++ b/bonus-admin/src/main/resources/mapper/ArchiveMapper.xml
@@ -182,5 +182,8 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		   left join da_ky_files_classify_mark dkfcm on dkfc.classify_mark = dkfcm.id
 		where dkfc.id = #{id}
 	</select>
+	<select id="getLevelById" resultType="java.lang.Integer">
+		select `level` from da_ky_files_contents where id = #{id}
+	</select>
 
 </mapper>