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

fixed: codegen error for "blob" type

Browse Source
This commit is contained in:
开源海哥 2023-03-29 12:09:24 +08:00
parent 8bfdf1c0f6
commit c2ee7cf042
3 changed files with 36 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,7 +38,10 @@ public abstract class JdbcDialect implements IDialect {
for (int i = 1; i <= columnCount; i++) {
Column column = new Column();
column.setName(columnMetaData.getColumnName(i));
column.setPropertyType(columnMetaData.getColumnClassName(i));
String jdbcType = columnMetaData.getColumnClassName(i);
column.setPropertyType(JdbcTypeMapping.getType(jdbcType));
column.setAutoIncrement(columnMetaData.isAutoIncrement(i));
//注释

28
mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/JdbcTypeMapping.java Normal file
View File

@ -0,0 +1,28 @@
package com.mybatisflex.codegen.dialect;
import com.mybatisflex.core.util.StringUtil;
import java.util.HashMap;
import java.util.Map;
public class JdbcTypeMapping {
private static Map<String, String> mapping = new HashMap<>();
static {
mapping.put("[B", "byte[]");
}
public void registerMapping(Class<?> from, Class<?> to) {
mapping.put(from.getName(), to.getName());
}
public void registerMapping(String from, String to) {
mapping.put(from, to);
}
static String getType(String jdbcType) {
String registered = mapping.get(jdbcType);
return StringUtil.isNotBlank(registered) ? registered : jdbcType;
}
}

5
mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Column.java
View File

@ -203,7 +203,10 @@ public class Column {
List<String> importClasses = new ArrayList<>();
//lang 包不需要显式导入
if (!propertyType.startsWith("java.lang.")) {
if (!propertyType.startsWith("java.lang.")
&& !"byte[]".equals(propertyType)
&& !"Byte[]".equals(propertyType)
) {
importClasses.add(propertyType);
}