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

时间:2024-04-08 13:09:29

一、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负责这一部分的类型转换,看其中的代码可以看到:
自定义mybatis-generator自动生成代码时的类型转换
这里默认把TINYINT转为了Byte,所以我们要做的是,自己写一个实现类来替代默认的实现类,如下:
自定义mybatis-generator自动生成代码时的类型转换
这里自己写一个实现类实现JavaTypeResolver接口,其他代码直接拿过来不变,只把自己需要改的转为类型修改。

generator-config.xml配置修改

最后在generator-config.xml文件中加一行,指向刚才自己写的实现类,如图:
自定义mybatis-generator自动生成代码时的类型转换

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

完成以上步骤,还不能保证可以成功生成,如果你执行的话,会报以下错误:
自定义mybatis-generator自动生成代码时的类型转换
cannot instantiate obeject of type,这是说明在生成过程中,这个类无法被实例化,这是因为在使用maven插件执行的时候,并没有将你刚写的类包含进依赖。
所以这里有两种解决方法:
一是将刚才写的类放入mybatis-generator的jar包,二是比较推荐的方法是如果你有独立于全局的自己维护的common包,将类放入传到maven nexus上,然后在pom文件中加一下引用,如图:
自定义mybatis-generator自动生成代码时的类型转换

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

谢谢阅读!