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
    评论
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、付费专栏及课程。

余额充值