Mybatis 源码6 结果集映射流程 ，mybatis插件实现原理和基于mybatis插件实现参数化类型TypeHandler

一丶前情回顾

书接上回，下面是SimpleExecutor执行查询的主要逻辑

+ instantiateStatement 根据数据库连接，创建Statement，具体什么类型的Statement取决于当前用的什么StatementHandler

至此我们以及将参数设置到PreparedStatement 上了
相当于
select * from A where a=#{a}
这里的a=？已经成功设置上去了

二丶执行查询

这些查询交由StatementHandler处理，可以看下图，

还有个RoutingStatementHandler，就是调用被装饰着的query方法

三丶处理结果集

终于到了这篇总结的重点，处理结果集，上面图可以看到Statement执行excute后，使用ResultSetHandler处理结果集，ResultSetHandler只有一个实现类 DefaultResultSetHandler

1.处理多结果集

存储过程存在多结果集的情况，

2.处理结果集

不存在嵌套子查询的时候，使用handleRowValuesForSimpleResultMap

这里出现一个类DefaultResultContext，实现了ResultContext，这是结果上下文，主要的职责是控制处理结果行的停止，配合rowBounds实现内存分页，后面的storeObject就是将一行对应的对象存在list（似乎对map这种出惨有特殊处理，对于嵌套子查询也有特殊处理）

这里的自动映射应该是处理，没有指定resultMap 凭借对象属性和数据库列名进行映射的情况 ，后面applyPropertyMappings 处理指定resultMap 中column和 property的情况

例如我们执行下面test这条sql，

        SELECT *
        FROM user
        WHERE id = #{id}

User对象是

对于id和name 会执行 applyAutomaticMappings赋值对应属性，对于eList执行applyPropertyMappings赋值到属性

四丶mybatis插件实现原理

1.拦截器接口

public interface Interceptor {

    //拦截        Invocation ：当前被拦截对象 参数 和被拦截方法
  Object intercept(Invocation invocation) throws Throwable;

  //动态代理
  default Object plugin(Object target) {
    return Plugin.wrap(target, this);
  }

  default void setProperties(Properties properties) {
    // NOP  可以在这里给拦截器赋值一些属性
  }

}

2.Plugin.wrap(target, this)方法如何实现拦截

• Plugin 实现了InvocationHandler——基于JDK动态代理

• 读取注解信息，获取当前拦截器要拦截什么类的什么方法

注意获取方法的方式

Method method = sig.type().getMethod(sig.method(), sig.args());

getMethod方法是没有办法获取到私有方法的，所有无法拦截一个私有方法

• 获取被代理类的接口

• 动态代理对象生成

3.动态代理生成的对象是怎么被使用的

如上我们知道了mybatis 是怎么支持插件的，根据拦截器上的信息生成动态代理对象，动态代理对象在执行方法的时候会进入拦截器的intercept拦截方法，那么动态代理的生成的对象在哪里被使用到昵

• Configuration 类 也就是mybatis的大管家，在new一些mybatis四大对象的时候会使用到插件

也就是说mybatis 只支持拦截ParmeterHandler,ResultSetHandler,StatementHandler,Excutor这四种对象

在mybatis执行中的使用的四大对象其实是被动态代理后的对象

五丶基于mybatis插件实现参数化类型TypeHandler

1.需求

• 存在实体

@Data
public class User {
    @NotNull(message = "id")
    private Integer id;
    @NotNull(message = "name")
    private String name;

    private List eList;

    @Data
    @AllArgsConstructor
    @NoArgsConstructor
    static class E {
        Integer i;
        String j;
    }
}

* 我希望在插入User对象到数据表的时候，自动到eList转换为Json格式存入 查询User对象返回的时候，自动从Json格式字符串转换为 List<e></e>类型

2.实现TypeHandler

阅读上面结果集处理源码后，我们知道对一个字段的处理最终是使用了对应TypeHanlder进行读取，我们实现一个我们的TypeHandler实现Json的序列化和反序列化

@MappedJdbcTypes(JdbcType.VARCHAR)
@MappedTypes({List.class})
public class ParameterizeType2JsonStrTypeHandler extends BaseTypeHandler {

    //设置参数 也就是eList插入的时候怎么设置到对应的PrepareStatement占位符上
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException {
        //直接序列化成字符串
        ps.setString(i, JSON.toJSONString(parameter));
    }

    @Override
    public Object getNullableResult(ResultSet rs, String columnName) throws SQLException {
        //怎么根据字段名，返回属性值
        //比如我们这里是List,不能简单的序列化成List.class ，因为这样List内部其实存储的是JsonObject对象，没办法泛型擦除是这样的
        //那么怎么拿到参数化类型昵 我们要的是List

    }
}

这里的getNullableResult 入参只有结果集 和列名，这显然不足以让我们拿到 List<确切的类型><!--确切的类型-->

3.”拦截”ResultSetHandler

我们知道处理一行数据到对象映射使用的方法是ResultSetHandler的handleRowValues方法

加上上面实现TypeHandler遇到的问题，看来我们需要实现一个自己的ResultSetHandler，在handleRowValues执行之前把当前处理对象的class 存储到一个位置，然后我们实现的TypeHandler在getNullableResult 执行的时候拿到class 根据字段名词找到对应属性，然后获取到数据的通用类型，然后再进行Json的反序列化

