【Mybatis源码解析第五章】Mybatis是如何执行sql的

快来打我* 2023-07-06 06:30 33阅读 0赞

# 缘起 #

在上一章中，我们讲述了Mybatis是如何解析它的配置文件的。在解析的过程中，我们得到了一个至关重要的对象：Configuration对象，那么我们的sql又是在什么时候执行的呢？如何执行的呢？让我们来一起探究`selectOne()`背后的故事

Test test = sqlSession.selectOne("TestMapper.selectTest", 1);

`selectOne()`其实是调用的`selectList()`然后判断，如果结果集大于一条，就报错：

Expected one result (or null) to be returned by selectOne(), but found:XXX

关键代码如下

public <T> T selectOne(String statement, Object parameter) { 
      // Popular vote was to return null on 0 results and throw exception on too many.
      List<T> list = this.selectList(statement, parameter);
        if (list.size() == 1) { 
          return list.get(0);
        } else if (list.size() > 1) { 
          throw new TooManyResultsException("Expected one result (or null) to be returned by selectOne(), but found: " + list.size());
        } else { 
        return null;
      }
    }

`selectLis`方法如下

public <E> List<E> selectList(String statement, Object parameter, RowBounds rowBounds) { 
      try { 
        MappedStatement ms = configuration.getMappedStatement(statement);
        return executor.query(ms, wrapCollection(parameter), rowBounds, Executor.NO_RESULT_HANDLER);
      } catch (Exception e) { 
        throw ExceptionFactory.wrapException("Error querying database. Cause: " + e, e);
      } finally { 
        ErrorContext.instance().reset();
      }
    }

`configuration.getMappedStatement(statement);`就是前往`Configuration`对象的`mappedStatements`属性中获取到`MappedStatement`，我们知道`MappedStatement`保存着我们的`sql`

public MappedStatement getMappedStatement(String id, boolean validateIncompleteStatements) { 
      if (validateIncompleteStatements) { 
        buildAllStatements();
      }
      return mappedStatements.get(id);
    }

拿到了我们的`MappedStatement`之后，想必就是去取出对应的`sql`然后执行了。我们看`executor.query()`方法  
该方法最终会调用到`SimpleExecutor`的`doQuery`方法

public <E> List<E> doQuery(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException { 
      Statement stmt = null;
      try { 
        Configuration configuration = ms.getConfiguration();
        StatementHandler handler = configuration.newStatementHandler(wrapper, ms, parameter, rowBounds, resultHandler, boundSql);
        stmt = prepareStatement(handler, ms.getStatementLog());
        return handler.query(stmt, resultHandler);
      } finally { 
        closeStatement(stmt);
      }
    }

上诉`prepareStatement`方法最终会调用到`instantiateStatement`方法，该方法代码如下

protected Statement instantiateStatement(Connection connection) throws SQLException { 
      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() == ResultSetType.DEFAULT) { 
        return connection.prepareStatement(sql);
      } else { 
        return connection.prepareStatement(sql, mappedStatement.getResultSetType().getValue(), ResultSet.CONCUR_READ_ONLY);
      }
    }

看到了吧？从对应的`MapperStatement`对象中取出`BoundSql`对象，然后取出`BoundSql`中的`sql`，然后使用与`JDBC`中相同的方式封装成一个`Statement`对象。  
还记得`JDBC`中的代码吧？`JDBC`关键代码

PreparedStatement preparedStatement = connection.prepareStatement("select * from test where id = ?");
    preparedStatement.setString(1,"1");
    ResultSet resultSet = preparedStatement.executeQuery();

这个`connection.prepareStatement`是不是与`Mybatis`源码中的一模一样呢？  
我们再回到前面的`doQuery`方法，其中有一句代码`handler.query(stmt, resultHandler);`  
这个query方法的代码如下

public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException { 
      PreparedStatement ps = (PreparedStatement) statement;
      ps.execute();
      return resultSetHandler.handleResultSets(ps);
    }

想必你已经看到了那个`execute()`了，现在明白了吧？

## 总结 ##

当我们执行`sqlSession.selectOne()`这句代码时，`Mybatis`会去解析配置文件的时候就构建好的`Configuration`对象中的`mapperStatements`属性中通过我们传入的id取到对应的`MapperStatement`对象，然后从`MapperStatement`对象中取出`BoundSql`对象，然后取出`BoundSql`中的SQL语句，最后使用JDBC的方式执行。

那么下一章，我们来讲解，`Mybatis`是如何封装返回结果到我们的属性映射中的。