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/GeneraterManagerV2.java | 42 +++++++++++++++++++++++++++++++++++++-----
1 files changed, 37 insertions(+), 5 deletions(-)
diff --git a/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java b/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java
index 3f679a8..99af5a4 100644
--- a/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java
+++ b/src/main/java/org/yeshi/utils/generater/GeneraterManagerV2.java
@@ -9,6 +9,7 @@
import org.yeshi.utils.generater.util.NameUtil;
import org.yeshi.utils.generater.vo.ExceptionVO;
import org.yeshi.utils.generater.vo.admin.controller.AdminControllerInfoVO;
+import org.yeshi.utils.generater.vo.admin.controller.AdminVOVO;
import org.yeshi.utils.generater.vo.dao.MongoDBDaoVO;
import org.yeshi.utils.generater.vo.dao.MyBatisDBDaoVO;
import org.yeshi.utils.generater.vo.service.QueryVO;
@@ -82,7 +83,7 @@
FileUtil.saveAsFileByte(data.getBytes("UTF-8"), path);
}
- public void createException(Class base) throws Exception {
+ public void createException(ClassInfo base) throws Exception {
if (config.getService().getException() == null) {
throw new Exception("灏氭湭閰嶇疆寮傚父淇℃伅");
}
@@ -90,7 +91,7 @@
String exceptionName = NameUtil.getDefaultExceptionName(config.getEntity());
String path = getDirPath(config.getService().getException().getPkg());
try {
- ExceptionVO vo = new ExceptionVO.Builder().setEntity(config.getEntity()).setPackageName(config.getService().getException().getPkg()).setBase(new ClassInfo(base.getSimpleName(), base.getName())).build();
+ ExceptionVO vo = new ExceptionVO.Builder().setEntity(config.getEntity()).setPackageName(config.getService().getException().getPkg()).setBase(base).build();
String result = GeneraterUtil.createException(vo);
//淇濆瓨
saveFile(result, new File(path, exceptionName + ".java").getAbsolutePath());
@@ -98,6 +99,25 @@
e.printStackTrace();
}
}
+
+
+ public void createAdminVO() throws Exception {
+ if (StringUtil.isNullOrEmpty(config.getService().getAdminVOPackage())) {
+ throw new Exception("灏氭湭閰嶇疆AdminVO");
+ }
+
+ String name = NameUtil.getDefaultExceptionName(config.getEntity());
+ String path = getDirPath(config.getService().getAdminVOPackage());
+ try {
+ AdminVOVO vo = new AdminVOVO.Builder().setEntity(config.getEntity()).setPackageName(config.getService().getAdminVOPackage()).build();
+ String result = GeneraterUtil.createAdminVO(vo);
+ //淇濆瓨
+ saveFile(result, new File(path, name + ".java").getAbsolutePath());
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
public void createDao() throws Exception {
@@ -220,7 +240,15 @@
}
if (config.getService().getException() != null) {
- createException(Exception.class);
+ if (config.getService().getException().getBase() != null) {
+ createException(config.getService().getException().getBase());
+ } else {
+ createException(ClassInfo.create(Exception.class));
+ }
+ }
+
+ if (config.getService().getAdminVOPackage() != null) {
+ createAdminVO();
}
QueryVO queryVO = createQuery(config.getService().getQueryPackage());
@@ -388,8 +416,12 @@
}
- public static void main(String[] args) {
- System.out.print(getHtmlNameFromEntityName("TTestEEntity"));
+ public static void main(String[] args) throws Exception {
+// System.out.print(getHtmlNameFromEntityName("TTestEEntity"));
+// AdminVOVO vo = new AdminVOVO.Builder().setEntity(TestEntity3.class).setPackageName("test.vo.admin").build();
+// String result = GeneraterUtil.createAdminVO(vo);
+// System.out.println(result);
+
}
--
Gitblit v1.8.0