更新代码生成2.0

Author: dolyw

pom.xml

@@ -31,6 +31,7 @@
         <generator.version>1.3.6</generator.version>
         <guava.version>23.0</guava.version>
         <commons-lang3.version>3.7</commons-lang3.version>
+        <antlr.version>3.2</antlr.version>
     </properties>
 
     <dependencies>
@@ -111,6 +112,12 @@
             <version>${generator.version}</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.antlr</groupId>
+            <artifactId>antlr</artifactId>
+            <version>${antlr.version}</version>
+            <scope>test</scope>
+        </dependency>
 
         <!-- Guava工具包 -->
         <dependency>

…com/example/generator/CodeGenerator.java …/example/generator/CodeGeneratorOne.javasrc/test/java/com/example/generator/CodeGenerator.java renamed to src/test/java/com/example/generator/CodeGeneratorOne.java src/test/java/com/example/generator/CodeGenerator.java renamed to src/test/java/com/example/generator/CodeGeneratorOne.java

@@ -15,9 +15,9 @@
 import java.util.*;
 
 /**
- * 代码生成(根据表生成对应的Model、Mapper、Service、Controller)
+ * 代码生成1.0(根据表生成对应的Model、Mapper、Service、Controller)(推荐使用2.0)
  */
-public class CodeGenerator {
+public class CodeGeneratorOne {
     // Controller生成类型 REST POST
     private static final String CONTROLLER_TYPE = "REST";
 
@@ -45,7 +45,7 @@ public class CodeGenerator {
     // 项目在硬盘上的基础路径
     private static final String PROJECT_PATH = System.getProperty("user.dir");
     // 模板位置
-    private static final String TEMPLATE_FILE_PATH = PROJECT_PATH + "/src/test/resources/template";
+    private static final String TEMPLATE_FILE_PATH = PROJECT_PATH + "/src/test/resources/template/one";
     // Java文件路径
     private static final String JAVA_PATH = "/src/main/java";
     // 资源文件路径
@@ -58,7 +58,7 @@ public class CodeGenerator {
     private static final String PACKAGE_PATH_CONTROLLER = packageConvertPath(CONTROLLER_PACKAGE);
 
     // 作者
-    private static final String AUTHOR = "CodeGenerator";
+    private static final String AUTHOR = "CodeGeneratorOne";
     // 时间
     private static final String DATE = new SimpleDateFormat("yyyy/MM/dd HH:ss").format(new Date());
 
@@ -233,7 +233,6 @@ public static void genController(String tableName, String modelName) {
         } catch (Exception e) {
             throw new RuntimeException("生成Controller失败(请检查数据库是否连接正常及表名是否正确以及权限是否缺失)", e);
         }
-
     }
 
     private static freemarker.template.Configuration getConfiguration() throws IOException {
@@ -250,7 +249,6 @@ private static String tableNameConvertLowerCamel(String tableName) {
 
     private static String tableNameConvertUpperCamel(String tableName) {
         return CaseFormat.LOWER_UNDERSCORE.to(CaseFormat.UPPER_CAMEL, tableName.toLowerCase());
-
     }
 
     private static String tableNameConvertMappingPath(String tableName) {

src/test/java/com/example/generator/CodeGeneratorTwo.java

@@ -0,0 +1,58 @@
+package com.example.generator;
+
+import com.uframe.generator.Generator;
+import com.uframe.generator.GeneratorFacade;
+import com.uframe.generator.GeneratorProperties;
+
+/**
+ * 代码生成2.0(根据表生成对应模板的的Model、Mapper、Service、Controller、JS、JSP)
+ * @author Wang926454
+ * @date 2019/4/1 18:38
+ */
+public class CodeGeneratorTwo {
+
+    /**
+     * 项目在硬盘上的基础路径
+     */
+    private static final String PROJECT_PATH = System.getProperty("user.dir");
+    /**
+     * 生成代码模板位置
+     */
+    private static final String TEMPLATE_FILE_PATH = PROJECT_PATH + "/src/test/resources/template/two/LayUI";
+
+    /**
+     * 有三点需要引起特别的注意
+     * (1)数据库连接必须配置
+     * (2)表无主键，无法生成
+     * (3)外键关联的表无读权限，无法生成
+     */
+    public static void main(String[] args) throws Exception {
+        // 配置表名
+        genCode("user", "role", "permission");
+    }
+
+    /**
+     * 通过表名称生成代码，Model名称通过解析数据表名称获得，下划线转大驼峰的形式
+     * @param tableNames 表名称...
+     */
+    public static void genCode(String... tableNames) {
+        // GeneratorFacade
+        GeneratorFacade generatorFacade = new GeneratorFacade();
+        // 开始执行
+        try {
+            for (String tableName : tableNames) {
+                // 删除旧文件
+                generatorFacade.deleteByTable(tableName, TEMPLATE_FILE_PATH);
+                // 生成新文件
+                generatorFacade.generateByTable(tableName, TEMPLATE_FILE_PATH);
+            }
+            System.out.println("----- 生成成功 -----");
+            // 打开生成目录
+            // Runtime.getRuntime().exec("cmd.exe /c start " + GeneratorProperties.getRequiredProperty("outRoot"));
+            Runtime.getRuntime().exec("cmd.exe /c start " + Generator.getOutRootDir());
+        } catch (Exception e) {
+            System.out.println("----- 生成失败 请检查数据库是否连接正常及表名是否正确以及权限是否缺失 -----");
+            e.printStackTrace();
+        }
+    }
+}