1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374 |
- package com.szwl;
- import com.baomidou.mybatisplus.annotation.IdType;
- import com.baomidou.mybatisplus.generator.AutoGenerator;
- import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
- import com.baomidou.mybatisplus.generator.config.GlobalConfig;
- import com.baomidou.mybatisplus.generator.config.PackageConfig;
- import com.baomidou.mybatisplus.generator.config.StrategyConfig;
- import com.baomidou.mybatisplus.generator.config.rules.DateType;
- import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
- import lombok.extern.slf4j.Slf4j;
- import org.junit.jupiter.api.Test;
- import org.springframework.boot.test.context.SpringBootTest;
- @Slf4j
- @SpringBootTest
- class AutoGeneratorTests {
- @Test
- void contextLoads() {
- AutoGenerator autoGenerator = new AutoGenerator();
- //获取工程的根目录
- String projectPath = System.getProperty("user.dir");//得到当前项目的路径
- //全局配置
- GlobalConfig globalConfig = new GlobalConfig();
- globalConfig.setAuthor("wuhs")
- .setOutputDir(projectPath + "/src/main/java")//设置输出路径,项目所在路径
- .setFileOverride(true)//设置文件覆盖
- .setOpen(false)//是否生成代码后打开本地目录
- .setSwagger2(true) //实体属性 Swagger2 注解
- .setIdType(IdType.ASSIGN_ID)//设置主键生成策略
- .setServiceName("%sService")//service接口的名称
- .setDateType(DateType.ONLY_DATE)
- .setBaseResultMap(true)//基本结果集合
- .setBaseColumnList(true)//设置基本的列
- .setControllerName("%sController");
- //配置数据源
- //配置数据源
- DataSourceConfig dataSourceConfig = new DataSourceConfig();
- dataSourceConfig.setDriverName("com.mysql.cj.jdbc.Driver")
- .setUrl("jdbc:mysql://rm-wz995mu26a1479kz0so.mysql.rds.aliyuncs.com:3306/szwl-test")
- .setUsername("root").setPassword("sunzee@020");
- //策略配置
- StrategyConfig strategyConfig = new StrategyConfig();
- strategyConfig
- // .setCapitalMode(true)//设置全局大写命名
- .setInclude(new String[]{
- "t_hot_update"
- })//只会生成该表
- .setEntityLombokModel(true)//实体类生成之后自动添加lombok注解
- .setNaming(NamingStrategy.underline_to_camel)//数据库表映射到实体的命名策略
- .setColumnNaming(NamingStrategy.underline_to_camel)//将数据库中带下划线的转成驼峰规则
- //.setTablePrefix("tbl_")//设置表名前缀
- .setRestControllerStyle(true)//controller是否使用restful风格
- ;
- //包名配置
- PackageConfig packageConfig = new PackageConfig();
- packageConfig.setParent("com.szwl")
- .setMapper("mapper")//包路径
- .setService("service")
- .setController("controller")
- .setEntity("model.entity")
- // .setXml()
- ;
- autoGenerator.setGlobalConfig(globalConfig).setDataSource(dataSourceConfig)
- .setStrategy(strategyConfig).setPackageInfo(packageConfig);
- autoGenerator.execute();
- }
- }
|