MyBatis自定义范例处理器 TypeHandler（通俗易懂，效率腾飞），处理jsonb、 ...

一 使用场景

在项目开发过程中，我们常常碰到这样的情况：Java 对象中的数据范例与数据库中的字段范例不一致。这时，我们需要在保存数据到数据库和从数据库检索数据时进行范例转换。例如：

• 对于一些数据库特有的数据范例（如 PostgreSQL 的 jsonb 或数组范例），这些范例可能不被 MyBatis 默认支持，因此需要特殊处理。
  
• 在 Java 实体（JavaBean）中，可能有一些字段是罗列（Enum）范例或特殊范例，而在数据库中，这些数据可能需要存储为字符串（String）或整数（Integer）。
  
• 同样，你的 Java 实体可能有日期（Date）范例的字段，而在数据库中，相应的字段可能是以字符串（varchar）形式存储的日期。
  

这些范例不匹配的情况会导致大量的手动数据范例转换，这不仅贫苦，而且轻易出错。为了解决这个问题，MyBatis 提供了一种功能强大的机制：TypeHandler 范例处理器。通过实现和使用范例处理器，我们可以自动化地进行数据范例转换，简化代码，提高开发效率。范例处理器使得 MyBatis 能够智能地处理那些它默认不支持的数据库字段范例，同时也方便了开发者在复杂数据范例和数据库范例之间进行无缝转换。
二 范例处理器 TypeHandler简介

在 MyBatis 中，TypeHandler（范例处理器）的主要作用是帮助我们在 Java 代码中使用的数据范例（JavaType）和数据库中的数据范例（JdbcType）之间进行转换。这就像是在 Java 世界和数据库世界之间搭建了一座桥梁。

• 当你需要把数据从 Java 发送到数据库时（比如，插入或更新数据），TypeHandler 确保 Java 范例的数据能够转换成数据库能够理解的格式。这个过程涉及到使用 PreparedStatement，它是一种预编译的 SQL 语句。TypeHandler 负责把 Java 范例的数据精确地放置到 SQL 语句的参数中。
  
• 当你从数据库获取数据时（比如，查询操作），TypeHandler 确保从数据库中获取的数据（通过 ResultSet 或 CallableStatement）能够转换成 Java 步伐中能够使用的格式。这样，你就可以在 Java 步伐中方便地处理数据库返回的数据。
  

紧张的一点是，MyBatis 已经内置了许多常见根本范例（如整数、字符串等）的范例处理器。这意味着对于这些根本数据范例，MyBatis 能够自动进行 Java 范例和数据库范例之间的转换，你无需做额外工作。
但是，假如你需要处理一些特殊的数据范例（这些范例可能不是根本范例，比如某种特定格式的字符串，大概是你自定义的复杂范例），MyBatis 就无法直接处理了。在这种情况下，你就需要自定义范例处理器。通过自定义范例处理器，你可以指定怎样将这些特殊的 Java 范例数据转换为数据库可以理解的范例，反之亦然。
三 自定义 TypeHandler

TypeHandler<T> 接口在 MyBatis 中起着桥梁的作用，它连接了 Java 步伐中的数据范例和数据库中的数据范例。这个接口确保了你在 Java 代码中使用的数据范例可以精确地转换成数据库能理解的格式，反之亦然。简单来说，它就像是一个翻译器，帮助 Java 代码和数据库之间进行数据交换。

1. public interface TypeHandler<T> {
3.     /**
4.      * 设置 PreparedStatement 的指定参数。
5.      *
6.      * @param ps        PreparedStatement 对象。
7.      * @param index     参数在 PreparedStatement 中的位置。
8.      * @param parameter 要设置的参数值。
9.      * @param jdbcType  JDBC 类型。这是一个可选参数，可以用来控制设置参数时的行为。
10.      * @throws SQLException 如果在设置参数时发生 SQL 异常。
11.      */
12.     void setParameter(PreparedStatement ps, int index, T parameter, JdbcType jdbcType) throws SQLException;
14.     /**
15.      * 从 ResultSet 中获取数据并转换为 Java 类型。
16.      *
17.      * @param rs        ResultSet 对象。
18.      * @param columnName 要获取的数据的列名。
19.      * @return 转换后的 Java 类型数据。
20.      * @throws SQLException 如果在获取数据时发生 SQL 异常。
21.      */
22.     T getResult(ResultSet rs, String columnName) throws SQLException;
24.     /**
25.      * 从 ResultSet 中获取数据并转换为 Java 类型。
26.      *
27.      * @param rs         ResultSet 对象。
28.      * @param columnIndex 要获取的数据的列索引。
29.      * @return 转换后的 Java 类型数据。
30.      * @throws SQLException 如果在获取数据时发生 SQL 异常。
31.      */
32.     T getResult(ResultSet rs, int columnIndex) throws SQLException;
34.     /**
35.      * 从 CallableStatement 中获取数据并转换为 Java 类型。
36.      *
37.      * @param cs         CallableStatement 对象。
38.      * @param columnIndex 要获取的数据的列索引。
39.      * @return 转换后的 Java 类型数据。
40.      * @throws SQLException 如果在获取数据时发生 SQL 异常。
41.      */
42.     T getResult(CallableStatement cs, int columnIndex) throws SQLException;
43. }

