StatementHandler 接口是 MyBatis 的核心接口之一,它完成了 MyBatis 中最核心的工作,也是 Executor 接口实现的基础。

StatementHandler 接口中的功能很多,例如创建 Statement 对象,为 SQL 语句绑定实参,执行 select、insert、update、delete 等多种类型的 SQL 语句,批量执行 SQL 语句,将结果集映射成结果对象。

  1. public interface StatementHandler {
  2. // 从连接中获取一个Statement
  3. Statement prepare(Connection connection, Integer transactionTimeout)
  4. throws SQLException;
  5. // 绑定statement执行时所需的实参
  6. void parameterize(Statement statement)
  7. throws SQLException;
  8. // 批量执行SQL语句
  9. void batch(Statement statement)
  10. throws SQLException;
  11. // 执行update/insert/delete语句
  12. int update(Statement statement)
  13. throws SQLException;
  14. // 执行select语句
  15. <E> List<E> query(Statement statement, ResultHandler resultHandler)
  16. throws SQLException;
  17. <E> Cursor<E> queryCursor(Statement statement)
  18. throws SQLException;
  19. BoundSql getBoundSql();
  20. // 获取参数处理器
  21. ParameterHandler getParameterHandler();
  22. }

RoutingStatementHandler

RoutingStatementHandler 使用了策略模式,RoutingStatementHandler 是策略类,而 SimpleStatementHandler、PreparedStatementHandler、CallableStatementHandler 则是实现了具体算法的实现类,RoutingStatementHandler 对象会根据 MappedStatement 对象的 StatementType 属性值选择使用相应的策略去执行。

  1. public class RoutingStatementHandler implements StatementHandler {
  2. // 持有的真正实现StatementHandler接口功能的对象
  3. private final StatementHandler delegate;
  4. public RoutingStatementHandler(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
  5. // RoutingStatementHandler的作用就是根据ms的配置,生成一个相对应的StatementHandler对象
  6. // 并设置到持有的delegate属性中,本对象的所有方法都是通过调用delegate的相应方法实现的
  7. switch (ms.getStatementType()) {
  8. case STATEMENT:
  9. delegate = new SimpleStatementHandler(executor, ms, parameter, rowBounds, resultHandler, boundSql);
  10. break;
  11. case PREPARED:
  12. delegate = new PreparedStatementHandler(executor, ms, parameter, rowBounds, resultHandler, boundSql);
  13. break;
  14. case CALLABLE:
  15. delegate = new CallableStatementHandler(executor, ms, parameter, rowBounds, resultHandler, boundSql);
  16. break;
  17. default:
  18. throw new ExecutorException("Unknown statement type: " + ms.getStatementType());
  19. }
  20. }
  21. @Override
  22. public Statement prepare(Connection connection, Integer transactionTimeout) throws SQLException {
  23. return delegate.prepare(connection, transactionTimeout);
  24. }
  25. @Override
  26. public void parameterize(Statement statement) throws SQLException {
  27. delegate.parameterize(statement);
  28. }
  29. @Override
  30. public void batch(Statement statement) throws SQLException {
  31. delegate.batch(statement);
  32. }
  33. @Override
  34. public int update(Statement statement) throws SQLException {
  35. return delegate.update(statement);
  36. }
  37. @Override
  38. public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
  39. return delegate.query(statement, resultHandler);
  40. }
  41. @Override
  42. public <E> Cursor<E> queryCursor(Statement statement) throws SQLException {
  43. return delegate.queryCursor(statement);
  44. }
  45. @Override
  46. public BoundSql getBoundSql() {
  47. return delegate.getBoundSql();
  48. }
  49. @Override
  50. public ParameterHandler getParameterHandler() {
  51. return delegate.getParameterHandler();
  52. }
  53. }

BaseStatementHandler

