Browse Source

fixed gitee issues/IHN8Y

= 7 years ago
parent
commit
4f40a14fd3

+ 1 - 1
mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/test/generator/PostgreSQLGenerator.java

@@ -121,7 +121,7 @@ public class PostgreSQLGenerator extends GeneratorTest {
             }
         };
         List<FileOutConfig> focList = new ArrayList<>();
-        focList.add(new FileOutConfig("/templates/entity.java" + ((1 == result) ? ".ftl" : ".vm" )) {
+        focList.add(new FileOutConfig("/templates/dto.java" + ((1 == result) ? ".ftl" : ".vm" )) {
             @Override
             public String outputFile(TableInfo tableInfo) {
                 // 自定义输入文件名称

+ 11 - 0
mybatis-plus-generator/src/test/resources/templates/dto.java.ftl

@@ -0,0 +1,11 @@
+package ${package.Entity};
+
+
+/**
+ * <p>
+ * 测试自定义DTO模板 ${table.comment}
+ * </p>
+ *
+ * 测试注入 ${cfg.abc}
+ */
+

+ 9 - 0
mybatis-plus-generator/src/test/resources/templates/dto.java.vm

@@ -0,0 +1,9 @@
+package ${package.Entity};
+
+
+/**
+ * <p>
+ * 测试自定义DTO模板  $!{table.comment}
+ * </p>
+ * 测试注入 ${cfg.abc}
+ */