复制代码

接口中的方法分别处理不同的数据转换场景：

• setParameter(PreparedStatement ps, int index, T parameter, JdbcType jdbcType)
  
  
  
  • 当你在 Java 代码中实行一个 SQL 语句并且需要向这个语句中传入参数时，这个方法就发挥作用了。
    
  • 它告诉 MyBatis 怎样将 Java 范例的数据（parameter）转换成数据库能理解的格式，并把它放在 SQL 语句的精确位置（index）。
    
  
  
• getResult(ResultSet rs, String columnName)
  
  
  
  • 当你实行了一个 SQL 查询并从数据库得到结果集（ResultSet）时，这个方法帮助你把结果集中某一列的数据取出，并转换成 Java 范例的数据。
    
  • 你告诉它具体要转换哪一列（columnName），它就会处理这一列的数据。
    
  
  
• getResult(ResultSet rs, int columnIndex)
  
  
  
  • 这个方法和上一个方法类似，但它是通过列的索引（位置）而不是列的名称来获取数据的。
    
  • 它也是用来把结果集中的数据转换成 Java 范例的数据。
    
  
  
• getResult(CallableStatement cs, int columnIndex)
  
  
  
  • 这个方法用在存储过程的场景。存储过程是在数据库中实行的一系列操作，它可以返回多个结果。
    
  • 当你调用一个存储过程并想要处理返回的结果时，这个方法就会根据你指定的列索引来获取并转换这些结果。
    
  
  

总的来说，TypeHandler<T> 就像是一个双向翻译器，它确保 Java 步伐和数据库在数据范例上的沟通是流畅和精确的。
四 创建自定义处理器

实际开发中，我们可以继承 org.apache.ibatis.type.BaseTypeHandler 范例来实现自定义范例处理器。
这个范例是抽象范例，实现了 TypeHandler 的方法进行通用流程的封装，做了异常处理，并定义了几个类似的抽象方法，如下所示。继承 BaseTypeHandler 范例可以极大地低落开发难度。
下面定义三种常用的自定义处理器：
自定义处理器1：数组范例

1. public class IntegerArrayTypeHandler extends BaseTypeHandler<Integer[]> {
3.     @Override
4.     public void setNonNullParameter(PreparedStatement ps, int i, Integer[] parameter, JdbcType jdbcType) throws SQLException {
5.         // 将Java类型转换为数据库类型
6.         Array array = ps.getConnection().createArrayOf("integer", parameter);
7.         ps.setArray(i, array);
8.     }
10.     @Override
11.     public Integer[] getNullableResult(ResultSet rs, String columnName) throws SQLException {
12.         // 从数据库类型转换为Java类型
13.         Array array = rs.getArray(columnName);
14.         return (Integer[]) array.getArray();
15.     }
17.     @Override
18.     public Integer[] getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
19.         Array array = rs.getArray(columnIndex);
20.         return (Integer[]) array.getArray();
21.     }
23.     @Override
24.     public Integer[] getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
25.         Array array = cs.getArray(columnIndex);
26.         return (Integer[]) array.getArray();
27.     }
28. }

复制代码

自定义处理器2：jsonb范例

