MyBatis源码解析之Transaction事务模块

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

1、回顾

之前介绍了Environment环境类，这其实是一个单例类，在MyBatis运行开启后只会存在一个唯一的环境实例，虽然我们可以在Configuration配置文件中配置多个环境，但是项目运行中只会存在其中的一个，一般项目会存在开发环境和测试环境、生产环境三大环境，其是否可以设置到配置文件中，在开发时使用开发环境，测试时使用测试环境，正式运营时可以使用生产环境。

之前还提到Environment类中有三个字段，除了id之外，TransactionFactory和DataSource都是比较复杂的模块，这一次我们介绍Transaction模块（即事务模块）。

2、事务模块

事务模块位于org.apache.ibatis.transaction包，这个包内的类均是事务相关的类：

　　org.apache.ibatis.transaction
　　-----org.apache.ibatis.transaction.jdbc
　　----------JdbcTransaction.java
　　----------JdbcTransactionFactory.java
　　-----org.apache.ibatis.transaction.managed
　　----------ManagedTransaction.java
　　----------ManagedTransactionFactory.java
　　-----Transaction.java
　　-----TransactionException.java
　　-----TransactionFactory.java

从上面的类结构中也能看出来，MyBatis的事务模块采用的是工厂模式。

2.1 事务接口

位于org.apache.ibatis.transaction包的Transaction和TransactionFactory都是接口类。

Transaction是事务接口，其中定义了四个方法：

commit()-事务提交
rollBack()-事务回滚
close()-关闭数据库连接
getConnection()-获取数据库连接

如下代码所示：


package org.apache.ibatis.transaction;
import java.sql.Connection;
import java.sql.SQLException;

public interface Transaction {
  Connection getConnection() throws SQLException;
 void commit() throws SQLException;
 void rollback() throws SQLException;
 void close() throws SQLException;
}

TransactionFactory是事务工厂接口，其中定义了三个方法：

setProperties(Properties props)-设置属性
newTransaction(Connection conn)-创建事务实例
newTransaction(DataSource dataSource,TransactionIsolationLevel level,boolean autoCommit)-创建事务实例

如下代码所示：


package org.apache.ibatis.transaction;
import java.sql.Connection;
import java.util.Properties;
import javax.sql.DataSource;
import org.apache.ibatis.session.TransactionIsolationLevel;

public interface TransactionFactory {
 //设置属性
 void setProperties(Properties props);
 //根据Connection创建Transaction
 Transaction newTransaction(Connection conn);
 //根据数据源和事务隔离级别创建Transaction
 Transaction newTransaction(DataSource dataSource, TransactionIsolationLevel level, boolean autoCommit);
}

Transacrion接口定义的目的就是为了对具体的事务类型进行抽象，便于扩展；TransactionFactory与其一样，是对事务工厂的抽象，同样便于具体类型的事务工厂的扩展实现。

2.2 MyBatis事务类型

说到这里，就不得不提到MyBatis里内置的两种事务类型及对应的事务工厂了，还记得在上一文中给出的environment配置信息，有这么一句：


 <transactionManager type="JDBC"/>

这里的<transactionManager>标签就是用于定义项目所使用的事务类型，具体的类型由type属性来指定，此处指定使用“JDBC”类型事务，当然MyBatis还提供了另外一种“MANAGED”型事务。

---JDBC
---MANAGED

这里的“JDBC”和“MANAGED”是在Configuration配置类的类型别名注册器中注册的别名，其对应的类分别是：JdbcTransactionFactory.class和ManagedTransactionFactory.class。具体的配置如下所述：


typeAliasRegistry.registerAlias("JDBC", JdbcTransactionFactory.class);
typeAliasRegistry.registerAlias("MANAGED", ManagedTransactionFactory.class);

上面的代码是在Configuration类的无参构造器中定义的，这里拿来仅用于展示，具体说明以后会介绍。

这里提一句：类型别名注册器额原理就是将别名与具体的类类型以键值对的方式保存到一个HashMap中，这样只要知道别名（键），就可以从Map中得到对应的值（Class类型），很简单！

现在只要知道MyBatis能够根据你在配置文件中设置的事务类型，直接找到对应的事务工厂类就行了。

下面对上面提到的两种事务类型进行解读。