1. 如何存储结果集转换目标对象类型的class 这里我们使用ThreadLocal进行存储ResultMap 可以根据 ResultMap中的方法拿到当前的类型

public class ResultSetProcessTypeMemory {

    private final static ThreadLocal MEMORY = new ThreadLocal<>();

    public static ResultMap get() {
        return MEMORY.get();
    }

    public static void set(ResultMap resultMap) {
        MEMORY.set(resultMap);
    }

    public static void remove() {
        MEMORY.remove();
    }

}

public class MyResultSetHandler extends DefaultResultSetHandler {

    public MyResultSetHandler(Executor executor, MappedStatement mappedStatement, ParameterHandler parameterHandler, ResultHandler resultHandler, BoundSql boundSql, RowBounds rowBounds) {
        super(executor, mappedStatement, parameterHandler, resultHandler, boundSql, rowBounds);
    }

    @Override
    public void handleRowValues(ResultSetWrapper rsw, ResultMap resultMap, ResultHandler resultHandler, RowBounds rowBounds, ResultMapping parentMapping) throws SQLException {
        //放入ThreadLocal
        ResultSetProcessTypeMemory.set(resultMap);
        super.handleRowValues(rsw, resultMap, resultHandler, rowBounds, parentMapping);
        //移除
        ResultSetProcessTypeMemory.remove();
    }
}

private static MyResultSetHandler create(RoutingStatementHandler routingStatementHandler) {
    BaseStatementHandler delegate = (BaseStatementHandler) ReflectUtil.getFieldValue(routingStatementHandler, "delegate");
    return create(delegate);
}

private static MyResultSetHandler create(BaseStatementHandler base) {
    Executor executor = (Executor) ReflectUtil.getFieldValue(base, "executor");
    MappedStatement mappedStatement = (MappedStatement) ReflectUtil.getFieldValue(base, "mappedStatement");
    ParameterHandler parameterHandler = (ParameterHandler) ReflectUtil.getFieldValue(base, "parameterHandler");
    ResultHandler resultHandler = (ResultHandler) ReflectUtil.getFieldValue(base, "resultHandler");
    BoundSql boundSql = (BoundSql) ReflectUtil.getFieldValue(base, "boundSql");
    RowBounds rowBounds = (RowBounds) ReflectUtil.getFieldValue(base, "rowBounds");
    return new MyResultSetHandler(executor, mappedStatement, parameterHandler, resultHandler, boundSql, rowBounds);
}

public static MyResultSetHandler create(StatementHandler statementHandler) {
    if (statementHandler instanceof BaseStatementHandler){
        return create((BaseStatementHandler)statementHandler);
    }
    if (statementHandler instanceof RoutingStatementHandler){
        return create((RoutingStatementHandler)statementHandler);
    }
    throw new UnsupportedOperationException();
}

4.拦截StatementHandler

上面的一通操作，我们搞出来一个自己的ResultSetHandler 但是怎么让mybatis 用我们的ResultSetHandler昵，我们可以拦截StatementHandler的query方法

@Intercepts({
        @Signature(
                type = StatementHandler.class,
                method = "query",
                args = {Statement.class, ResultHandler.class}
        )
})
public class ListVarcharTypeHandlerInterceptor implements Interceptor {

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        PreparedStatement ps = (PreparedStatement) args[0];
        ps.execute();
        StatementHandler target = (StatementHandler) invocation.getTarget();
        //最后一步用我们自己的ResultSetHandler 处理
        return MyResultSetHandler.create(target).handleResultSets(ps);
    }

}

5.注入拦截器

/**
 * mybatis配置
 */
@Configuration
public class MybatisConfiguration {

    /**
     * 注册拦截器
     */
    @Bean
    public ListVarcharTypeHandlerInterceptor mybatisInterceptor() {
        return new ListVarcharTypeHandlerInterceptor();
    }

}

6.完善TypeHandler

实现完第五步，我们可以从ThreadLocal中拿到ResultMap，接下来就是如何利用Json反序列化到属性上了

@MappedJdbcTypes(JdbcType.VARCHAR)
@MappedTypes({List.class})
public class ParameterizeType2JsonStrTypeHandler extends BaseTypeHandler {

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Object parameter, JdbcType jdbcType) throws SQLException {
        ps.setString(i, JSON.toJSONString(parameter));
    }

    @Override
    public Object getNullableResult(ResultSet rs, String columnName) throws SQLException {
        //从TreadLocal拿
        ResultMap resultMap = ResultSetProcessTypeMemory.get();
        //拿到封装目标类型class
        Class type = resultMap.getType();
        //根据字段名词 拿到field 这里可以优化下，忽略大小写吧
        Field field = ReflectUtil.getField(type, columnName);
        //拿到通用类型 可以拿到List
        Type genericType = field.getGenericType();
        //从resutSet拿到字符串内容，并且反序列化回去
        return JSON.parseObject(rs.getString(columnName), genericType);
    }
}

7.简单的一个测试

    //指定用我们的参数化类型TypeHandler

    SELECT *
    FROM user
    WHERE id = #{id}

最后成功反序列化且类型没有错误

Original: https://www.cnblogs.com/cuzzz/p/16609763.html
Author: Cuzzz
Title: Mybatis 源码6 结果集映射流程 ，mybatis插件实现原理和基于mybatis插件实现参数化类型TypeHandler