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(); } }