Mybatis 自定义ResultHandler(结果处理器)正确使用姿势

本文介绍了如何在Mybatis中使用自定义ResultHandler将查询结果的Map转换为Record类型,以简化批量处理过程,避免冗余代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一、使用场景

对Mybatis查询的结果集进行统一批量处理,如进行类型转换、数据变更等等。

二、前提条件

  1. 必须是查询语句。
  2. mapper方法的返回值必须是void,这也就是不能直接从这个方法获取到查询结果集,需要在ResultHandler里面自己存放查询结果。(这个很重要,否则ResultHandler里面的逻辑不会执行,目前网上大部分资料都没有提到这一点,巨坑)
  3. mapper方法必须传入自定义的结果处理器。

三、源码对照

  1. 方法执行入口,org.apache.ibatis.binding.MapperMethod#execute:
    org.apache.ibatis.binding.MapperMethod#execute
  2. 根据获取方法参数中ResultHandler.class类型的变量作为结果处理器:
    在这里插入图片描述
  3. org.apache.ibatis.binding.MapperMethod#executeWithResultHandler,可以看到executeWithResultHandler方法无返回值。
    在这里插入图片描述

四、使用示例

目标

将原查询结果的Map转为Record,Record可以理解为将Map进行包装了一次的实体类,代码结构如下:

public class Record implements Serializable {
	
	private static final long serialVersionUID = 905784513600884082L;
	
	private Map<String, Object> columns;	

	void setColumnsMap(Map<String, Object> columns) {
		this.columns = columns;
	}
	......
}

这种如果直接使用mybatis进行构造返回结果是比较麻烦,特别是使用的地方比较多,会有很多冗余代码。

实现方式

  1. 自定义ResultHandler代码
public class RecordResultHandler implements ResultHandler<Record> {

    private final List<Record> list = new ArrayList<>();

    @Override
    public void handleResult(ResultContext context) {
        Record record = new Record();
        record.setColumns((Map<String, Object>) context.getResultObject());
        list.add(record);
    }
    
    public List<Record> getResults(){
        return list;
    }
}
  1. xml代码
    <select id="test" resultType="map">
        select * from user_test
    </select>
  1. Mapper层代码
	//必须返回void且传入ResultHandler
	void test(ResultHandler recordResultHandler);
  1. 调用层代码
	RecordResultHandler recordResultHandler = new RecordResultHandler();
	//integDataMapper正常注入即可
    integDataMapper.test(recordResultHandler);
    List<Record> results = recordResultHandler.getResults();
  1. 执行结果
    可以看到已经返回了Record类型的数据
MyBatis中,你可以使用自定义的类型处理器(TypeHandler)来处理自定义的Map类型。类型处理器用于在Java对象和数据库字段之间进行转换。 以下是一种实现自定义类型处理器处理自定义的Map的示例: 1. 定义一个自定义的Map类型,例如`CustomMap`: ```java public class CustomMap extends HashMap<String, Object> { // 添加自定义的方法或属性 } ``` 2. 实现一个自定义的类型处理器,继承自`org.apache.ibatis.type.BaseTypeHandler`类,并实现对应的方法。 ```java import org.apache.ibatis.type.BaseTypeHandler; import org.apache.ibatis.type.JdbcType; import java.sql.CallableStatement; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; public class CustomMapTypeHandler extends BaseTypeHandler<CustomMap> { @Override public void setNonNullParameter(PreparedStatement ps, int i, CustomMap parameter, JdbcType jdbcType) throws SQLException { // 将CustomMap转换为需要的数据类型,并设置到PreparedStatement中 // ps.setXXX(i, convertedValue); } @Override public CustomMap getNullableResult(ResultSet rs, String columnName) throws SQLException { // 从ResultSet中获取指定列名的值,并将其转换为CustomMap类型 // Object columnValue = rs.getXXX(columnName); // CustomMap map = convertToCustomMap(columnValue); // return map; return null; } @Override public CustomMap getNullableResult(ResultSet rs, int columnIndex) throws SQLException { // 与上面类似,只是根据列索引获取值 return null; } @Override public CustomMap getNullableResult(CallableStatement cs, int columnIndex) throws SQLException { // 与上面类似,只是在CallableStatement中获取值 return null; } } ``` 在上述示例中,我们继承了`BaseTypeHandler`类,并重写了父类的方法,在这些方法中进行了自定义类型的转换逻辑。 3. 在MyBatis的配置文件中,注册自定义的类型处理器。 ```xml <typeHandlers> <typeHandler handler="com.example.CustomMapTypeHandler"/> </typeHandlers> ``` 通过以上步骤,你就可以使用自定义的Map类型,并通过自定义的类型处理器来处理该类型的转换逻辑。在数据库操作时,MyBatis会自动调用类型处理器来进行转换。你可以根据实际需求,在类型处理器中编写相应的转换逻辑,将自定义的Map类型与数据库字段进行转换。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值