From 211840b64fa1132d76d6dff6c779e9ba2c0c450f Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 07 四月 2022 19:13:54 +0800
Subject: [PATCH] 优化自动化代码

---
 src/main/java/org/yeshi/utils/generater/vo/admin/controller/AdminControllerInfoVO.java |   56 +++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 53 insertions(+), 3 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 e55a0eb..124e173 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
@@ -19,11 +19,17 @@
     private ClassInfo service;
     private ClassInfo query;
     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 {
@@ -34,6 +40,8 @@
         private ClassInfo query;
         //
         private ClassInfo service;
+
+        private boolean responseBody;
 
 
         public Builder setAdminInfo(AdminGeneraterInfo generaterInfo) {
@@ -56,6 +64,11 @@
             return this;
         }
 
+        public Builder setResponseBody(boolean responseBody) {
+            this.responseBody = responseBody;
+            return this;
+        }
+
         public AdminControllerInfoVO build() throws Exception {
             if (generaterInfo == null) {
                 throw new Exception("generaterInfo涓嶈兘涓虹┖");
@@ -71,14 +84,19 @@
             }
 
             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.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");
+            vo.setControllerName(vo.getEntity().getName() + "AdminController");
             return vo;
         }
 
@@ -152,4 +170,36 @@
     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;
+    }
 }

--
Gitblit v1.8.0