---JDBC事务模型：JdbcTransaction
---MANAFED事务模型：ManagedTransaction

二者的不同之处在于：前者是直接使用JDK提供的JDBC来管理事务的各个环节：提交、回滚、关闭等操作，而后者则什么都不做，那么后者有什么意义呢，当然很重要。　

当我们单独使用MyBatis来构建项目时，我们要在Configuration配置文件中进行环境（environment）配置，在其中要设置事务类型为JDBC，意思是说MyBatis被单独使用时就需要使用JDBC类型的事务模型，因为在这个模型中定义了事务的各个方面，使用它可以完成事务的各项操作。

而MANAGED类型的事务模型其实是一个托管模型，也就是说它自身并不实现任何事务功能，而是托管出去由其他框架来实现，你可能还不明白，这个事务的具体实现就交由如Spring之类的框架来实现，而且在使用SSM整合框架后已经不再需要单独配置环境信息（包括事务配置与数据源配置），因为在在整合jar包（mybatis-spring.jar）中拥有覆盖mybatis里面的这部分逻辑的代码，实际情况是即使你显式设置了相关配置信息，系统也会视而不见......

托管的意义显而易见，正是为整合而设。

我们学习MyBatis的目的正是由于其灵活性和与Spring等框架的无缝整合的能力，所以有关JDBC事务模块的内容明显不再是MyBatis功能中的重点，也许只有在单独使用MyBatis的少量系统中才会使用到。

2.3 JDBC事务模型

虽然JDBC事务类型很少使用到，但是作为MyBatis不可分割的一部分，我们还是需要进行一定的了解，JDBC事务的实现是对JDK中提供的JDBC事务模块的再封装，以适用于MyBatis环境。

MyBatis中的JDBC事务模块包括两个部分，分别为JDBC事务工厂和JDBC事务，整个事务模块采用的是抽象工厂模式，那么对应于每一项具体的事务处理模块必然拥有自己的事务工厂，事务模块实例通过事务工厂来创建（事务工厂将具体的事务实例的创建封装起来）。

首先我们来看JDBC事务工厂：JdbcTransactionFactory

JdbcTransactionFactory继承自TransactionFactory接口，实现了其中的所有方法。分别为一个设置属性的方法和两个新建事务实例的方法（参数不同），内容很简单，作用也很简单。

其中setProperties()方法用于设置属性，这个方法在XMLConfigBuilder中解析事务标签时调用，用于解析事务标签的下级属性标签<property>（一般情况下我们并不会进行设置，但是如果我们进行了设置，那就会覆盖MyBatis中的默认设置）之后将其设置到创建的事务实例中。然而针对JDBC事务模型，在事务工厂的设置属性方法中没有任何执行代码，也就说明JDBC事务模块并不支持设置属性的功能，即使你在配置文件中设置的一些信息，也不会有任何作用。

那么这个方法有什么用呢？前面提到，这个设置用于覆盖默认设置，只是JDBC事务模块并不支持而已，但并不代表别的事务模型不支持，同时这个方法也可用于功能扩展。

另外两个方法显而易见，就是用于创建JDBC事务实例的生产方法，只是参数不同，方法的重载而已。其中一个生产方法仅需传递一个实例Connection，这代表一个数据库连接。而另一个方法需要传递三个参数(DataSource、TransactionIsolationLevel、boolean)，其实这对应于MyBatis中SqlSession的两种生产方式，其参数与这里一一对应，这部分内容以后介绍，此处不再赘述。

然后我们来看看JDBC事务类：JdbcTransaction

其中有四个参数：


 protected Connection connection;
 protected DataSource dataSource;
 protected TransactionIsolationLevel level;
 protected boolean autoCommmit;

这四个参数分别对应事务工厂中的两个生产方法中的总共四个参数，对应的在事务类中定义了两个构造器，构造实例的同时进行赋值：


public JdbcTransaction(DataSource ds, TransactionIsolationLevel desiredLevel, boolean desiredAutoCommit) {
   dataSource = ds;
   level = desiredLevel;
   autoCommmit = desiredAutoCommit;
 }
 public JdbcTransaction(Connection connection) {
   this.connection = connection;
 }

