From 74ff03182e0ec07fffebb44a2ad4f684f6a46200 Mon Sep 17 00:00:00 2001
From: Bunny <1319900154@qq.com>
Date: Tue, 1 Jul 2025 15:31:58 +0800
Subject: [PATCH] =?UTF-8?q?:poop:=20=E5=BC=82=E5=B8=B8=E5=A4=84=E7=90=86?=
 =?UTF-8?q?=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../bunny/service/impl/TableServiceImpl.java  |  2 -
 .../cn/bunny/service/impl/VmsServiceImpl.java | 42 ++++++++++++-------
 2 files changed, 27 insertions(+), 17 deletions(-)

diff --git a/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java b/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java
index 2ed62b7..00a9018 100644
--- a/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java
+++ b/generator-code-server/src/main/java/cn/bunny/service/impl/TableServiceImpl.java
@@ -5,7 +5,6 @@ import cn.bunny.domain.entity.DatabaseInfoMetaData;
 import cn.bunny.domain.entity.TableMetaData;
 import cn.bunny.service.TableService;
 import lombok.RequiredArgsConstructor;
-import lombok.SneakyThrows;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
@@ -22,7 +21,6 @@ public class TableServiceImpl implements TableService {
      *
      * @return 当前连接的数据库信息属性
      */
-    @SneakyThrows
     @Override
     public DatabaseInfoMetaData databaseInfoMetaData() {
         List<TableMetaData> databaseTableList = databaseMetadataProvider.getTableMetadataBatch(null);
diff --git a/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java b/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java
index 541a070..3f1117b 100644
--- a/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java
+++ b/generator-code-server/src/main/java/cn/bunny/service/impl/VmsServiceImpl.java
@@ -10,7 +10,6 @@ import cn.bunny.utils.ResourceFileUtil;
 import cn.bunny.utils.VmsUtil;
 import cn.hutool.crypto.digest.MD5;
 import lombok.RequiredArgsConstructor;
-import lombok.SneakyThrows;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
@@ -35,24 +34,37 @@ public class VmsServiceImpl implements VmsService {
         return codeGeneratorService.generateCode(dto);
     }
 
-    @SneakyThrows
+    /**
+     * 获取vms文件路径
+     *
+     * @return vms下的文件路径
+     */
     @Override
     public Map<String, List<VmsPathVo>> vmsResourcePathList() {
-        List<String> vmsRelativeFiles = ResourceFileUtil.getRelativeFiles("vms");
+        List<String> vmsRelativeFiles;
+        Map<String, List<VmsPathVo>> listMap;
 
-        return vmsRelativeFiles.stream()
-                .map(vmFile -> {
-                    String[] filepathList = vmFile.split("/");
-                    String filename = filepathList[filepathList.length - 1].replace(".vm", "");
+        try {
+            vmsRelativeFiles = ResourceFileUtil.getRelativeFiles("vms");
+            listMap = vmsRelativeFiles.stream()
+                    .map(vmFile -> {
+                        String[] filepathList = vmFile.split("/");
+                        String filename = filepathList[filepathList.length - 1].replace(".vm", "");
 
-                    return VmsPathVo.builder()
-                            .id(VmsUtil.generateDivId())
-                            .name(vmFile)
-                            .label(filename)
-                            .type(filepathList[0])
-                            .build();
-                })
-                .collect(Collectors.groupingBy(VmsPathVo::getType));
+                        return VmsPathVo.builder()
+                                .id(VmsUtil.generateDivId())
+                                .name(vmFile)
+                                .label(filename)
+                                .type(filepathList[0])
+                                .build();
+                    })
+                    .collect(Collectors.groupingBy(VmsPathVo::getType));
+        } catch (Exception e) {
+            throw new RuntimeException("Get error of VMS path:" + e.getMessage());
+        }
+
+
+        return listMap;
     }
 
     @Override