From 17055fd8d36504b79a5def28f5d4b4740faf012d Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期五, 25 三月 2022 13:47:35 +0800
Subject: [PATCH] TDMQ工具集成

---
 src/test/java/com/generater/GeneratorTest.java |   21 ++++++++++++++++-----
 1 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/src/test/java/com/generater/GeneratorTest.java b/src/test/java/com/generater/GeneratorTest.java
index 33d9472..ba1ba09 100644
--- a/src/test/java/com/generater/GeneratorTest.java
+++ b/src/test/java/com/generater/GeneratorTest.java
@@ -3,11 +3,15 @@
 import org.springframework.data.annotation.Id;
 import org.yeshi.utils.FileUtil;
 import org.yeshi.utils.FreemarkerUtils;
-import org.yeshi.utils.generater.SpringComponentGenerater;
+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.Table;
+import org.yeshi.utils.generater.params.AdminGeneraterParams;
+import org.yeshi.utils.generater.params.DaoGeneraterParams;
+import org.yeshi.utils.generater.params.ServiceGeneraterParams;
 import org.yeshi.utils.generater.vo.ExceptionVO;
 import org.yeshi.utils.generater.vo.dao.MongoDBDaoVO;
 import org.yeshi.utils.generater.vo.dao.MyBatisDBDaoVO;
@@ -47,7 +51,7 @@
         //
 //        boolean match = Pattern.matches("^(https://){1}.*(\\.m3u8)$", "https://n1.szjal.cn/20210805/qnDXhXBa/index.m3u8123");
 //        System.out.println(match);
-        generateMybatisDao();
+        generater();
     }
 
     private static void generateAdminPage() throws Exception {
@@ -69,7 +73,7 @@
 
         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(new FormUpdateInfoVO.Builder().setAdminInfo(info).build(), "admin/update.ftl", map);
         FileUtil.saveAsFileByte(result.getBytes("UTF-8"), "C:\\Users\\Administrator\\Documents\\HBuilderProjects\\LayuiTest\\form.html");
@@ -81,7 +85,7 @@
         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().setDao(dao).setEntity(Entity.class).setPackageName("com.test.service").setQuery(query).setService(service).build();
+        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);
     }
@@ -144,9 +148,16 @@
         System.out.println(result);
     }
 
+    private static void generater() throws Exception {
+//        GeneraterManager.getInstance().init("D:\\workspace\\server\\utils\\src\\test\\java", TestEntity.class);
+//        AdminGeneraterParams adminGeneraterParams=new AdminGeneraterParams("com.generater.temp.controller","D:\\workspace\\server\\utils\\src\\test");
+//
+//        GeneraterManager.getInstance().createWholeFunction(new DaoGeneraterParams("com.generater.temp.dao", "D:\\workspace\\server\\utils\\src\\test"), new ServiceGeneraterParams("com.generater.temp.query","com.generater.temp.service","com.generater.temp.service.impl"), adminGeneraterParams);
+    }
+
 
     @Table("tb_test")
-    public static class Entity {
+     static class Entity {
 
         @Id
         @Column(name = "id")

--
Gitblit v1.8.0