Browse Source

优化自动生成代码

wangqiong 2 years ago
parent
commit
8f6458d800

+ 16
- 7
qomo-kn-cost-manage/src/test/java/com/zzsmart/qomo/kn/cost/manage/ApplicationTests.java View File

3
 import com.baomidou.mybatisplus.annotation.IdType;
3
 import com.baomidou.mybatisplus.annotation.IdType;
4
 import com.baomidou.mybatisplus.generator.FastAutoGenerator;
4
 import com.baomidou.mybatisplus.generator.FastAutoGenerator;
5
 import com.baomidou.mybatisplus.generator.config.OutputFile;
5
 import com.baomidou.mybatisplus.generator.config.OutputFile;
6
+import com.baomidou.mybatisplus.generator.config.TemplateType;
6
 import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
7
 import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
7
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
8
 import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
8
 import org.junit.Test;
9
 import org.junit.Test;
23
         FastAutoGenerator.create("jdbc:mysql://121.40.189.20:3306/"+"costmanage"+"?&useSSL=true&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai","root","YMzc157#")
24
         FastAutoGenerator.create("jdbc:mysql://121.40.189.20:3306/"+"costmanage"+"?&useSSL=true&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai","root","YMzc157#")
24
                 .globalConfig(builder -> {
25
                 .globalConfig(builder -> {
25
                     builder.author(" ")
26
                     builder.author(" ")
27
+                            .fileOverride()//是否覆盖
26
                             //启用swagger
28
                             //启用swagger
27
                             //.enableSwagger()
29
                             //.enableSwagger()
28
                             //指定输出目录
30
                             //指定输出目录
40
                             .pathInfo(Collections.singletonMap(OutputFile.mapperXml,System.getProperty("user.dir")+"/src/main/java/com/zzsmart/qomo/kn/cost/manage/mapper/xml"));
42
                             .pathInfo(Collections.singletonMap(OutputFile.mapperXml,System.getProperty("user.dir")+"/src/main/java/com/zzsmart/qomo/kn/cost/manage/mapper/xml"));
41
                 })
43
                 })
42
                 .strategyConfig(builder -> {
44
                 .strategyConfig(builder -> {
43
-                    builder.addInclude("7525_replacement_cost")// todo 设置需要生成的表名
45
+                    builder.addInclude("authority")// todo 设置需要生成的表名
44
                             //.addTablePrefix("sys_")//设置表前缀过滤
46
                             //.addTablePrefix("sys_")//设置表前缀过滤
45
                             .entityBuilder()
47
                             .entityBuilder()
46
                             .enableLombok()
48
                             .enableLombok()
49
                             .columnNaming(NamingStrategy.underline_to_camel)//表字段映射实体属性命名规则:默认null,不指定按照naming执行
51
                             .columnNaming(NamingStrategy.underline_to_camel)//表字段映射实体属性命名规则:默认null,不指定按照naming执行
50
                             .idType(IdType.AUTO)//添加全局主键类型
52
                             .idType(IdType.AUTO)//添加全局主键类型
51
                             .formatFileName("%s")//格式化实体名称,%s取消首字母I,
53
                             .formatFileName("%s")//格式化实体名称,%s取消首字母I,
54
+                            .enableTableFieldAnnotation()
55
+                            .enableColumnConstant()
52
                             .mapperBuilder()
56
                             .mapperBuilder()
53
                             .enableMapperAnnotation()//开启mapper注解
57
                             .enableMapperAnnotation()//开启mapper注解
54
                             .enableBaseResultMap()//启用xml文件中的BaseResultMap 生成
58
                             .enableBaseResultMap()//启用xml文件中的BaseResultMap 生成
55
                             .enableBaseColumnList()//启用xml文件中的BaseColumnList
59
                             .enableBaseColumnList()//启用xml文件中的BaseColumnList
56
                             .formatMapperFileName("%sMapper")//格式化Dao类名称
60
                             .formatMapperFileName("%sMapper")//格式化Dao类名称
57
-                            .formatXmlFileName("%sMapper");//格式化xml文件名称
58
-//                            .serviceBuilder()
59
-//                            .formatServiceFileName("%sService")//格式化 service 接口文件名称
60
-//                            .formatServiceImplFileName("%sServiceImpl")//格式化 service 接口文件名称
61
-//                            .controllerBuilder()
62
-//                            .enableRestStyle();
61
+                            .formatXmlFileName("%sMapper")//格式化xml文件名称
62
+                            .serviceBuilder()
63
+                            .formatServiceFileName("%sService")//格式化 service 接口文件名称
64
+                            .formatServiceImplFileName("%sServiceImpl")//格式化 service 接口文件名称
65
+                            .controllerBuilder()
66
+                            .enableRestStyle();
63
                 })
67
                 })
64
                 // 使用Freemarker引擎模板,默认的是Velocity引擎模板
68
                 // 使用Freemarker引擎模板,默认的是Velocity引擎模板
69
+                .templateConfig(builder -> {
70
+                    builder.disable(TemplateType.CONTROLLER)
71
+                            .disable(TemplateType.SERVICE)
72
+                            .disable(TemplateType.SERVICEIMPL);
73
+                })
65
                 .templateEngine(new FreemarkerTemplateEngine())
74
                 .templateEngine(new FreemarkerTemplateEngine())
66
                 .execute();
75
                 .execute();
67
 
76