自定义mybatis-generator自动生成代码时的类型转换

一、mybatis-generator问题

大家在使用mybatis的时候，有些时候需要使用mybatis generator来逆向生成mapper和entity层。

这其中一个问题在于，表设计的时候，出于最大程度节省表空间的考虑，不少枚举使用了tinyint类型，但是在进行自动生成的时候，默认生成的java字段是Byte类型，但是我想要的是Integer类型，这就对程序中使用产生了不便。

其他转换也同理，比如decimal想转为float而不是BigDecimal，smallint想转为Integer而不是Short

二、解决方法

继承JavaTypeResolver接口

mybatis-generator中由JavaTypeResolver的实现类org.mybatis.generator.internal.types.JavaTypeResolverDefaultImpl负责这一部分的类型转换，看其中的代码可以看到：

这里默认把TINYINT转为了Byte，所以我们要做的是，自己写一个实现类来替代默认的实现类，如下：

这里自己写一个实现类实现JavaTypeResolver接口，其他代码直接拿过来不变，只把自己需要改的转为类型修改。

generator-config.xml配置修改

最后在generator-config.xml文件中加一行，指向刚才自己写的实现类，如图：

将重写的类包含进运行依赖

完成以上步骤，还不能保证可以成功生成，如果你执行的话，会报以下错误：

cannot instantiate obeject of type，这是说明在生成过程中，这个类无法被实例化，这是因为在使用maven插件执行的时候，并没有将你刚写的类包含进依赖。
所以这里有两种解决方法：
一是将刚才写的类放入mybatis-generator的jar包，二是比较推荐的方法是如果你有独立于全局的自己维护的common包，将类放入传到maven nexus上，然后在pom文件中加一下引用，如图：

这样就可以成功按照自己的需要生成代码了。

谢谢阅读！