看它以 Base 开头,就可以猜到 它是一个实现了 StatementHandler 接口的抽象类,这个类只提供了一些参数绑定相关的方法,并没有实现操作数据库的方法。

  1. public abstract class BaseStatementHandler implements StatementHandler {
  2. // 持有的这些属性都是通过构造方法完成初始化的,typeHandlerRegistry、
  3. // objectFactory、parameterHandler等则是通过configuration属性获得的
  4. protected final Configuration configuration;
  5. protected final ObjectFactory objectFactory;
  6. protected final TypeHandlerRegistry typeHandlerRegistry;
  7. protected final ResultSetHandler resultSetHandler;
  8. // parameterHandler的功能主要是为SQL语句绑定实参,也就是使用传入的实参
  9. // 替换SQL语句中的占位符"?"
  10. protected final ParameterHandler parameterHandler;
  11. // 用来执行SQL语句的执行器
  12. protected final Executor executor;
  13. protected final MappedStatement mappedStatement;
  14. // 记录了用户设置的offset和limit,用于在结果集中定位
  15. // 映射的起始位置和结束位置
  16. protected final RowBounds rowBounds;
  17. protected BoundSql boundSql;
  18. // BaseStatementHandler的构造方法主要用于属性的初始化
  19. protected BaseStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameterObject, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
  20. this.configuration = mappedStatement.getConfiguration();
  21. this.executor = executor;
  22. this.mappedStatement = mappedStatement;
  23. this.rowBounds = rowBounds;
  24. this.typeHandlerRegistry = configuration.getTypeHandlerRegistry();
  25. this.objectFactory = configuration.getObjectFactory();
  26. if (boundSql == null) { // issue #435, get the key before calculating the statement
  27. // 其中调用了KeyGenerator的processBefore()方法
  28. // 用于初始化SQL语句的主键
  29. generateKeys(parameterObject);
  30. boundSql = mappedStatement.getBoundSql(parameterObject);
  31. }
  32. this.boundSql = boundSql;
  33. this.parameterHandler = configuration.newParameterHandler(mappedStatement, parameterObject, boundSql);
  34. this.resultSetHandler = configuration.newResultSetHandler(executor, mappedStatement, rowBounds, parameterHandler, resultHandler, boundSql);
  35. }
  36. protected void generateKeys(Object parameter) {
  37. KeyGenerator keyGenerator = mappedStatement.getKeyGenerator();
  38. ErrorContext.instance().store();
  39. keyGenerator.processBefore(executor, mappedStatement, null, parameter);
  40. ErrorContext.instance().recall();
  41. }
  42. @Override
  43. public BoundSql getBoundSql() {
  44. return boundSql;
  45. }
  46. @Override
  47. public ParameterHandler getParameterHandler() {
  48. return parameterHandler;
  49. }
  50. @Override
  51. public Statement prepare(Connection connection, Integer transactionTimeout) throws SQLException {
  52. ErrorContext.instance().sql(boundSql.getSql());
  53. Statement statement = null;
  54. try {
  55. // 这是一个抽象方法,用于初始化java.sql.Statement对象
  56. statement = instantiateStatement(connection);
  57. // 为Statement对象设置超时时间及fetchSize
  58. setStatementTimeout(statement, transactionTimeout);
  59. setFetchSize(statement);
  60. return statement;
  61. } catch (SQLException e) {
  62. closeStatement(statement);
  63. throw e;
  64. } catch (Exception e) {
  65. closeStatement(statement);
  66. throw new ExecutorException("Error preparing statement. Cause: " + e, e);
  67. }
  68. }
  69. protected abstract Statement instantiateStatement(Connection connection) throws SQLException;
  70. protected void setStatementTimeout(Statement stmt, Integer transactionTimeout) throws SQLException {
  71. Integer queryTimeout = null;
  72. if (mappedStatement.getTimeout() != null) {
  73. queryTimeout = mappedStatement.getTimeout();
  74. } else if (configuration.getDefaultStatementTimeout() != null) {
  75. queryTimeout = configuration.getDefaultStatementTimeout();
  76. }
  77. if (queryTimeout != null) {
  78. stmt.setQueryTimeout(queryTimeout);
  79. }
  80. StatementUtil.applyTransactionTimeout(stmt, queryTimeout, transactionTimeout);
  81. }
  82. protected void setFetchSize(Statement stmt) throws SQLException {
  83. Integer fetchSize = mappedStatement.getFetchSize();
  84. if (fetchSize != null) {
  85. stmt.setFetchSize(fetchSize);
  86. return;
  87. }
  88. Integer defaultFetchSize = configuration.getDefaultFetchSize();
  89. if (defaultFetchSize != null) {
  90. stmt.setFetchSize(defaultFetchSize);
  91. }
  92. }
  93. protected void closeStatement(Statement statement) {
  94. try {
  95. if (statement != null) {
  96. statement.close();
  97. }
  98. } catch (SQLException e) {
  99. //ignore
  100. }
  101. }
  102. }

