mybatis通过ResultSets返回自定义结果集

说明：

提示：这篇文章仅是记录自己成长路上的愚笨，不喜勿喷，同时也希望各路大神提出优秀的观点。谢谢!!

---

问题需求：

项目中想通过mybatis查询后直接返回DataFrame数据结构的数据，但是mybatis的ResultType又不支持，所以“曲线救国”通过ResultSets来返回需要的结果

---

代码：

ResultSets结果集拦截器

package com.application.interceptor.mybatis;

import joinery.DataFrame;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.ibatis.executor.resultset.DefaultResultSetHandler;
import org.apache.ibatis.executor.resultset.ResultSetHandler;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.*;
import org.springframework.util.ReflectionUtils;

import java.lang.reflect.Field;
import java.sql.Statement;
import java.util.Arrays;
import java.util.Properties;

/**
 * mybatisResultSet结果集拦截器
 *
 * @author wo
 * @createDate 2021/5/24 16:00
 */
@Intercepts({@Signature(type = ResultSetHandler.class, method = "handleResultSets", args = {Statement.class})})
public class ResultHandlerInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        ResultSetHandler resultSetHandler = (ResultSetHandler) invocation.getTarget();
        Field mappedStatementField = ReflectionUtils.findField(DefaultResultSetHandler.class, "mappedStatement");
        mappedStatementField.setAccessible(true);
        MappedStatement mappedStatement = (MappedStatement) ReflectionUtils.getField(mappedStatementField, resultSetHandler);
        SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
        if (sqlCommandType == SqlCommandType.SELECT) {
            String[] resultSets = mappedStatement.getResultSets();
            if (ArrayUtils.isNotEmpty(resultSets) && ArrayUtils.contains(resultSets, "dataFrame")) {
                Statement statement = (Statement) invocation.getArgs()[0];
                // 这里需要注意：一定要注意使用返回list，因为mybatis所有的查询最终都是使用selectList方法。
                // 这里即使返回的是list，但是在mapper文件里可以直接使用DataFrame接收参数（也就是说selectOne方法会获取list的第一个值来返回）
                return Arrays.asList(DataFrame.readSql(statement.getResultSet()));
            }
        }
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

    @Override
    public void setProperties(Properties properties) {

    }
}

mybatis的Xml配置，拦截器的注册

<plugins>
        <plugin interceptor="com.application.interceptor.mybatis.ResultHandlerInterceptor"></plugin>
</plugins>

至此已经配置完成了! 接下来就可以使用了

mybatis的Xml配置

<select id="exportData" resultSets="dataFrame">
        select * from device
</select>

mybatis的mapper配置

DataFrame<Object> exportData();

---

结尾：

说明：关于mybatis的ResultSets具体使用说明请参考mybatis官网 或者自行度娘。
有关mybatis的查询结果集处理请参考mybatis的源码或者自行度娘。
谢谢!!!