一、简介

我们在用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;

}

}