MyBatis里字段到枚举类型的转换/映射

一、简介

  我们在用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 ;
     }
}
原作者: 水岩
         
本博客中未标明转载的文章归作者 水岩和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。





  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值