一、简介
我们在用MyBatis里，很多时间有这样一个需求：bean里有个属性是枚举，在DB存储时我们想存的枚举的代号，从DB拿出来时想直接映射成目标枚举类型，也即代号字段与Java枚举类的相互类型转换。
当然，你可以为每个枚举写一个MyEnumTypeHandler，但问题是要为每个类都写一个TypeHandler，过于繁琐。
有了泛型，一个通用的TypeHandler直接搞定。
二、源码
源码详见：spring-mybatis-test
2.1 EnumTypeHandler类
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | package com.adu.spring_test.mybatis.typehandler;import java.sql.CallableStatement;import java.sql.PreparedStatement;import java.sql.ResultSet;import java.sql.SQLException;import com.adu.spring_test.mybatis.util.CodeEnumUtil;import org.apache.ibatis.type.BaseTypeHandler;import org.apache.ibatis.type.JdbcType;import com.adu.spring_test.mybatis.enums.CodeBaseEnum;/** * mapper里字段到枚举类的映射。 * 用法一: * 入库：#{enumDataField, typeHandler=com.adu.spring_test.mybatis.typehandler.EnumTypeHandler} * 出库： * <resultMap> * <result property="enumDataField" column="enum_data_field" javaType="com.xxx.MyEnum" typeHandler="com.adu.spring_test.mybatis.typehandler.EnumTypeHandler"/> * </resultMap> * * 用法二： * 1）在mybatis-config.xml中指定handler: * <typeHandlers> * <typeHandler handler="com.adu.spring_test.mybatis.typehandler.EnumTypeHandler" javaType="com.xxx.MyEnum"/> * </typeHandlers> * 2)在MyClassMapper.xml里直接select/update/insert。 */public class EnumTypeHandler<E extends Enum<?> & CodeBaseEnum> extends BaseTypeHandler<CodeBaseEnum> { private Class<E> clazz; public EnumTypeHandler(Class<E> enumType) { if (enumType == null) throw new IllegalArgumentException("Type argument cannot be null"); this.clazz = enumType; } @Override public void setNonNullParameter(PreparedStatement ps, int i, CodeBaseEnum parameter, JdbcType jdbcType) throws SQLException { ps.setInt(i, parameter.code()); } @Override public E getNullableResult(ResultSet rs, String columnName) throws SQLException { return CodeEnumUtil.codeOf(clazz, rs.getInt(columnName)); } @Override public E getNullableResult(ResultSet rs, int columnIndex) throws SQLException { return CodeEnumUtil.codeOf(clazz, rs.getInt(columnIndex)); } @Override public E getNullableResult(CallableStatement cs, int columnIndex) throws SQLException { return CodeEnumUtil.codeOf(clazz, cs.getInt(columnIndex)); }} |
2.2 CodeBaseEnum类
1 2 3 4 5 | package com.adu.spring_test.mybatis.enums;public interface CodeBaseEnum { int code();} |
2.3 CodeEnumUtil类
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | package com.adu.spring_test.mybatis.util;import com.adu.spring_test.mybatis.enums.CodeBaseEnum;public class CodeEnumUtil { /** * @param enumClass * @param code * @param <E> * @return */ public static <E extends Enum<?> & CodeBaseEnum> E codeOf(Class<E> enumClass, int code) { E[] enumConstants = enumClass.getEnumConstants(); for (E e : enumConstants) { if (e.code() == code) return e; } return null; }} |