一、简介
我们在用MyBatis里，很多时间有这样一个需求：bean里有个属性是枚举，在DB存储时我们想存的枚举的代号，从DB拿出来时想直接映射成目标枚举类型，也即代号字段与Java枚举类的相互类型转换。
当然，你可以为每个枚举写一个MyEnumTypeHandler，但问题是要为每个类都写一个TypeHandler，过于繁琐。
有了泛型，一个通用的TypeHandler直接搞定。
二、源码
2.1 EnumTypeHandler类
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}
* 出库：
*
*
*
*
* 用法二：
* 1)在mybatis-config.xml中指定handler:
*
*
*
* 2)在MyClassMapper.xml里直接select/update/insert。
*/
public class EnumTypeHandler & CodeBaseEnum> extends BaseTypeHandler {
private Class clazz;
public EnumTypeHandler(Class 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类
package com.adu.spring_test.mybatis.enums;
public interface CodeBaseEnum {
int code();
}
2.3 CodeEnumUtil类
package com.adu.spring_test.mybatis.util;
import com.adu.spring_test.mybatis.enums.CodeBaseEnum;
public class CodeEnumUtil {
/**
* @param enumClass
* @param code
* @param
* @return
*/
public static & CodeBaseEnum> E codeOf(Class enumClass, int code) {
E[] enumConstants = enumClass.getEnumConstants();
for (E e : enumConstants) {
if (e.code() == code)
return e;
}
return null;
}
}
