一、简介
我们在用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
;
}
}
|