BaseStatementHandler 主要实现了 StatementHandler 接口中的 prepare()方法,BaseStatementHandler 依赖两个重要的组件,ParameterHandler 和 ResultSetHandler。

ParameterHandler 系列组件

我们要执行的 SQL 语句中可能包含占位符”?”,而每个”?”都对应了 BoundSql 中 parameterMappings 集合中的一个元素,在该 ParameterMapping 对象中记录了对应的参数名称以及该参数的相关属性。ParameterHandler 接口定义了一个非常重要的方法 setParameters(),该方法主要负责调用 PreparedStatement 的 set*()系列方法,为 SQL 语句绑定实参。MyBatis 只为 ParameterHandler 接口提供了唯一一个实现类 DefaultParameterHandler。

  1. public interface ParameterHandler {
  2. // 获取用户传入的实参对象
  3. Object getParameterObject();
  4. // 本方法主要负责调用PreparedStatement.set*()方法,为SQL语句绑定实参。
  5. void setParameters(PreparedStatement ps)
  6. throws SQLException;
  7. }
  8. public class DefaultParameterHandler implements ParameterHandler {
  9. // 管理mybatis中所有的TypeHandler对象
  10. private final TypeHandlerRegistry typeHandlerRegistry;
  11. // 其中记录了SQL节点相应的配置信息
  12. private final MappedStatement mappedStatement;
  13. // 用户传入的实参对象
  14. private final Object parameterObject;
  15. // 其中记录了要执行的SQL语句,及参数信息
  16. private final BoundSql boundSql;
  17. private final Configuration configuration;
  18. // 构造方法主要为持有的属性 进行初始化
  19. public DefaultParameterHandler(MappedStatement mappedStatement, Object parameterObject, BoundSql boundSql) {
  20. this.mappedStatement = mappedStatement;
  21. this.configuration = mappedStatement.getConfiguration();
  22. this.typeHandlerRegistry = mappedStatement.getConfiguration().getTypeHandlerRegistry();
  23. this.parameterObject = parameterObject;
  24. this.boundSql = boundSql;
  25. }
  26. @Override
  27. public Object getParameterObject() {
  28. return parameterObject;
  29. }
  30. // 为PreparedStatement对象要执行的SQL语句中的占位符 设置对应的参数值
  31. @Override
  32. public void setParameters(PreparedStatement ps) {
  33. ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());
  34. // 获取参数列表
  35. List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
  36. if (parameterMappings != null) {
  37. for (int i = 0; i < parameterMappings.size(); i++) {
  38. ParameterMapping parameterMapping = parameterMappings.get(i);
  39. // 过滤掉存储过程中的输出参数
  40. if (parameterMapping.getMode() != ParameterMode.OUT) {
  41. // 记录绑定的实参
  42. Object value;
  43. // 获取参数对应的属性名
  44. String propertyName = parameterMapping.getProperty();
  45. // 根据属性名 获取 实参值
  46. if (boundSql.hasAdditionalParameter(propertyName)) { // issue #448 ask first for additional params
  47. value = boundSql.getAdditionalParameter(propertyName);
  48. // 整个实参为空
  49. } else if (parameterObject == null) {
  50. value = null;
  51. // 如果实参可以直接通过TypeHandler转换成JdbcType
  52. } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
  53. value = parameterObject;
  54. } else {
  55. // 获取对象中相应的属性值 或查找Map对象中的值
  56. MetaObject metaObject = configuration.newMetaObject(parameterObject);
  57. value = metaObject.getValue(propertyName);
  58. }
  59. // 获取当前parameterMapping中的TypeHandler对象 及JdbcType对象
  60. TypeHandler typeHandler = parameterMapping.getTypeHandler();
  61. JdbcType jdbcType = parameterMapping.getJdbcType();
  62. if (value == null && jdbcType == null) {
  63. jdbcType = configuration.getJdbcTypeForNull();
  64. }
  65. try {
  66. // TypeHandler的setParameter()方法会调用PreparedStatement对象的
  67. // set*()系列方法,为SQL语句绑定相应的实参
  68. typeHandler.setParameter(ps, i + 1, value, jdbcType);
  69. } catch (TypeException | SQLException e) {
  70. throw new TypeException("Could not set parameters for mapping: " + parameterMapping + ". Cause: " + e, e);
  71. }
  72. }
  73. }
  74. }
  75. }
  76. }

