使用 mybatis-generator 發現數據庫類型是 tinyint(4) , 生成 model 時字段類型是 Byte ,使用的時候有點不便html
數據庫的類型和 Model 中 Java 類型的關係是由 JavaTypeResolver 控制的, 查資料發現能夠在生成配置中指定表中的某一個字段的 JavaType, 但不能全局配置.java
若是想改某一個表的話某個字段, 只須要在 table 標籤下指定 <columnOverride column="" javaType=""/>
便可git
若是想要改全局就沒那麼容易了.github
官方文檔:http://www.mybatis.org/genera...sql
查看配置文檔發現有一個 javaTypeResolver
標籤, 有一個生成器的默認實現數據庫
package org.mybatis.generator.internal.types; import java.math.BigDecimal; import java.sql.Types; import java.util.Date; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Properties; import org.mybatis.generator.api.IntrospectedColumn; import org.mybatis.generator.api.JavaTypeResolver; import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType; import org.mybatis.generator.config.Context; import org.mybatis.generator.config.PropertyRegistry; import org.mybatis.generator.internal.util.StringUtility; /** * * @author Jeff Butler */ public class JavaTypeResolverDefaultImpl implements JavaTypeResolver { protected List<String> warnings; protected Properties properties; protected Context context; protected boolean forceBigDecimals; protected Map<Integer, JdbcTypeInformation> typeMap; public JavaTypeResolverDefaultImpl() { super(); properties = new Properties(); typeMap = new HashMap<Integer, JdbcTypeInformation>(); typeMap.put(Types.ARRAY, new JdbcTypeInformation("ARRAY", //$NON-NLS-1$ new FullyQualifiedJavaType(Object.class.getName()))); typeMap.put(Types.BIGINT, new JdbcTypeInformation("BIGINT", //$NON-NLS-1$ new FullyQualifiedJavaType(Long.class.getName()))); typeMap.put(Types.BINARY, new JdbcTypeInformation("BINARY", //$NON-NLS-1$ new FullyQualifiedJavaType("byte[]"))); //$NON-NLS-1$ typeMap.put(Types.BIT, new JdbcTypeInformation("BIT", //$NON-NLS-1$ new FullyQualifiedJavaType(Boolean.class.getName()))); 等等等...
能夠發現是把數據庫類型和 JavaType 使用 map 映射起來了, 要是能改這個東西就行了
在網上搜索沒有找到合適的擴展點, 在官方倉庫中搜到了一點相關的東西 api
參考:https://github.com/mybatis/ge...mybatis
若是想控制映射關係, 有兩個方案dom
顯然第二個實現起來比較簡單, 咱們的要求不是很複雜, 不用修改 calculateJavaType, 因此能夠直接覆蓋 org.mybatis.generator.internal.types.JavaTypeResolverDefaultImpl#typeMap 中的映射關係ide
首先把 mybatis-generator 源碼拉下來 ,打開 org.mybatis.generator.internal.types 包新建一個類, 繼承原有的 JavaTypeResolverDefaultImpl, 而後在構造中覆蓋原有配置
package org.mybatis.generator.internal.types; import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType; import java.sql.Types; public class MyJavaTypeResolverDefaultImpl extends JavaTypeResolverDefaultImpl { public MyJavaTypeResolverDefaultImpl() { super(); //把數據庫的 TINYINT 映射成 Integer super.typeMap.put(Types.TINYINT, new JdbcTypeInformation("TINYINT", new FullyQualifiedJavaType(Integer.class.getName()))); } }
完成之後總體 install 到本地, 拉下來的版本號是 1.3.7-SNAPSHOT, 注意改一下本身項目依賴的版本號
在生成配置文件中把 javaTypeResolver 修改爲本身的實現類
<javaTypeResolver type="org.mybatis.generator.internal.types.MyJavaTypeResolverDefaultImpl"/>
這樣的話生成代碼時就會把數據庫的 tinyint 變成 Java 中的 Integer