目录
- 引言
- 关键代码
- 源码分析
- 记录一坑
- 类型转换的第二种方式
- 完整源码地址
引言
当默认生成的数据类型不满足时,就需要自定义指定要生成的类型
关键代码
FastAutoGenerator.create(url, username, password)
.dataSourceConfig(builder -> {
builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
// 包装类转基本数据类型
IColumnType columnType = typeRegistry.getColumnType(metaInfo);
if (DbColumnType.LONG.equals(columnType)) {
return DbColumnType.BASE_LONG;
} else if (DbColumnType.INTEGER.equals(columnType)) {
return DbColumnType.BASE_INT;
} else if (DbColumnType.BOOLEAN.equals(columnType)) {
return DbColumnType.BASE_BOOLEAN;
}
return columnType;
});
})
.execute();
源码分析
如果不自定义类型转换器 typeConvertHandler 就会是空的,进入else 调用 columnType = typeRegistry.getColumnType(metaInfo);
因此自定义类型转换器时,可以基于默认的基础上来改。
记录一坑
看官方文档时,看到类型转换器就会下意识的整个页面搜索关键字查看使用,但你会神奇的发现,根本就没有具体的使用示例,凉了。。。这时只能去看源码打断点,研究默认的使用,再重写自己的;
如何你再看下指南,就会发现这边居然会有示例。。。
类型转换的第二种方式
可以自定义模板,直接在模板上修改,以下是freemaker模板示例
private <#if field.propertyType == 'int'>Integer<#elseif field.propertyType == 'long'>Long<#elseif field.propertyType == 'boolean'>Boolean<#else>${field.propertyType}</#if> ${field.propertyName};
完整源码地址
https://gitee.com/qjqlzwh/mybatis-plus-code-generator/tree/master