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

发布时间:2019-11-17 发布网站:脚本宝典
脚本宝典收集整理的这篇文章主要介绍了修改 mybatis-generator 中数据库类型和 Java 类型的映射关系脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。

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

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

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

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

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

参考: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 修改成自己的实现类

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

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

脚本宝典总结

以上是脚本宝典为你收集整理的修改 mybatis-generator 中数据库类型和 Java 类型的映射关系全部内容,希望文章能够帮你解决修改 mybatis-generator 中数据库类型和 Java 类型的映射关系所遇到的问题。

如果觉得脚本宝典网站内容还不错,欢迎将脚本宝典推荐好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。