修改 mybatis-generator 中數據庫類型和 Java 類型的映射關係

使用 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

  1. 編寫一個插件
  2. 擴展 JavaTypeResolverDefaultImpl 並更改 calculateJavaType 方法, 在 TypeResolver 配置本身的實現

顯然第二個實現起來比較簡單, 咱們的要求不是很複雜, 不用修改 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

相關文章
相關標籤/搜索