为 SQL 语句绑定完实参之后,就可以调用 Statement 对象 相应的 execute 方法,将 SQL 语句交给数据库执行了。

SimpleStatementHandler

SimpleStatementHandler 继承了 BaseStatementHandler 抽象类。其底层使用 java.sql.Statement 来完成数据库的相关操作,所以 SQL 语句中不存在占位符,所以 SimpleStatementHandler 的 parameterize()方法是空实现。SimpleStatementHandler 的 instantiateStatement()方法直接通过 JDBC Connection 创建 Statement 对象。

  1. public class SimpleStatementHandler extends BaseStatementHandler {
  2. // 构造方法主要用于属性的初始化
  3. public SimpleStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
  4. super(executor, mappedStatement, parameter, rowBounds, resultHandler, boundSql);
  5. }
  6. // 直接通过Connection创建Statement对象
  7. @Override
  8. protected Statement instantiateStatement(Connection connection) throws SQLException {
  9. if (mappedStatement.getResultSetType() == ResultSetType.DEFAULT) {
  10. // 如果结果集类型是DEFAULT默认的,则直接用connection创建Statement对象
  11. return connection.createStatement();
  12. } else {
  13. // 否则,设置结果集类型,设置结果集 只读
  14. return connection.createStatement(mappedStatement.getResultSetType().getValue(), ResultSet.CONCUR_READ_ONLY);
  15. }
  16. }
  17. // 上面创建的Statement对象会被本方法用于完成数据库查询操作
  18. @Override
  19. public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
  20. // 获取SQL语句
  21. String sql = boundSql.getSql();
  22. // 发送请求 执行SQL语句
  23. statement.execute(sql);
  24. // 从statement中获取结果集,并进行映射处理
  25. return resultSetHandler.handleResultSets(statement);
  26. }
  27. // 下面的batch()及queryCursor()方法的实现与上面的query()方法非常类似
  28. @Override
  29. public void batch(Statement statement) throws SQLException {
  30. String sql = boundSql.getSql();
  31. statement.addBatch(sql);
  32. }
  33. @Override
  34. public <E> Cursor<E> queryCursor(Statement statement) throws SQLException {
  35. String sql = boundSql.getSql();
  36. statement.execute(sql);
  37. return resultSetHandler.handleCursorResultSets(statement);
  38. }
  39. // 本方法用于执行insert、delete、update等类型的SQL语句,并且会根据配置的
  40. // KeyGenerator获取数据库生成的主键
  41. @Override
  42. public int update(Statement statement) throws SQLException {
  43. // 获取SQL语句 及parameterObject
  44. String sql = boundSql.getSql();
  45. Object parameterObject = boundSql.getParameterObject();
  46. // 获取配置的KeyGenerator 数据库主键生成器
  47. KeyGenerator keyGenerator = mappedStatement.getKeyGenerator();
  48. int rows;
  49. if (keyGenerator instanceof Jdbc3KeyGenerator) {
  50. // 执行SQL语句
  51. statement.execute(sql, Statement.RETURN_GENERATED_KEYS);
  52. // 获取更新的条数
  53. rows = statement.getUpdateCount();
  54. // 将数据库生成的主键添加到parameterObject中
  55. keyGenerator.processAfter(executor, mappedStatement, statement, parameterObject);
  56. } else if (keyGenerator instanceof SelectKeyGenerator) {
  57. // 执行SQL语句
  58. statement.execute(sql);
  59. // 获取更新的条数
  60. rows = statement.getUpdateCount();
  61. // 执行<selectKey>节点中配置的SQL语句,将从数据库获取到的主键 添加到parameterObject中
  62. keyGenerator.processAfter(executor, mappedStatement, statement, parameterObject);
  63. } else {
  64. statement.execute(sql);
  65. rows = statement.getUpdateCount();
  66. }
  67. return rows;
  68. }
  69. @Override
  70. public void parameterize(Statement statement) {
  71. // N/A
  72. }
  73. }