其次在该类中实现了Transaction接口，实现了其中的四个方法，三个功能性方法，和一个获取数据库连接的方法。三个功能方法分别是提交、回滚和关闭。


  @Override
  public void commit() throws SQLException {
    if (connection != null && !connection.getAutoCommit()) {
      if (log.isDebugEnabled()) {
        log.debug("Committing JDBC Connection [" + connection + "]");
      }
      connection.commit();
    }
  }

 @Override
 public void rollback() throws SQLException {
   if (connection != null && !connection.getAutoCommit()) {
     if (log.isDebugEnabled()) {
       log.debug("Rolling back JDBC Connection [" + connection + "]");
     }
     connection.rollback();
   }
 }

 @Override
 public void close() throws SQLException {
   if (connection != null) {
     resetAutoCommit();
     if (log.isDebugEnabled()) {
       log.debug("Closing JDBC Connection [" + connection + "]");
     }
     connection.close();
   }
 }

通过观察这三个方法，可以发现，其中都使用了connection来进行具体操作，因此这些方法使用的前提就是先获取connection数据库连接，Connection的获取使用getConnection()方法


 @Override
 public Connection getConnection() throws SQLException {
   if (connection == null) {
     openConnection();
   }
   return connection;
 }

在上面的方法中调用了openConnection()方法：


 protected void openConnection() throws SQLException {
    if (log.isDebugEnabled()) {
      log.debug("Opening JDBC Connection");
    }
    connection = dataSource.getConnection();
    if (level != null) {
      connection.setTransactionIsolation(level.getLevel());
    }
    setDesiredAutoCommit(autoCommmit);
 }

可见connection是从数据源dataSource中获取的，最后会调用setDesiredAutoCommit()方法：


  protected void setDesiredAutoCommit(boolean desiredAutoCommit) {
    try {
      if (connection.getAutoCommit() != desiredAutoCommit) {
        if (log.isDebugEnabled()) {
          log.debug("Setting autocommit to " + desiredAutoCommit + " on JDBC Connection [" + connection + "]");
        }
        connection.setAutoCommit(desiredAutoCommit);
      }
    } catch (SQLException e) {
     // Only a very poorly implemented driver would fail here,
     // and there's not much we can do about that.
     throw new TransactionException("Error configuring AutoCommit.  "
         + "Your driver may not support getAutoCommit() or setAutoCommit(). "
         + "Requested setting: " + desiredAutoCommit + ".  Cause: " + e, e);
   }
 }

这个方法的目的就是为connection中的自动提交赋值（真或假）。

这么看来，我们创建事务实例所提供的三个参数就是为connection服务的，其中DataSource是用来获取Connection实例的，而TransactionIsolationLevel（事务级别）和boolean（自动提交）是用来填充connection的，通过三个参数我们获得了一个圆满的Connection实例，然后我们就可以使用这个实例来进行事务操作：提交、回滚、关闭。

2.4 关于自动提交

在之前的代码中我们能看到在关闭操作之前调用了一个方法：resetAutoCommit()：


 protected void resetAutoCommit() {
   try {
     if (!connection.getAutoCommit()) {
       // MyBatis does not call commit/rollback on a connection if just selects were performed.
       // Some databases start transactions with select statements
       // and they mandate a commit/rollback before closing the connection.
       // A workaround is setting the autocommit to true before closing the connection.
       // Sybase throws an exception here.
       if (log.isDebugEnabled()) {
        log.debug("Resetting autocommit to true on JDBC Connection [" + connection + "]");
      }
      connection.setAutoCommit(true);
    }
  } catch (SQLException e) {
    log.debug("Error resetting autocommit to true "
        + "before closing the connection.  Cause: " + e);
  }
}

这里相对自动提交做个解说，如果设置自动提交为真，那么数据库将会将每一个SQL语句当做一个事务来执行，为了将多条SQL当做一个事务进行提交，必须将自动提交设置为false，然后进行手动提交。一般在我们的项目中，都需要将自动提交设置为false，即将自动提交关闭，使用手动提交

这个方法中通过对connection实例中的自动提交设置（真或假）进行判断，如果为false，表明不执行自动提交，则复位，重新将其设置为true。（自动提交的默认值为true）这个操作执行在connection关闭之前。可以看做是连接关闭之前的复位操作。