初看Mybatis 源码 （三） SQL是怎么执行的

前面说到Java动态代理，Mybatis通过这种方式实现了我们通过getMapper方式得到的Dao接口，可以直接通过接口的没有实现的方法来执行sql。

AuthUserDao mapper = session.getMapper(AuthUserDao.class);

getMapper方法到底做了什么。跟踪getMapper方法，进入到 MapperProxyFactory 类的 newInstance(SqlSession sqlSession) 方法。

  @SuppressWarnings("unchecked")

  protected T newInstance(MapperProxy<T> mapperProxy) {

    return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy);

  }

  public T newInstance(SqlSession sqlSession) {

    final MapperProxy<T> mapperProxy = new MapperProxy<T>(sqlSession, mapperInterface, methodCache);

    return newInstance(mapperProxy);

  }

看过上一篇的例子，我们知道。最后是通过Proxy.newProxyInstance 来实现切入sql的。他的实现在MapperProxy的 invoke 方法里面。看下invoke方法：

  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    if (Object.class.equals(method.getDeclaringClass())) {
      return method.invoke(this, args);
    }
    final MapperMethod mapperMethod = cachedMapperMethod(method);
    return mapperMethod.execute(sqlSession, args);
  }

简单来说，在invoke里面做了如下事情：

1. 通过我们调用的方法，如本例中的 selectAuthUserByName ，接口名来组合成语句。本例中是 com.mybatis.dao.AuthUserDao.selectAuthUserByName 。其实使用过sqlSession的selectOne(String statmet)之类的语句都知道。这个可以唯一定位到我们在sql映射文件中配置的sql语句

2. 通过返回值类型，定位到的语句的类型。确定最后应该执行的方法。是执行查询、删除、添加、修改等等。

看下mybatis的是如何做的：

  public Object execute(SqlSession sqlSession, Object[] args) {

    Object result;

    if (SqlCommandType.INSERT == command.getType()) {

      Object param = method.convertArgsToSqlCommandParam(args);

      result = rowCountResult(sqlSession.insert(command.getName(), param));

    } else if (SqlCommandType.UPDATE == command.getType()) {

      Object param = method.convertArgsToSqlCommandParam(args);

      result = rowCountResult(sqlSession.update(command.getName(), param));

    } else if (SqlCommandType.DELETE == command.getType()) {

      Object param = method.convertArgsToSqlCommandParam(args);

      result = rowCountResult(sqlSession.delete(command.getName(), param));

    } else if (SqlCommandType.SELECT == command.getType()) {

      if (method.returnsVoid() && method.hasResultHandler()) {

        executeWithResultHandler(sqlSession, args);

        result = null;

      } else if (method.returnsMany()) {

        result = executeForMany(sqlSession, args);

      } else if (method.returnsMap()) {

        result = executeForMap(sqlSession, args);

      } else {

        Object param = method.convertArgsToSqlCommandParam(args);

        result = sqlSession.selectOne(command.getName(), param);

      }

    } else {

      throw new BindingException("Unknown execution method for: " + command.getName());

    }

    if (result == null && method.getReturnType().isPrimitive() && !method.returnsVoid()) {

      throw new BindingException("Mapper method '" + command.getName() 

          + " attempted to return null from a method with a primitive return type (" + method.getReturnType() + ").");

    }

    return result;

  }

可以看到的是，他是通过我们在sql配置文件中的语句类型来判断是执行哪种操作。然后通过返回类型来确定查询是查一条还是查多条记录。最后这种操作的方式，也是回归到了sqlSession中的CRUD的操作的方法。

下面看下，一条sql语句到底是怎么执行的？我们知道Mybatis其实是对JDBC的一个封装。假如我执行

session.update("com.mybatis.dao.AuthUserDao.updateAuthUserEmailByName", test@email.com);

语句，追踪下来，Executor、 BaseStatementHandler等等。在 SimpleExecutor 中有如下代码：

  public int doUpdate(MappedStatement ms, Object parameter) throws SQLException {

    Statement stmt = null;

    try {

      Configuration configuration = ms.getConfiguration();

      StatementHandler handler = configuration.newStatementHandler(this, ms, parameter, RowBounds.DEFAULT, null, null);

      stmt = prepareStatement(handler, ms.getStatementLog());

      return handler.update(stmt);

    } finally {

      closeStatement(stmt);

    }

  }

1. 首先获取相关配置信息，这个在初始化时，从配置文件中解析而来

2. 新建了一个handler

3. 做了执行statement之前的准备工作。看看准备了些什么，跟踪代码，最后进入了DataSource类的doGetConnection方法，该方法做如下操作：

  private Connection doGetConnection(Properties properties) throws SQLException {

    initializeDriver();

    Connection connection = DriverManager.getConnection(url, properties);

    configureConnection(connection);

    return connection;

  }

  private synchronized void initializeDriver() throws SQLException {

    if (!registeredDrivers.containsKey(driver)) {

      Class<?> driverType;

      try {

        if (driverClassLoader != null) {

          driverType = Class.forName(driver, true, driverClassLoader);

        } else {

          driverType = Resources.classForName(driver);

        }

        // DriverManager requires the driver to be loaded via the system ClassLoader.

        // http://www.kfu.com/~nsayer/Java/dyn-jdbc.html

        Driver driverInstance = (Driver)driverType.newInstance();

        DriverManager.registerDriver(new DriverProxy(driverInstance));

        registeredDrivers.put(driver, driverInstance);

      } catch (Exception e) {

        throw new SQLException("Error setting driver on UnpooledDataSource. Cause: " + e);

      }

    }

  }

原来是通过prepareStatement 来执行了 我们初始化jdbc的操作。Class.forName  DriverManager.getConnection. 这两步是在这里面完成的。

4. 将执行sql的部分交给handler

继续跟踪handler 可以看到SimpleStatementHandler 中。如下执行这个update语句

  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 {

      statement.execute(sql);

      rows = statement.getUpdateCount();

    }

    return rows;

  }

这边就完成了statement的操作，整个过程就是我们Jdbc的过程。原来真的就是对JDBC的简单封装。

其实Mybatis的整个执行过程，理解起来分为如下几个过程：

1. 加载配置文件

2. 解析配置文件，从配置文件中解析出来 datasource、mapper文件、事务配置等等。将配置信息保存在对象内

3. 调用相关语句，执行sql。在执行的方法中分别完成JDBC的一系列操作。