Sharding-JDBC 源码之结果集归并

最新推荐文章于 2024-08-15 15:28:28 发布

朝花拾夕

最新推荐文章于 2024-08-15 15:28:28 发布

阅读量2k

点赞数 4

分类专栏： Sharding-JDBC 文章标签：数据库 sharding-jdbc java sql

本文链接：https://blog.youkuaiyun.com/smile_from_2015/article/details/112101676

版权

Sharding-JDBC 系列

将从各个数据节点获取的多数据结果集，组合成为一个结果集并正确的返回至请求客户端，称为结果归并。由于从数据库中返回的结果集是逐条返回的，并不需要将所有的数据一次性加载至内存中，因此，在进行结果归并时，沿用数据库返回结果集的方式进行归并，能够极大减少内存的消耗，是归并方式的优先选择。

归并类型

流式归并

指每一次从结果集中获取到的数据，都能够通过逐条获取的方式返回正确的单条数据，它与数据库原生的返回结果集的方式最为契合。遍历、排序以及流式分组都属于流式归并的一种。

内存归并

需要将结果集的所有数据都遍历并存储在内存中，再通过统一的分组、排序以及聚合等计算之后，再将其封装成为逐条访问的数据结果集返回。

具体归并类型请参考官网：内核剖析-归并引擎

源码分析

将执行完成的 PreparedStatement 对象交给 resultSetHandler 处理

org.apache.ibatis.executor.statement.PreparedStatementHandler#query

    public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
   
    PreparedStatement ps = (PreparedStatement) statement;
    // 执行完成
    ps.execute();
    // 处理结果集
    return resultSetHandler.handleResultSets(ps);
  }

执行完成后，将 PreparedStatement 对象 ps 传给结果处理器 resultSetHandler 的 handleResultSets 方法；

执行 DefaultResultSetHandler 中的 handleResultSets 方法

    public List<Object> handleResultSets(Statement stmt) throws SQLException {
   
    ErrorContext.instance().activity("handling results").object(mappedStatement.getId());

    final List<Object> multipleResults = new ArrayList<>();

    int resultSetCount = 0;
    // 1. 获取第一个结果集
    ResultSetWrapper rsw = getFirstResultSet(stmt);

    // ...... 暂时省略
  }
  
  private ResultSetWrapper getFirstResultSet(Statement stmt) throws SQLException {
   
    //  2. stmt 是 ShardingPreparedStatement 类型，所以执行该类中的方法
    ResultSet rs = stmt.getResultSet();
    while (rs == null) {
   
      // move forward to get the first resultset in case the driver
      // doesn't return the resultset as the first result (HSQLDB 2.1)
      if (stmt.getMoreResults()) {
   
        rs = stmt.getResultSet();
      } else {
   
        if (stmt.getUpdateCount() == -1) {
   
          // no more results. Must be no resultset
          break;
        }
      }
    }
    return rs != null ? new ResultSetWrapper(rs, configuration) : null;
  }

1 .获取第一个结果集，这里的 stmt 对象是 ShardingPreparedStatement 类型；
2. getFirstResultSet 方法中，stmt 调用 getResultSet 方法获取结果集。

执行 ShardingPreparedStatement 中的 getResultSet 方法

    public ResultSet getResultSet() throws SQLException {
   
        if (null != currentResultSet) {
   
            return currentResultSet;
        }
        
        // 1. 如果只查询一张库表，则不用进行结果归并
        if (1 == routedStatements.size() && routeResult.getSqlStatement() instanceof DQLStatement) {
   
            currentResultSet = routedStatements.iterator().next().getResultSet();
            return currentResultSet;
        }
        List<ResultSet> resultSets = new ArrayList<>(routedStatements.size()