From 210ddec5ffc105ca48153d745ae2b8ecc0984250 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Wed, 10 Jan 2024 10:42:22 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=87=8D=E5=91=BD=E5=90=8D?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../coal/netDisk/service/NetDiskService.java     | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/main/java/cn/lihongjie/coal/netDisk/service/NetDiskService.java b/src/main/java/cn/lihongjie/coal/netDisk/service/NetDiskService.java
index fe4e9388..fa17f33c 100644
--- a/src/main/java/cn/lihongjie/coal/netDisk/service/NetDiskService.java
+++ b/src/main/java/cn/lihongjie/coal/netDisk/service/NetDiskService.java
@@ -899,6 +899,7 @@ public class NetDiskService extends BaseService<NetDiskEntity, NetDiskRepository
 
         NetDiskEntity entity = get(dto.getId());
 
+        validateEntityName(entity);
         if (entity.getParent() != null) {
             if (entity.getParent().getChildren().stream()
                     .anyMatch(x -> x.getId() != dto.getId() && x.getName().equals(dto.getName()))) {
@@ -906,6 +907,21 @@ public class NetDiskService extends BaseService<NetDiskEntity, NetDiskRepository
             }
         }
         entity.setName(dto.getName());
+
+        validateEntityName(entity);
+
+
         save(entity);
     }
+
+    private void validateEntityName(NetDiskEntity entity) {
+        if (StringUtils.equalsIgnoreCase(entity.getEntryType(), "0")){
+
+            validateDirName(entity.getName());
+
+        }else if (StringUtils.equalsIgnoreCase(entity.getEntryType(), "1")){
+
+            validateFileName(entity.getName());
+        }
+    }
 }