使用 mybatis-generator 发现数据库类型是 tinyint(4) , 生成 model 时字段类型是 Byte ,使用的时候有点不便

数据库的类型和 Model 中 Java 类型的关系是由 JavaTypeResolver 控制的, 查资料发现可以在生成配置中指定表中的某一个字段的 JavaType, 但不能全局配置.

如果想改某一个表的话某个字段, 只需要在 table 标签下指定 <columnOverride column="" javaType=""/> 即可

如果想要改全局就没那么容易了.

官方文档：...

查看配置文档发现有一个 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 映射起来了, 要是能改这个东西就好了

在网上搜索没有找到合适的扩展点, 在官方仓库中搜到了一点相关的东西

参考：...

如果想控制映射关系, 有两个方案

编写一个插件扩展 JavaTypeResolverDefaultImpl 并更改 calculateJavaType 方法, 在 TypeResolver 配置自己的实现

显然第二个实现起来比较简单, 我们的要求不是很复杂, 不用修改 calculateJavaType, 所以可以直接覆盖 org.mybatis.generator.internal.types.JavaTypeResolverDefaultImpl#typeMap 中的映射关系

首先把 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