mybatis ResultSet结果集自定义

/**
 * mybatisResultSet结果集拦截器
 */
@Component
@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);
        //获取节点属性的集合
        List<ResultMap> resultMaps = mappedStatement.getResultMaps();
        int resultMapCount = resultMaps.size();
        //获取当前resutType的类型
        Class<?> resultType = resultMaps.get(0).getType();
        if (resultMapCount > 0 && resultType.getName().equals("grp.pt3.model.ElementValue3")){
//            Statement statement = (Statement) invocation.getArgs()[0];
//            //获得结果集
//            ResultSet resultSet = statement.getResultSet();
            Object proceed = invocation.proceed();
            Object handleResultSet = handleResultSet(proceed);
            return handleResultSet;
        }
        return invocation.proceed();
    }

    private Object handleResultSet(Object resultSet){
        if (resultSet instanceof ArrayList) {
            ArrayList resultList = (ArrayList) resultSet;
            for (int i = 0; i < resultList.size(); i++) {
                if (resultList.get(i) instanceof Map) {
                    Map resultMap = (Map) resultList.get(i);
                    if(resultMap.get("id") != null ){
                        resultMap.put("ele_id",resultMap.get("id"));
                    }else if(resultMap.get("ele_id") != null){
                        resultMap.put("id",resultMap.get("ele_id"));
                    }
                    if(resultMap.get("code") != null ){
                        resultMap.put("ele_code",resultMap.get("code"));
                    }else if(resultMap.get("ele_code") != null){
                        resultMap.put("code",resultMap.get("ele_code"));
                    }
                    if(resultMap.get("name") != null ){
                        resultMap.put("ele_name",resultMap.get("name"));
                    }else if(resultMap.get("ele_name") != null){
                        resultMap.put("name",resultMap.get("ele_name"));
                    }
                }
            }
            return resultList;
        }
        return resultSet;
    }

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

    @Override
    public void setProperties(Properties properties) {

    }
}

mybatis-ResultHandler解析:
自定义ResultHandler类处理结果集,适用范围小。
https://blog.csdn.net/qq_28695733/article/details/107212027

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值