test: 更新测试代码,支持新生成器格式。

This commit is contained in:
Suomm 2023-05-15 20:04:55 +08:00
parent 19069a129a
commit 9165d8d44a
2 changed files with 58 additions and 57 deletions

View File

@ -38,40 +38,36 @@ public class GeneratorTest {
// JdbcTypeMapping.registerMapping(Integer.class, Long.class); // JdbcTypeMapping.registerMapping(Integer.class, Long.class);
GlobalConfig globalConfig = new GlobalConfig(); GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setSourceDir(System.getProperty("user.dir") + "/src/test/java");
// globalConfig.setTablePrefix("tb_");
// globalConfig.setEntityWithLombok(true);
globalConfig.setEntitySupperClass(BaseEntity.class); //设置生成文件目录和根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setBasePackage("com.test");
//设置只生成哪些表 //设置只生成哪些表
globalConfig.addGenerateTable("account", "account_session"); globalConfig.getStrategyConfig()
.addGenerateTable("account", "account_session");
//设置 entity 的包名 //设置生成 entity
globalConfig.setEntityPackage("com.test.entity"); globalConfig.enableEntity()
globalConfig.setEntityClassPrefix("My"); .setWithLombok(true)
globalConfig.setEntityClassSuffix("Entity"); .setClassPrefix("My")
.setClassSuffix("Entity")
.setSupperClass(BaseEntity.class);
//设置 entity 的包名 //设置生成 tableDef
globalConfig.setTableDefGenerateEnable(true); globalConfig.enableTableDef();
globalConfig.setTableDefPackage("com.test.entity.tables");
globalConfig.setTableDefClassPrefix("My");
globalConfig.setTableDefClassSuffix("TableDef");
//是否生成 mapper 默认为 false
globalConfig.setMapperGenerateEnable(true);
globalConfig.setMapperClassPrefix("Flex");
globalConfig.setMapperClassSuffix("Dao");
//设置 mapper 类的包名
globalConfig.setMapperPackage("com.test.mapper");
globalConfig.setMapperSupperClass(MyBaseMapper.class);
//设置生成 mapper
globalConfig.enableMapper()
.setClassPrefix("Flex")
.setClassSuffix("Dao")
.setSupperClass(MyBaseMapper.class);
TableConfig tableConfig = new TableConfig(); TableConfig tableConfig = new TableConfig();
tableConfig.setTableName("account"); tableConfig.setTableName("account");
tableConfig.setUpdateListenerClass(MyUpdateListener.class); tableConfig.setUpdateListenerClass(MyUpdateListener.class);
globalConfig.addTableConfig(tableConfig); globalConfig.getStrategyConfig().addTableConfig(tableConfig);
//可以单独配置某个列 //可以单独配置某个列
@ -79,7 +75,7 @@ public class GeneratorTest {
columnConfig.setColumnName("tenant_id"); columnConfig.setColumnName("tenant_id");
columnConfig.setLarge(true); columnConfig.setLarge(true);
columnConfig.setVersion(true); columnConfig.setVersion(true);
globalConfig.addColumnConfig("account", columnConfig); globalConfig.getStrategyConfig().addColumnConfig("account", columnConfig);
//通过 datasource globalConfig 创建代码生成器 //通过 datasource globalConfig 创建代码生成器
@ -91,39 +87,44 @@ public class GeneratorTest {
@Test @Test
public void testCodeGen() { public void testCodeGen() {
// 配置数据源 //配置数据源
HikariDataSource dataSource = new HikariDataSource(); HikariDataSource dataSource = new HikariDataSource();
dataSource.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/test?characterEncoding=utf-8"); dataSource.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/test?characterEncoding=utf-8");
dataSource.setUsername("root"); dataSource.setUsername("root");
dataSource.setPassword("12345678"); dataSource.setPassword("12345678");
GlobalConfig globalConfig = new GlobalConfig(); GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setSourceDir(System.getProperty("user.dir") + "/src/test/java");
globalConfig.setTablePrefix("sys_");
globalConfig.setBasePackage("com.test");
globalConfig.setEntityWithLombok(true);
globalConfig.setEntitySupperClass(BaseEntity.class); //设置生成文件目录和根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setBasePackage("com.test");
// 设置只生成哪些表 //设置表前缀和只生成哪些表
globalConfig.addGenerateTable("sys_user"); globalConfig.getStrategyConfig()
.setTablePrefix("sys_")
.addGenerateTable("sys_user");
// 设置 entity 的包名 //配置生成 entity
globalConfig.setTableDefGenerateEnable(true); globalConfig.enableEntity()
.setWithLombok(true)
.setSupperClass(BaseEntity.class);
// 是否生成 mapper 默认为 false //配置生成 mapper
globalConfig.setMapperGenerateEnable(true); globalConfig.enableMapper();
// 是否生成 service 默认为 false //配置生成 service
globalConfig.setServiceGenerateEnable(true); globalConfig.enableService();
// 是否生成 serviceImpl 默认为 false //配置生成 serviceImpl
globalConfig.setServiceImplGenerateEnable(true); globalConfig.enableServiceImpl();
// 是否生成 controller 默认为 false //配置生成 controller
globalConfig.setControllerGenerateEnable(true); globalConfig.enableController();
//配置生成 tableDef
globalConfig.enableTableDef();
// 通过 datasource globalConfig 创建代码生成器 //通过 datasource globalConfig 创建代码生成器
Generator generator = new Generator(dataSource, globalConfig); Generator generator = new Generator(dataSource, globalConfig);
// 开始生成代码 //开始生成代码
generator.generate(); generator.generate();
} }

View File

@ -4,14 +4,13 @@ import com.mybatisflex.codegen.Generator;
import com.mybatisflex.codegen.config.GlobalConfig; import com.mybatisflex.codegen.config.GlobalConfig;
import com.mybatisflex.codegen.dialect.IDialect; import com.mybatisflex.codegen.dialect.IDialect;
import com.zaxxer.hikari.HikariDataSource; import com.zaxxer.hikari.HikariDataSource;
import org.junit.Test;
import java.sql.Connection; import java.sql.Connection;
import java.sql.Statement; import java.sql.Statement;
public class SqliteGeneratorTest { public class SqliteGeneratorTest {
// @Test // @Test
public void testGenerator3() { public void testGenerator3() {
//配置数据源 //配置数据源
@ -24,20 +23,21 @@ public class SqliteGeneratorTest {
GlobalConfig globalConfig = new GlobalConfig(); GlobalConfig globalConfig = new GlobalConfig();
globalConfig.setSourceDir(System.getProperty("user.dir") + "/src/test/java");
//配置生成文件目录与根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setBasePackage("com.test");
//设置只生成哪些表 //设置只生成哪些表
globalConfig.addGenerateTable("person"); globalConfig.getStrategyConfig()
.addGenerateTable("person");
//设置 entity 的包名 globalConfig.enableEntity()
globalConfig.setEntityPackage("com.test.entity"); .setWithLombok(true);
//是否生成 mapper 默认为 false //设置生成 mapper
globalConfig.setMapperGenerateEnable(true); globalConfig.enableMapper();
globalConfig.setEntityWithLombok(true);
//设置 mapper 类的包名
globalConfig.setMapperPackage("com.test.mapper");
Generator generator = new Generator(dataSource, globalConfig, IDialect.SQLITE); Generator generator = new Generator(dataSource, globalConfig, IDialect.SQLITE);