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 |   83 ++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 81 insertions(+), 2 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 baf35ef..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,9 +18,19 @@
     private ClassInfo entity;
     private ClassInfo service;
     private ClassInfo query;
+    private ClassInfo adminVO;
     private boolean jsonp;
+
+    private boolean responseBody;
     //涓婚敭绫诲瀷
     private String identifyIdType;
+
+    //鍚嶇О
+    private String controllerName;
+
+    private boolean add;
+    private boolean update;
+    private boolean delete;
 
 
     public static class Builder {
@@ -31,6 +41,10 @@
         private ClassInfo query;
         //
         private ClassInfo service;
+
+        private boolean responseBody;
+
+        private ClassInfo adminVO;
 
 
         public Builder setAdminInfo(AdminGeneraterInfo generaterInfo) {
@@ -53,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涓嶈兘涓虹┖");
@@ -68,13 +92,20 @@
             }
 
             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);
+            vo.setDelete(true);
+            vo.setResponseBody(generaterInfo.getControllerData().isResponseBody());
+
             vo.setIdentifyIdType(EntityUtil.getIdentifyId(generaterInfo.getEntity()).getType().getSimpleName());
+            vo.setControllerName(vo.getEntity().getName() + "AdminController");
             return vo;
         }
 
@@ -140,4 +171,52 @@
     public void setIdentifyIdType(String identifyIdType) {
         this.identifyIdType = identifyIdType;
     }
+
+    public String getControllerName() {
+        return controllerName;
+    }
+
+    public void setControllerName(String controllerName) {
+        this.controllerName = controllerName;
+    }
+
+    public boolean isAdd() {
+        return add;
+    }
+
+    public void setAdd(boolean add) {
+        this.add = add;
+    }
+
+    public boolean isUpdate() {
+        return update;
+    }
+
+    public void setUpdate(boolean update) {
+        this.update = update;
+    }
+
+    public boolean isDelete() {
+        return delete;
+    }
+
+    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