PreparedStatementHandler

PreparedStatementHandler 底层依赖于 java.sql.PreparedStatement 来完成数据库的相关操作。其中的 parameterize()方法中,会调用前面介绍的 ParameterHandler 的 setParameters()方法 完成 SQL 语句的参数绑定。instantiateStatement()方法直接调用 JDBC Connection 的 prepareStatement()方法创建 PreparedStatement 对象。

  1. public class PreparedStatementHandler extends BaseStatementHandler {
  2. // 构造方法主要用于属性的初始化
  3. public PreparedStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
  4. super(executor, mappedStatement, parameter, rowBounds, resultHandler, boundSql);
  5. }
  6. @Override
  7. protected Statement instantiateStatement(Connection connection) throws SQLException {
  8. // 获取SQL语句
  9. String sql = boundSql.getSql();
  10. // 根据mappedStatement持有的KeyGenerator的类型进行不同的处理
  11. if (mappedStatement.getKeyGenerator() instanceof Jdbc3KeyGenerator) {
  12. // 获取主键列
  13. String[] keyColumnNames = mappedStatement.getKeyColumns();
  14. if (keyColumnNames == null) {
  15. // 返回数据库生成的主键
  16. return connection.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
  17. } else {
  18. // 在insert语句执行完后,会将keyColumnNames指定的列返回
  19. return connection.prepareStatement(sql, keyColumnNames);
  20. }
  21. } else if (mappedStatement.getResultSetType() == ResultSetType.DEFAULT) {
  22. // 如果结果集类型是DEFAULT默认的,则直接通过connection获取PreparedStatement对象
  23. return connection.prepareStatement(sql);
  24. } else {
  25. // 否则,设置结果集类型,设置结果集为只读
  26. return connection.prepareStatement(sql, mappedStatement.getResultSetType().getValue(), ResultSet.CONCUR_READ_ONLY);
  27. }
  28. }
  29. // 因为是PrepareStatement对象,所以需要处理占位符"?"
  30. // 使用了前面介绍的ParameterHandler组件完成
  31. @Override
  32. public void parameterize(Statement statement) throws SQLException {
  33. parameterHandler.setParameters((PreparedStatement) statement);
  34. }
  35. // 下面的这些方法,除了多了一步 将Statement对象强转成PreparedStatement对象
  36. // 其它的几乎与SimpleStatementHandler一样
  37. @Override
  38. public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
  39. PreparedStatement ps = (PreparedStatement) statement;
  40. ps.execute();
  41. return resultSetHandler.handleResultSets(ps);
  42. }
  43. @Override
  44. public void batch(Statement statement) throws SQLException {
  45. PreparedStatement ps = (PreparedStatement) statement;
  46. ps.addBatch();
  47. }
  48. @Override
  49. public <E> Cursor<E> queryCursor(Statement statement) throws SQLException {
  50. PreparedStatement ps = (PreparedStatement) statement;
  51. ps.execute();
  52. return resultSetHandler.handleCursorResultSets(ps);
  53. }
  54. @Override
  55. public int update(Statement statement) throws SQLException {
  56. PreparedStatement ps = (PreparedStatement) statement;
  57. ps.execute();
  58. int rows = ps.getUpdateCount();
  59. Object parameterObject = boundSql.getParameterObject();
  60. KeyGenerator keyGenerator = mappedStatement.getKeyGenerator();
  61. keyGenerator.processAfter(executor, mappedStatement, ps, parameterObject);
  62. return rows;
  63. }
  64. }

另外,StatementHandler 接口还有一个 CallableStatementHandler 的实现。其底层依赖于 java.sql.CallableStatement 调用指定的存储过程,其 parameterize()方法也会调用 ParameterHandler 的 setParameters()方法完成 SQL 语句的参数绑定,并指定输出参数的索引位置和 JDBC 类型。其余方法与前面介绍的 ResultSetHandler 实现类似,唯一区别是会调用 ResultSetHandler 的 handleOutputParameters()方法 处理输出参数。

看到这里,我们可以发现 StatementHandler 组件依赖 ParameterHandler 组件 和 ResultSetHandler 组件 完成了 MyBatis 的核心功能,它控制着参数绑定、SQL 语句执行、结果集映射等一系列核心流程。