优化自动生成代码

qomo-kn-cost-manage/src/test/java/com/zzsmart/qomo/kn/cost/manage/ApplicationTests.java

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