mybatis 源码分析（六）StatementHandler 主体结构分析

  • 2019 年 10 月 3 日
  • 笔记

分析到这里的时候，mybatis 初始化、接口、事务、缓存等主要功能都已经讲完了，现在就还剩下 StatementHandler 这个真正干活的家伙没有分析了；所以接下来的博客内容主要和数据库的关系比较密切，而 StatementHandler 的主要流程也基本是和 JDBC 的流程是一一对应的；

一、StatementHandler 执行流程

在 mybatis 系列文章的第一篇，我放了一张 mybatis 整体的执行流程图：

从上面的图中也能比较清楚的看到 StatementHandler 的职责：获取 Statement -> 设置参数 -> 查询数据库 -> 将查询结果映射为 JavaBean，从这里也能看到是和我们使用原生 JDBC 的流程是一样的；而整个过程 StatementHandler 又将其拆分成了部分：

  • KeyGenerator：主键设置
  • ParameterHandler：参数设置
  • ResultSetHandler：结果集设置

这里我们首先介绍 StatementHandler 的类结构：

  • RoutingStatementHandler：路由处理器，这个相当于一个静态代理，根据 MappedStatement.statementType 创建对应的对处理器；
  • SimpleStatementHandler：不需要预编译的简单处理器；
  • PreparedStatementHandler：预编译的 SQL 处理器；
  • CallableStatementHandler：主要用于存储过程的调度；

这里的 SimpleStatementHandler、PreparedStatementHandler、CallableStatementHandler 同时也和 JDBC 的三个 Statement 一一对应；首先我们还是先看一下接口方法：

