引言
typeHandlers
{:target="_blank"}
MyBatis 在预处理语句(PreparedStatement)中设置一个参数时,Java对象将通过ps.setInt、ps.setString、ps.setTimeStamp等方法转换成数据库需要的数据
在从结果集(ResultSet)中取出一个值时,将使用rs.getInt、rs.getString、rs.getTimeStamp等方法将数据转换为Java对象
这两类操作通过 类型处理器 完成
类型处理器均实现
TypeHandler<T>
接口,所有基本类型均有对应的类型处理器Enum对应有两个类型处理器,分别为
EnumTypeHandler
、EnumOrdinalTypeHandler
若需将Enum字段映射为字符串,则使用
EnumTypeHandler
。 (默认使用)若需将Enum字段映射为int数值,则使用
EnumOrdinalTypeHandler
然而, EnumOrdinalTypeHandler
局限性非常明显,其映射的数据直接使用枚举值的 ordinal
数值,因此与枚举值定义顺序紧耦合, 本文将解决此问题
一般合理实现方案(针对每一种特定值Enum定义专属TypeHandler)
假设存在一个产品类型枚举类型定义,其产品均对应有特定int值,实现如下
public enum ProductType { AAA(100), BBB(200), CCC(300); private int value; private ProductType(int value) { this.value = value; } public static ProductType fromValue(int value) { for (ProductType productType : ProductType.values()) { if (productType.value == value) { return productType; } } throw new IllegalArgumentException("Cannot create evalue from value: " + value + "!"); } public int toValue() { return value; }}
同时,需要定义对应的 ProductTypeHandler
public class ProductTypeHandler implements TypeHandler{ @Overridepublic void setParameter(PreparedStatement preparedStatement, int i, ProductType productType, JdbcType jdbcType)throws SQLException { preparedStatement.setInt(i, productType.toValue()); } @Overridepublic ProductType getResult(ResultSet resultSet, String s) throws SQLException { return ProductType.fromValue(resultSet.getInt(s)); } @Overridepublic ProductType getResult(ResultSet resultSet, int i) throws SQLException { return ProductType.fromValue(resultSet.getInt(i)); } @Overridepublic ProductType getResult(CallableStatement callableStatement, int i) throws SQLException { return ProductType.fromValue(callableStatement.getInt(i)); }}
最佳实践
同样以上方的 ProductType
举例,枚举类型的定义稍作修改
public enum ProductType implements ValuedEnum { AAA(100), BBB(200), CCC(300); private int value; private ProductType(int value) { this.value = value; } public int getValue() { return value; }}
可以看出,该类型实现了接口 ValuedEnum
,同时去除了 fromValue
类方法, ValuedEnum
接口内容如下,仅声明一个 getValue
方法
public interface ValuedEnum { int getValue();}
不再需要为 ProductType
专属定义类型转换器,使用通用转换器 ValuedEnumTypeHanlder
,使用方式同内置转换器 EnumOrdinalTypeHandler
完全一致。 实现如下
/** * Created by sunlin05 on 2015/7/6. * @author sunlin */public class ValuedEnumTypeHandler> extends BaseTypeHandler { private Class type; private Map map = new HashMap<>(); public ValuedEnumTypeHandler(Class type) { if (type == null) { throw new IllegalArgumentException("Type argument cannot be null"); } this.type = type; E[] enums = type.getEnumConstants(); if (enums == null) { throw new IllegalArgumentException(type.getSimpleName() + " does not represent an enum type."); } for (E e : enums) { ValuedEnum valuedEnum = (ValuedEnum) e; map.put(valuedEnum.getValue(), e); } } @Override public void setNonNullParameter(PreparedStatement ps, int i, E parameter, JdbcType jdbcType) throws SQLException { ValuedEnum valuedEnum = (ValuedEnum) parameter; ps.setInt(i, valuedEnum.getValue()); } @Override public E getNullableResult(ResultSet rs, String columnName) throws SQLException { int i = rs.getInt(columnName); if (rs.wasNull()) { return null; } else { return getValuedEnum(i); } } @Override public E getNullableResult(ResultSet rs, int columnIndex) throws SQLException { int i = rs.getInt(columnIndex); if (rs.wasNull()) { return null; } else { return getValuedEnum(i); } } @Override public E getNullableResult(CallableStatement cs, int columnIndex) throws SQLException { int i = cs.getInt(columnIndex); if (cs.wasNull()) { return null; } else { return getValuedEnum(i); } } private E getValuedEnum(int value) { try { return map.get(value); } catch (Exception ex) { throw new IllegalArgumentException( "Cannot convert " + value + " to " + type.getSimpleName() + " by value.", ex); } }}
该实现参考 EnumOrdinalTypeHandler
源码,核心逻辑见构造器,加注释说明如下
// 获取所有枚举值E[] enums = type.getEnumConstants();for (E e : enums) { // 类型转换为ValuedEnum接口对象 ValuedEnum valuedEnum = (ValuedEnum) e; // 通过getValue()方法获取枚举值对应的Value int值,通过map记录映射关系 map.put(valuedEnum.getValue(), e);}