package com.yami.trading; import com.baomidou.mybatisplus.generator.FastAutoGenerator; import com.baomidou.mybatisplus.generator.config.DataSourceConfig; import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert; import com.baomidou.mybatisplus.generator.config.querys.MySqlQuery; import com.baomidou.mybatisplus.generator.config.rules.DateType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import com.baomidou.mybatisplus.generator.keywords.MySqlKeyWordsHandler; // 代码自动生成器 public class MybatisPlusGenerator { // 数据库连接字段配置 private static final String JDBC_URL = "jdbc:mysql://localhost:3306/trading_order?useUnicode=true&characterEncoding=utf8&autoReconnect=true&allowMultiQueries=true"; private static final String JDBC_USER_NAME = "root"; private static final String JDBC_PASSWORD = "root"; // 包名和模块名 private static final String PACKAGE_NAME = "com.yami.trading"; private static final String MODULE_NAME = "chat"; //覆盖文件设置 /* StrategyConfig.Builder() entityBuilder().fileOverride() //下面 例子有 mapperBuilder().fileOverride() serviceBuilder().fileOverride() controllerBuilder().fileOverride()*/ // 表名,多个表使用英文逗号分割 private static final String[] TBL_NAMES = {"t_otc_message_user","t_otc_onlinechat_message"}; // 表名的前缀,从表生成代码时会去掉前缀 private static final String TABLE_PREFIX = "t_"; public static void main(String[] args) { //获取当前工程路径(这里无需修改) //String projectPath = System.getProperty("user.dir"); String projectPath = System.getProperty("user.dir")+"/trading-order-mbg"; /** * 1.数据库配置(设置数据源) 配置数据库连接以及需要使用的字段 */ DataSourceConfig.Builder dataSourceConfigBuilder = new DataSourceConfig.Builder(JDBC_URL, JDBC_USER_NAME, JDBC_PASSWORD) .dbQuery(new MySqlQuery()) .typeConvert(new MySqlTypeConvert()) .keyWordsHandler(new MySqlKeyWordsHandler()); FastAutoGenerator fastAutoGenerator = FastAutoGenerator.create(dataSourceConfigBuilder); /** * 2.全局配置 */ fastAutoGenerator.globalConfig( globalConfigBuilder -> globalConfigBuilder .fileOverride() // 覆盖已生成文件 .disableOpenDir() // 不打开生成文件目录 .outputDir(projectPath + "/src/main/java") // 指定输出目录,注意斜杠的表示 .author("lucas") // 设置注释的作者 .commentDate("yyyy-MM-dd HH:mm:ss") // 设置注释的日期格式 .dateType(DateType.TIME_PACK) // 使用java8新的时间类型 //.enableSwagger() // 开启swagger文档 ); /** 日期类型 DateType DateType.ONLY_DATE 使用 java.util.date包下的 Date DateType.SQL_PACK 使用 java.sql包下的 Date DateType.TIME_PACK 因为会使用 java.time.LocalDateTime jdk1.8以上才支持 (推荐使用) */ /** * 3.包配置 */ fastAutoGenerator.packageConfig( packageConfigBuilder -> packageConfigBuilder .parent(PACKAGE_NAME) // 设置父包名 // .moduleName(MODULE_NAME) // 设置父包模块名 .entity("model") // 设置MVC下各个模块的包名 .mapper("dao") .service("service") .serviceImpl("service.impl") .controller("controller") .xml("xml") // 设置XML资源文件的目录 ); /** * 4.模板配置 */ /* fastAutoGenerator.templateConfig( templateConfigBuilder -> templateConfigBuilder .disable(TemplateType.ENTITY) // 禁用模板 .entity("/templates/entity.java") // 设置实体模板路径(JAVA) .service("/templates/service.java") // 设置service模板路径 .serviceImpl("/templates/serviceImpl.java") // 设置serviceImpl模板路径 .mapper("/templates/mapper.java") // 设置mapper模板路径 .xml("/templates/mapper.xml") // 设置mapperXml模板路径 .controller("/templates/controller.java") // 设置controller模板路径 ); */ /** * 5.注入配置 TODO */ /** * 6.策略配置 */ fastAutoGenerator.strategyConfig( strategyConfigBuilder -> strategyConfigBuilder .enableCapitalMode() // 开启大写命名 .enableSkipView() // 开启跳过视图 .disableSqlFilter() // 禁用sql过滤 .addInclude(TBL_NAMES) // 设置需要生成的表名 .addTablePrefix(TABLE_PREFIX) // 设置过滤表前缀 ); /** * 6.1 Entity策略配置 */ fastAutoGenerator.strategyConfig( strategyConfigBuilder -> strategyConfigBuilder.entityBuilder() .fileOverride()//覆盖 entity //.enableTableFieldAnnotation() // 生成实体时生成字段的注解，包括@TableId注解等--- .naming(NamingStrategy.underline_to_camel) // 数据库表和字段映射到实体的命名策略,为下划线转驼峰 .columnNaming(NamingStrategy.underline_to_camel) //.idType(IdType.AUTO) // 全局主键类型为AUTO(自增) .enableLombok() // 支持lombok开启注解 // .logicDeleteColumnName("deleted") // 逻辑删除字段名(数据库) //.logicDeletePropertyName("deleted") // 逻辑删除属性名(实体) //.addTableFills(new Column("create_time", FieldFill.INSERT)) // 自动填充配置 create_time update_time 两种方式 //.addTableFills(new Property("updateTime", FieldFill.INSERT_UPDATE)) .versionColumnName("version") // 开启乐观锁 .disableSerialVersionUID() // 禁用生成 serialVersionUID，默认值:true .enableChainModel() // 开启实体类链式编程 //.formatFileName("%") // 实体名称格式化为XXXEntity formatFileName("%sEntity") ); /** * 6.2 Controller策略配置 */ fastAutoGenerator.strategyConfig( strategyConfigBuilder -> strategyConfigBuilder.controllerBuilder() .fileOverride()//覆盖文件 .enableRestStyle() // 开启生成@RestController控制器 .enableHyphenStyle() // 开启驼峰转连字符 localhost:8080/hello_id_2 ); /** * 6.3 Service策略配置 格式化service接口和实现类的文件名称，去掉默认的ServiceName前面的I ---- */ fastAutoGenerator.strategyConfig( strategyConfigBuilder -> strategyConfigBuilder.serviceBuilder() .fileOverride()//覆盖文件 .formatServiceFileName("%sService") .formatServiceImplFileName("%sServiceImpl")); /** * 6.4 Mapper策略配置 格式化 mapper文件名,格式化xml实现类文件名称 */ fastAutoGenerator.strategyConfig( strategyConfigBuilder -> strategyConfigBuilder.mapperBuilder() .fileOverride() //覆盖文件 .enableMapperAnnotation() // 开启 @Mapper 注解 .formatMapperFileName("%sMapper") .formatXmlFileName("%sMapper")); /** 7.生成代码 * */ // fastAutoGenerator.templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板 fastAutoGenerator.execute(); } }