From 052e1d5c47c4e536fde79074d53b0481c7d4f9b6 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期三, 13 十月 2021 18:33:42 +0800
Subject: [PATCH] 自动化代码生成优化
---
src/test/java/com/generater/GeneratorTest.java | 125 ++++++++++++++++++++++++++++++++---------
1 files changed, 98 insertions(+), 27 deletions(-)
diff --git a/src/test/java/com/generater/GeneratorTest.java b/src/test/java/com/generater/GeneratorTest.java
index 818a88f..b08f64b 100644
--- a/src/test/java/com/generater/GeneratorTest.java
+++ b/src/test/java/com/generater/GeneratorTest.java
@@ -1,29 +1,27 @@
package com.generater;
-import freemarker.cache.StringTemplateLoader;
-import freemarker.template.Template;
import org.springframework.data.annotation.Id;
-import org.springframework.data.annotation.Transient;
+import org.yeshi.utils.FileUtil;
import org.yeshi.utils.FreemarkerUtils;
-import org.yeshi.utils.generater.SpringComponentGenerater;
-import org.yeshi.utils.generater.annotation.admin.form.CheckBox;
-import org.yeshi.utils.generater.annotation.admin.form.Select;
-import org.yeshi.utils.generater.annotation.admin.form.Text;
-import org.yeshi.utils.generater.entity.KeyValue;
-import org.yeshi.utils.generater.entity.MongoDBDaoData;
+import org.yeshi.utils.generater.GeneraterManager;
+import org.yeshi.utils.generater.GeneraterUtil;
+import org.yeshi.utils.generater.entity.ClassInfo;
import org.yeshi.utils.generater.entity.admin.AdminGeneraterInfo;
import org.yeshi.utils.generater.mybatis.Column;
-import org.yeshi.utils.generater.mybatis.MyBatisMapperUtil;
import org.yeshi.utils.generater.mybatis.Table;
+import org.yeshi.utils.generater.params.DaoGeneraterParams;
+import org.yeshi.utils.generater.vo.ExceptionVO;
+import org.yeshi.utils.generater.vo.dao.MongoDBDaoVO;
+import org.yeshi.utils.generater.vo.dao.MyBatisDBDaoVO;
+import org.yeshi.utils.generater.vo.service.ServiceImplInfoVO;
+import org.yeshi.utils.generater.vo.service.ServiceInfoVO;
+import org.yeshi.utils.generater.vo.admin.controller.AdminControllerInfoVO;
+import org.yeshi.utils.generater.vo.admin.page.FormUpdateInfoVO;
-import java.io.File;
import java.io.InputStream;
-import java.io.StringWriter;
-import java.lang.annotation.Annotation;
-import java.lang.reflect.Field;
-import java.lang.reflect.Method;
-import java.util.*;
-import java.util.regex.Pattern;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
public class GeneratorTest {
@@ -48,8 +46,14 @@
// }
- AdminGeneraterInfo info = new AdminGeneraterInfo.Builder().setEntityClass(TestEntity.class).build();
+ //
+// boolean match = Pattern.matches("^(https://){1}.*(\\.m3u8)$", "https://n1.szjal.cn/20210805/qnDXhXBa/index.m3u8123");
+// System.out.println(match);
+ generater();
+ }
+ private static void generateAdminPage() throws Exception {
+ AdminGeneraterInfo info = new AdminGeneraterInfo.Builder().setEntityClass(TestEntity.class).build();
//灏嗘墍鏈夌殑鎺т欢妯℃澘鍔犺浇杩涘幓
String[] template = new String[]{
"admin/form/checkBox.ftl",
@@ -60,24 +64,91 @@
"admin/form/switch.ftl",
"admin/form/text.ftl",
"admin/form/textArea.ftl",
- "admin/add.ftl"
+ "admin/add.ftl",
+ "admin/update.ftl",
+ "admin/list.ftl"
};
Map<String, InputStream> map = new HashMap<>();
for (String st : template) {
- map.put(st, SpringComponentGenerater.class.getClassLoader().getResourceAsStream("generater/" + st));
+ map.put(st, GeneraterUtil.class.getClassLoader().getResourceAsStream("generater/" + st));
}
-
- String result = FreemarkerUtils.generateInputStream(info, "admin/add.ftl", map);
-
-
+ String result = FreemarkerUtils.generateInputStream(new FormUpdateInfoVO.Builder().setAdminInfo(info).build(), "admin/update.ftl", map);
+ FileUtil.saveAsFileByte(result.getBytes("UTF-8"), "C:\\Users\\Administrator\\Documents\\HBuilderProjects\\LayuiTest\\form.html");
System.out.println(result);
+ }
- //
-// boolean match = Pattern.matches("^(https://){1}.*(\\.m3u8)$", "https://n1.szjal.cn/20210805/qnDXhXBa/index.m3u8123");
-// System.out.println(match);
+ private static void generateService() throws Exception {
+ ClassInfo dao = new ClassInfo("TestDao", "com.test.dao.TestDao");
+ ClassInfo query = new ClassInfo("TestQuery", "com.test.query.TestQuery");
+ ClassInfo service = new ClassInfo("TestService", "com.test.service.TestService");
+ ServiceInfoVO serviceInfoVO = new ServiceInfoVO.Builder().setEntity(Entity.class).setPackageName("com.test.service").setQuery(query).setService(service).build();
+ String result = FreemarkerUtils.generateInputStream(serviceInfoVO, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\service\\service.template"));
+ System.out.println(result);
+ }
+ private static void generateServiceImpl() throws Exception {
+ ClassInfo dao = new ClassInfo("TestMapper", "com.test.dao.TestMapper");
+ ClassInfo query = new ClassInfo("TestQuery", "com.test.query.TestQuery");
+ ClassInfo service = new ClassInfo("TestService", "com.test.service.TestService");
+ ClassInfo daoQuery = new ClassInfo("DaoQuery", "com.test.dao.query.DaoQuery");
+ ServiceImplInfoVO serviceInfoVO = new ServiceImplInfoVO.Builder()
+ .setDao(dao)
+ .setEntity(Entity.class)
+ .setPackageName("com.test.service.impl")
+ .setQuery(query)
+ .setService(service)
+ .setDaoQuery(daoQuery)
+ .build();
+ String result = FreemarkerUtils.generateInputStream(serviceInfoVO, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\service\\serviceImpl.template"));
+ System.out.println(result);
+ }
+
+
+ private static void generateAdminController() throws Exception {
+ AdminGeneraterInfo adminGeneraterInfo = new AdminGeneraterInfo.Builder().setEntityClass(TestEntity.class).build();
+ ClassInfo query = new ClassInfo("TestQuery", "com.test.query.TestQuery");
+ ClassInfo service = new ClassInfo("TestService", "com.test.service.TestService");
+ AdminControllerInfoVO vo = new AdminControllerInfoVO.Builder().setAdminInfo(adminGeneraterInfo).setPackageName("com.test.controller.admin").setQueryVO(query).setService(service).build();
+ String result = FreemarkerUtils.generateInputStream(vo, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\admin\\adminController.template"));
+ System.out.println(result);
+ }
+
+
+ private static void generateMongoDBDao() throws Exception {
+ MongoDBDaoVO vo = new MongoDBDaoVO.Builder().setBaseDao(new ClassInfo(BaseDao.class.getSimpleName(), BaseDao.class.getName())).setDao(new ClassInfo("TestDao", "com.test.dao.TestDao")).setEntity(TestEntity.class).build();
+ String result = FreemarkerUtils.generateInputStream(vo, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\dao\\mongoDBDao.template"));
+ System.out.println(result);
+ }
+
+
+ private static void generateMybatisDao() throws Exception {
+
+ MyBatisDBDaoVO vo = new MyBatisDBDaoVO
+ .Builder()
+ .setEntity(TestEntity2.class)
+ .setBaseMapper(new ClassInfo("BaseMapper", "com.test.mapper.BaseMapper"))
+ .setMapper(new ClassInfo("TestEntity2Mapper", "com.test.mapper.TestEntity2Mapper"))
+ .build();
+
+
+// String result = FreemarkerUtils.generateInputStream(vo, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\dao\\mybatisDBDao.template"));
+ String result = FreemarkerUtils.generateInputStream(vo, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\dao\\mybatisDBXML.ftl"));
+ System.out.println(result);
+ }
+
+ private static void generateException() throws Exception {
+ ExceptionVO vo = new ExceptionVO.Builder()
+ .setEntity(Entity.class)
+ .setPackageName("com.test.exception").build();
+ String result = FreemarkerUtils.generateInputStream(vo, GeneratorTest.class.getClassLoader().getResourceAsStream("generater\\exception.template"));
+ System.out.println(result);
+ }
+
+ private static void generater() throws Exception {
+ GeneraterManager.getInstance().init("D:\\workspace\\server\\utils\\src\\test\\java", TestEntity.class);
+ GeneraterManager.getInstance().createWholeFunction(new DaoGeneraterParams("com.generater.temp.dao", null), null, null);
}
--
Gitblit v1.8.0