java long 对应mybati类型_修改 mybatis-generator 中数据库类型和 Java 类型的映射关系...

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

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

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

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

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

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

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

编写一个插件

扩展 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 修改成自己的实现类

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