1. import org.apache.ibatis.type.JdbcType;
2. import org.apache.ibatis.type.TypeHandler;
3. import org.postgresql.util.PGobject;
5. import java.sql.CallableStatement;
6. import java.sql.PreparedStatement;
7. import java.sql.ResultSet;
8. import java.sql.SQLException;
10. public class JsonbTypeHandler implements TypeHandler<String> {
12.     @Override
13.     public void setParameter(PreparedStatement ps, int i, String parameter, JdbcType jdbcType) throws SQLException {
14.         PGobject jsonObject = new PGobject();
15.         jsonObject.setType("jsonb");
16.         jsonObject.setValue(parameter);
17.         ps.setObject(i, jsonObject);
18.     }
20.     @Override
21.     public String getResult(ResultSet rs, String columnName) throws SQLException {
22.         return rs.getString(columnName);
23.     }
25.     @Override
26.     public String getResult(ResultSet rs, int columnIndex) throws SQLException {
27.         return rs.getString(columnIndex);
28.     }
30.     @Override
31.     public String getResult(CallableStatement cs, int columnIndex) throws SQLException {
32.         return cs.getString(columnIndex);
33.     }
34. }

复制代码

自定义处理器3：罗列范例

• 定义罗列类
  假设有一个罗列类 StatusEnum，它有两个值：ACTIVE 和 INACTIVE。
  

1. public enum StatusEnum {
2.     ACTIVE,
3.     INACTIVE;
5.     public static StatusEnum fromValue(String value) {
6.         for (StatusEnum status : values()) {
7.             if (status.name().equalsIgnoreCase(value)) {
8.                 return status;
9.             }
10.         }
11.         throw new IllegalArgumentException("Unknown enum value: " + value);
12.     }
13. }

复制代码

• 创建自定义范例处理器
  创建一个范例处理器来处理 StatusEnum：
  这个范例处理器将数据库中的字符串映射到 StatusEnum 罗列上。它假设数据库中存储的是罗列值的名称（例如，“ACTIVE” 或 “INACTIVE”）。
  

1. import org.apache.ibatis.type.BaseTypeHandler;
2. import org.apache.ibatis.type.JdbcType;
4. import java.sql.CallableStatement;
5. import java.sql.PreparedStatement;
6. import java.sql.ResultSet;
7. import java.sql.SQLException;
9. public class StatusEnumTypeHandler extends BaseTypeHandler<StatusEnum> {
11.     @Override
12.     public void setNonNullParameter(PreparedStatement ps, int i, StatusEnum parameter, JdbcType jdbcType) throws SQLException {
13.         ps.setString(i, parameter.name());
14.     }
16.     @Override
17.     public StatusEnum getNullableResult(ResultSet rs, String columnName) throws SQLException {
18.         String value = rs.getString(columnName);
19.         return value == null ? null : StatusEnum.fromValue(value);
20.     }
22.     @Override
23.     public StatusEnum getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
24.         String value = rs.getString(columnIndex);
25.         return value == null ? null : StatusEnum.fromValue(value);
26.     }
28.     @Override
29.     public StatusEnum getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
30.         String value = cs.getString(columnIndex);
31.         return value == null ? null : StatusEnum.fromValue(value);
32.     }
33. }

复制代码

五 把TypeHandler配置到步伐中有四种方法：

   每一种都测试过，把踩过的坑用黑体标识了。
  

• 在Mapper.xml中声明(应用单个指定字段)
  

1. <resultMap id="BaseResultMap" type="com.xxx.EntiyDto">
2.         <result column="enum1" jdbcType="INTEGER" property="enum1" typeHandler="com.xxx.handler.IntegerArrayTypeHandler"/>
3. </resultMap>

复制代码

• 在springboot的yml配置文件中设置范例处理器所在的包名，不是处理器路径（应用到全局）
  

1. mybatis-plus:
2.   type-handlers-package: com.xxx.handler

复制代码

• 实体类指定范例处理器。必须在实体类上加@TableName(autoResultMap = true),否则不见效
  

1. @Data
2. @Accessors(chain = true)
3. @TableName(autoResultMap = true)
4. public class User {
5.     private Long id;
6.     ...
7.     /**
8.      * 注意！！ 必须开启映射注解
9.      *
10.      * @TableName(autoResultMap = true)
11.      *
12.      * 以下两种类型处理器，二选一 也可以同时存在
13.      *
14.      * 注意！！选择对应的 JSON 处理器也必须存在对应 JSON 解析依赖包
15.      */
16.     @TableField(typeHandler = IntegerArrayTypeHandler.class)
17.     private Integer[] integerArray;
18. }

复制代码

• 在mybatis配置文件中设置
  

1.         <typeHandlers>
2.                 <typeHandler handler="com.xxx.handler.IntegerArrayTypeHandler"/>
3.         </typeHandlers>

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。