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/vo/admin/controller/AdminControllerInfoVO.java |   43 +++++++++++++++++++++++++++++++++++++++----
 1 files changed, 39 insertions(+), 4 deletions(-)

diff --git a/src/main/java/org/yeshi/utils/generater/vo/admin/controller/AdminControllerInfoVO.java b/src/main/java/org/yeshi/utils/generater/vo/admin/controller/AdminControllerInfoVO.java
index a90b56a..f889bf7 100644
--- a/src/main/java/org/yeshi/utils/generater/vo/admin/controller/AdminControllerInfoVO.java
+++ b/src/main/java/org/yeshi/utils/generater/vo/admin/controller/AdminControllerInfoVO.java
@@ -18,7 +18,10 @@
     private ClassInfo entity;
     private ClassInfo service;
     private ClassInfo query;
+    private ClassInfo adminVO;
     private boolean jsonp;
+
+    private boolean responseBody;
     //涓婚敭绫诲瀷
     private String identifyIdType;
 
@@ -38,6 +41,10 @@
         private ClassInfo query;
         //
         private ClassInfo service;
+
+        private boolean responseBody;
+
+        private ClassInfo adminVO;
 
 
         public Builder setAdminInfo(AdminGeneraterInfo generaterInfo) {
@@ -60,6 +67,16 @@
             return this;
         }
 
+        public Builder setAdminVO(ClassInfo adminVO) {
+            this.adminVO = adminVO;
+            return this;
+        }
+
+        public Builder setResponseBody(boolean responseBody) {
+            this.responseBody = responseBody;
+            return this;
+        }
+
         public AdminControllerInfoVO build() throws Exception {
             if (generaterInfo == null) {
                 throw new Exception("generaterInfo涓嶈兘涓虹┖");
@@ -75,15 +92,17 @@
             }
 
             AdminControllerInfoVO vo = new AdminControllerInfoVO();
-            vo.setMapping(generaterInfo.getControllerData().mapping());
+            vo.setMapping(generaterInfo.getControllerData().getMapping());
             vo.setEntity(new ClassInfo(generaterInfo.getEntity().getSimpleName(), generaterInfo.getEntity().getName()));
-            vo.setJsonp(generaterInfo.getControllerData().jsonp());
+            vo.setJsonp(generaterInfo.getControllerData().isJsonp());
             vo.setPackageName(packageName);
             vo.setQueryVO(query);
             vo.setService(service);
+            vo.setAdminVO(adminVO);
             vo.setAdd(generaterInfo.getAddFormRows() != null && generaterInfo.getAddFormRows().size() > 0);
-            vo.setUpdate(generaterInfo.getUpdateFormRows() != null && generaterInfo.getUpdateFormRows().size() > 0 && generaterInfo.getControllerData().edit());
-            vo.setDelete(generaterInfo.getControllerData().delete());
+            vo.setUpdate(generaterInfo.getUpdateFormRows() != null && generaterInfo.getUpdateFormRows().size() > 0);
+            vo.setDelete(true);
+            vo.setResponseBody(generaterInfo.getControllerData().isResponseBody());
 
             vo.setIdentifyIdType(EntityUtil.getIdentifyId(generaterInfo.getEntity()).getType().getSimpleName());
             vo.setControllerName(vo.getEntity().getName() + "AdminController");
@@ -184,4 +203,20 @@
     public void setDelete(boolean delete) {
         this.delete = delete;
     }
+
+    public boolean isResponseBody() {
+        return responseBody;
+    }
+
+    public void setResponseBody(boolean responseBody) {
+        this.responseBody = responseBody;
+    }
+
+    public ClassInfo getAdminVO() {
+        return adminVO;
+    }
+
+    public void setAdminVO(ClassInfo adminVO) {
+        this.adminVO = adminVO;
+    }
 }

--
Gitblit v1.8.0