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 |   19 ++++++++++++++++++-
 1 files changed, 18 insertions(+), 1 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 124e173..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,6 +18,7 @@
     private ClassInfo entity;
     private ClassInfo service;
     private ClassInfo query;
+    private ClassInfo adminVO;
     private boolean jsonp;
 
     private boolean responseBody;
@@ -43,6 +44,8 @@
 
         private boolean responseBody;
 
+        private ClassInfo adminVO;
+
 
         public Builder setAdminInfo(AdminGeneraterInfo generaterInfo) {
             this.generaterInfo = generaterInfo;
@@ -61,6 +64,11 @@
 
         public Builder setService(ClassInfo service) {
             this.service = service;
+            return this;
+        }
+
+        public Builder setAdminVO(ClassInfo adminVO) {
+            this.adminVO = adminVO;
             return this;
         }
 
@@ -90,8 +98,9 @@
             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 );
+            vo.setUpdate(generaterInfo.getUpdateFormRows() != null && generaterInfo.getUpdateFormRows().size() > 0);
             vo.setDelete(true);
             vo.setResponseBody(generaterInfo.getControllerData().isResponseBody());
 
@@ -202,4 +211,12 @@
     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