public interface StatementHandler {    Statement prepare(Connection connection) throws SQLException; // 获取 Statement    void parameterize(Statement statement) throws SQLException;   // 参数化    void batch(Statement statement) throws SQLException;          // 批处理    int update(Statement statement) throws SQLException;    <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException;    BoundSql getBoundSql();                 // 获取绑定sql    ParameterHandler getParameterHandler(); // 得到参数处理器  }

其中公共的方法都封装到了 BaseStatementHandler 中（这里使用的是模版模式）；

// 获取 Statement  @Override  public Statement prepare(Connection connection) throws SQLException {    ErrorContext.instance().sql(boundSql.getSql());    Statement statement = null;    try {      statement = instantiateStatement(connection); // 实例化 Statement      setStatementTimeout(statement);               // 设置超时      setFetchSize(statement);                      // 设置读取条数      return statement;    } catch (SQLException e) {      closeStatement(statement);      throw e;    } catch (Exception e) {      closeStatement(statement);      throw new ExecutorException("Error preparing statement.  Cause: " + e, e);    }  }    // 由子类提供不同的实例化  protected abstract Statement instantiateStatement(Connection connection) throws SQLException;

二、StatementHandler 子类

1. RoutingStatementHandler

静态代理模式，根据 MappedStatement.statementType 创建对应的对处理，默认是 PREPARED，可以使用 XML 配置或者注解的方式指定；

public StatementHandler newStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameterObject, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {    // 创建路由选择语句处理器    StatementHandler statementHandler = new RoutingStatementHandler(executor, mappedStatement, parameterObject, rowBounds, resultHandler, boundSql);    // 插件拦截    statementHandler = (StatementHandler) interceptorChain.pluginAll(statementHandler);    return statementHandler;  }
public RoutingStatementHandler(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {    switch (ms.getStatementType()) {  // 根据语句类型，委派到不同的语句处理器(STATEMENT|PREPARED|CALLABLE)      case STATEMENT:        delegate = new SimpleStatementHandler(executor, ms, parameter, rowBounds, resultHandler, boundSql);        break;      case PREPARED:        delegate = new PreparedStatementHandler(executor, ms, parameter, rowBounds, resultHandler, boundSql);        break;      case CALLABLE:        delegate = new CallableStatementHandler(executor, ms, parameter, rowBounds, resultHandler, boundSql);        break;      default:        throw new ExecutorException("Unknown statement type: " + ms.getStatementType());    }  }

2. SimpleStatementHandler

其功能和 JDBC.Statement 对应，从初始化方法也可以看到使用的是简单 Statement；

@Override  protected Statement instantiateStatement(Connection connection) throws SQLException {    // 调用 Connection.createStatement    if (mappedStatement.getResultSetType() != null) {      return connection.createStatement(mappedStatement.getResultSetType().getValue(), ResultSet.CONCUR_READ_ONLY);    } else {      return connection.createStatement();    }  }
@Override  public int update(Statement statement) throws SQLException {    String sql = boundSql.getSql();    Object parameterObject = boundSql.getParameterObject();    KeyGenerator keyGenerator = mappedStatement.getKeyGenerator();    int rows;    if (keyGenerator instanceof Jdbc3KeyGenerator) {      statement.execute(sql, Statement.RETURN_GENERATED_KEYS);      rows = statement.getUpdateCount();      keyGenerator.processAfter(executor, mappedStatement, statement, parameterObject);    } else if (keyGenerator instanceof SelectKeyGenerator) {      statement.execute(sql);      rows = statement.getUpdateCount();      keyGenerator.processAfter(executor, mappedStatement, statement, parameterObject);    } else {      //如果没有keyGenerator,直接调用Statement.execute和Statement.getUpdateCount      statement.execute(sql);      rows = statement.getUpdateCount();    }    return rows;  }

这里在更新的时候需要区分 KeyGenerator，因为使用的是简单 Statement，所以需要在查询的时候指定返回主键 statement.execute(sql, Statement.RETURN_GENERATED_KEYS);；

3. PreparedStatementHandler

其功能和 PreparedStatement 对应，这里初始化的时候可以看到比 SimpleStatementHandler 要复杂一些，因为 PreparedStatement 更新返回主键有三个方法，详细分析后面会单独放一篇详细讲解；

@Override  protected Statement instantiateStatement(Connection connection) throws SQLException {    //调用Connection.prepareStatement    String sql = boundSql.getSql();    if (mappedStatement.getKeyGenerator() instanceof Jdbc3KeyGenerator) {      String[] keyColumnNames = mappedStatement.getKeyColumns();      if (keyColumnNames == null) {        return connection.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);      } else {        return connection.prepareStatement(sql, keyColumnNames);      }    } else if (mappedStatement.getResultSetType() != null) {      return connection.prepareStatement(sql, mappedStatement.getResultSetType().getValue(), ResultSet.CONCUR_READ_ONLY);    } else {      return connection.prepareStatement(sql);    }  }

还有一点不同的就是 PreparedStatement 需要参数化设置，就是设置预编译 SQL 对应占位符的参数；

// DefaultParameterHandler  @Override  public void setParameters(PreparedStatement ps) throws SQLException {    ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());    List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();    if (parameterMappings != null) {      // 循环设参数      for (int i = 0; i < parameterMappings.size(); i++) {        ParameterMapping parameterMapping = parameterMappings.get(i);        if (parameterMapping.getMode() != ParameterMode.OUT) {          // 如果不是OUT，才设进去          Object value;          String propertyName = parameterMapping.getProperty();          if (boundSql.hasAdditionalParameter(propertyName)) {            // 若有额外的参数, 设为额外的参数            value = boundSql.getAdditionalParameter(propertyName);          } else if (parameterObject == null) {            // 若参数为null，直接设null            value = null;          } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {            // 若参数有相应的TypeHandler，直接设object            value = parameterObject;          } else {            // 除此以外，MetaObject.getValue反射取得值设进去            MetaObject metaObject = configuration.newMetaObject(parameterObject);            value = metaObject.getValue(propertyName);          }          TypeHandler typeHandler = parameterMapping.getTypeHandler();          JdbcType jdbcType = parameterMapping.getJdbcType();          if (value == null && jdbcType == null) {            // 不同类型的set方法不同，所以委派给子类的setParameter方法            jdbcType = configuration.getJdbcTypeForNull();          }          typeHandler.setParameter(ps, i + 1, value, jdbcType);        }      }    }  }

另外还有 CallableStatementHandler ，代码也很简单，这里就不详细分析了；

VirMach 便宜 VPS

QNews

QNews