From 47e3087067abd35e6337c011f96d2338c0bb1aae Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 25 七月 2024 13:39:31 +0800
Subject: [PATCH] 优化自动化代码

---
 src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java |   25 +++++++++++++++----------
 1 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java b/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java
index 99af5a4..2f70d3b 100644
--- a/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java
+++ b/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java
@@ -101,21 +101,24 @@
     }
 
 
-    public void createAdminVO() throws Exception {
-        if (StringUtil.isNullOrEmpty(config.getService().getAdminVOPackage())) {
-            throw new Exception("灏氭湭閰嶇疆AdminVO");
+    public AdminVOVO createAdminVO() throws Exception {
+        if (StringUtil.isNullOrEmpty(config.getAdmin().getController().getAdminVOPackage())) {
+            return null;
         }
 
-        String name = NameUtil.getDefaultExceptionName(config.getEntity());
-        String path = getDirPath(config.getService().getAdminVOPackage());
+        String name = NameUtil.getDefaultAdminVOName(config.getEntity());
+        String path = getDirPath(config.getAdmin().getController().getAdminVOPackage());
         try {
-            AdminVOVO vo = new AdminVOVO.Builder().setEntity(config.getEntity()).setPackageName(config.getService().getAdminVOPackage()).build();
+            AdminVOVO vo = new AdminVOVO.Builder().setEntity(config.getEntity()).setPackageName(config.getAdmin().getController().getAdminVOPackage()).build();
             String result = GeneraterUtil.createAdminVO(vo);
             //淇濆瓨
             saveFile(result, new File(path, name + ".java").getAbsolutePath());
+            return vo;
         } catch (Exception e) {
             e.printStackTrace();
         }
+
+        return null;
     }
 
 
@@ -247,9 +250,6 @@
             }
         }
 
-        if (config.getService().getAdminVOPackage() != null) {
-            createAdminVO();
-        }
 
         QueryVO queryVO = createQuery(config.getService().getQueryPackage());
 
@@ -315,13 +315,18 @@
             query = new ClassInfo(queryName, config.getService().getQueryPackage() + "." + queryName);
         }
 
+        AdminVOVO adminVOVO = null;
+        if (config.getAdmin().getController().getAdminVOPackage() != null) {
+            adminVOVO = createAdminVO();
+        }
+
 
         String path = getDirPath(config.getAdmin().getController().getPkg());
 
         AdminGeneraterInfo generaterInfo = new AdminGeneraterInfo.Builder().setEntityClass(config.getEntity()).setAdminData(config.getAdmin()).build();
 
 
-        AdminControllerInfoVO vo = new AdminControllerInfoVO.Builder().setAdminInfo(generaterInfo).setPackageName(generaterInfo.getControllerData().getPkg()).setQueryVO(query).setService(service).build();
+        AdminControllerInfoVO vo = new AdminControllerInfoVO.Builder().setAdminVO(adminVOVO == null ? null : new ClassInfo(adminVOVO.getName(), adminVOVO.getPackageName() + "." + adminVOVO.getName())).setAdminInfo(generaterInfo).setPackageName(generaterInfo.getControllerData().getPkg()).setQueryVO(query).setService(service).build();
         String controllerName = vo.getControllerName();
         //鐢熸垚controller
         String result = GeneraterUtil.createAdminController(vo);

--
Gitblit v1.8.0