修改 mybatis-generator 中数据库类型和 Java 类型的映射关系

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

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

如果想改某一个表的话某个字段, 只需要在 table 标签下指定  即可

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

官方文档：http://www.mybatis.org/genera...

查看配置文档发现有一个 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 typeMap;

    public JavaTypeResolverDefaultImpl() {
        super();
        properties = new Properties();
        typeMap = new HashMap();

        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 映射起来了, 要是能改这个东西就好了
在网上搜索没有找到合适的扩展点, 在官方仓库中搜到了一点相关的东西

参考：https://github.com/mybatis/ge...

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

1. 编写一个插件
2. 扩展 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 修改成自己的实现类

type="org.mybatis.generator.internal.types.MyJavaTypeResolverDefaultImpl"/>

这样的话生成代码时就会把数据库的 tinyint 变成 Java 中的 Integer