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

feat: add JdbcTypeMapper to customize bean property type

Browse Source
This commit is contained in:
开源海哥 2023-08-08 09:51:01 +08:00
parent 5fa138dee0
commit 2d60efe747
2 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/JdbcDialect.java
View File

@ -25,6 +25,7 @@ import java.util.Map;
/** /**
* 默认方言抽象类。 * 默认方言抽象类。
*
* @author michael * @author michael
*/ */
public abstract class JdbcDialect implements IDialect { public abstract class JdbcDialect implements IDialect {
@ -47,7 +48,7 @@ public abstract class JdbcDialect implements IDialect {
column.setRawLength(columnMetaData.getColumnDisplaySize(i)); column.setRawLength(columnMetaData.getColumnDisplaySize(i));
String jdbcType = columnMetaData.getColumnClassName(i); String jdbcType = columnMetaData.getColumnClassName(i);
column.setPropertyType(JdbcTypeMapping.getType(jdbcType)); column.setPropertyType(JdbcTypeMapping.getType(jdbcType, column.getRawLength()));
column.setAutoIncrement(columnMetaData.isAutoIncrement(i)); column.setAutoIncrement(columnMetaData.isAutoIncrement(i));

19
mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/JdbcTypeMapping.java
View File

@ -29,6 +29,7 @@ public class JdbcTypeMapping {
} }
private static final Map<String, String> mapping = new HashMap<>(); private static final Map<String, String> mapping = new HashMap<>();
private static JdbcTypeMapper mapper;
static { static {
registerMapping("[B", "byte[]"); registerMapping("[B", "byte[]");
@ -46,6 +47,14 @@ public class JdbcTypeMapping {
return mapping; return mapping;
} }
public static JdbcTypeMapper getMapper() {
return mapper;
}
public static void setMapper(JdbcTypeMapper mapper) {
JdbcTypeMapping.mapper = mapper;
}
/** /**
* 当只使用 date 类型来映射数据库的所有 "时间" 类型时，调用此方法 * 当只使用 date 类型来映射数据库的所有 "时间" 类型时，调用此方法
*/ */
@ -56,9 +65,17 @@ public class JdbcTypeMapping {
registerMapping("java.time.LocalDate", "java.util.Date"); registerMapping("java.time.LocalDate", "java.util.Date");
} }
static String getType(String jdbcType) { static String getType(String jdbcType, int length) {
if (mapper != null) {
return mapper.getType(jdbcType, length);
}
String registered = mapping.get(jdbcType); String registered = mapping.get(jdbcType);
return StringUtil.isNotBlank(registered) ? registered : jdbcType; return StringUtil.isNotBlank(registered) ? registered : jdbcType;
} }
public interface JdbcTypeMapper {
String getType(String jdbcType, int length);
}
} }