admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: 测试代码生成器。

Browse Source
This commit is contained in:
Suomm 2023-07-05 22:54:13 +08:00
parent ba10363c4f
commit dbdcc88959
1 changed files with 69 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
mybatis-flex-codegen/src/test/java/com/mybatisflex/codegen/test/GeneratorTest.java
View File

@ -16,17 +16,22 @@
package com.mybatisflex.codegen.test;
import com.mybatisflex.annotation.KeyType;
import com.mybatisflex.codegen.Generator;
import com.mybatisflex.codegen.config.ColumnConfig;
import com.mybatisflex.codegen.config.GlobalConfig;
import com.mybatisflex.codegen.config.TableConfig;
import com.mybatisflex.codegen.config.TableDefConfig;
import com.mybatisflex.spring.service.impl.CacheableServiceImpl;
import com.zaxxer.hikari.HikariDataSource;
import org.junit.Test;
import java.util.function.UnaryOperator;
public class GeneratorTest {
// @Test
// @Test
public void testCodeGen1() {
//配置数据源
HikariDataSource dataSource = new HikariDataSource();
@ -82,7 +87,7 @@ public class GeneratorTest {
generator.generate();
}
// @Test
// @Test
public void testCodeGen2() {
//配置数据源
HikariDataSource dataSource = new HikariDataSource();
@ -145,4 +150,66 @@ public class GeneratorTest {
generator.generate();
}
@Test
public void testCodeGen3() {
//配置数据源
HikariDataSource dataSource = new HikariDataSource();
dataSource.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/test?characterEncoding=utf-8");
dataSource.setUsername("root");
dataSource.setPassword("12345678");
GlobalConfig globalConfig = new GlobalConfig();
//用户信息表，用于存放用户信息。 -> 用户信息
UnaryOperator<String> tableFormat = (e) -> e.split("，")[0].replace("表", "");
//设置注解生成配置
globalConfig.getJavadocConfig()
.setAuthor("王帅")
.setTableCommentFormat(tableFormat);
//设置生成文件目录和根包
globalConfig.getPackageConfig()
.setSourceDir(System.getProperty("user.dir") + "/src/test/java")
.setMapperXmlPath(System.getProperty("user.dir") + "/src/test/java/resources/mapper")
.setBasePackage("com.test");
ColumnConfig columnConfig = new ColumnConfig();
columnConfig.setColumnName("phonenumber");
columnConfig.setLarge(true);
columnConfig.setKeyType(KeyType.None);
TableConfig tableConfig = new TableConfig();
tableConfig.setTableName("sys_user");
tableConfig.addColumnConfig(columnConfig);
//设置表前缀和只生成哪些表
globalConfig.getStrategyConfig()
.setTablePrefix("sys_")
.setGenerateTable("sys_user")
.setTableConfig(tableConfig);
//配置生成 tableDef
globalConfig.enableTableDef()
.setInstanceSuffix("Def")
.setPropertiesNameStyle(TableDefConfig.NameStyle.LOWER_CAMEL_CASE)
.setOverwriteEnable(true);
// 配置生成 entity
globalConfig.enableEntity()
.setOverwriteEnable(true)
.setWithLombok(true);
// 配置生成 mapper
globalConfig.enableMapper()
.setOverwriteEnable(true)
.setMapperAnnotation(true);
//通过 datasource 和 globalConfig 创建代码生成器
Generator generator = new Generator(dataSource, globalConfig);
//开始生成代码
generator.generate();
}
}