140.Mybatis连接池和事务

‍

我们在实际开发中都会使用连接池，因为它可以减少我们获取连接所消耗的时间。如果忘了什么是连接池，可以复习：JDBC连接池 (opens new window) 　　‍

连接池

连接池，本质上就是一个存储连接的容器，一个集合对象。该容器必须是线程安全的，不能两个线程拿到同一个连接；还必须实现队列的特性：先进先出。

使用连接池，可以省去一次查询中很耗时的部分：获取连接。

‍

Mybatis中的连接池

Mybatis连接池提供了3种方式的配置：在主配置文件SqlMapConfig.xml中的dataSource标签，type属性就是表示采用何种连接池方式，取值如下：

• POOLED：采用传统的javax.sql.DataSource​规范中的连接池，Mybatis中有针对规范的实现
• UNPOOLED：采用传统的获取连接的方式，虽然也实现javax.sql.DataSource​接口，但是并没有使用连接池的思想，相当于不使用连接池。
• JNDI 采用服务器提供的JNDI技术实现，来获取DataSource对象，不同的服务器所能拿到DataSource是不一样。注意：如果不是web或者Maven的war工程，是不能使用的。本系列使用的是Tomcat服务器，采用连接池就是dbcp连接池。

‍

我们可以比较下，分别使用POOLED和UNPOOLED，日志的输出：

​​

‍

‍

​​

‍

可以看到POOLED是有获取连接，用完了会将连接返回给连接池的（Return connection to pool），而UNPOOLED则没有，每次都是创建一个连接，用完就销毁。

‍

‍

MyBatis 内部分别定义了实现了​ java.sql.DataSource​ 接口的 PooledDataSource ，UnpooledDataSource类来表示 POOLED、UNPOOLED 类型的数据源。

public class PooledDataSource implements DataSource 
public class UnpooledDataSource implements DataSource{

‍

我们来看看UnpooledDataSource类获取Connection的源码，第5行就是使用原生的 JDBC 获取连接的方法，而initializeDriver方法则是注册驱动：也就是每次都注册驱动，获取连接，用完就关掉。

private Connection doGetConnection(Properties properties) throws SQLException {
this.initializeDriver();
    Connection connection = DriverManager.getConnection(this.url, properties);
    this.configureConnection(connection);
    return connection;
}

private synchronized void initializeDriver() throws SQLException {
    if (!registeredDrivers.containsKey(this.driver)) {
        try {
            Class driverType;
            if (this.driverClassLoader != null) {
                driverType = Class.forName(this.driver, true, this.driverClassLoader);
            } else {
                driverType = Resources.classForName(this.driver);
            }

            Driver driverInstance = (Driver)driverType.getDeclaredConstructor().newInstance();
            DriverManager.registerDriver(new DriverProxy(driverInstance));
            registeredDrivers.put(this.driver, driverInstance);
        } catch (Exception var3) {
            throw new SQLException("Error setting driver on UnpooledDataSource. Cause: " + var3);
        }
    }

}

‍

‍

而在PooledDataSource中，getConnection调用的是popConnection方法

public Connection getConnection() throws SQLException {
    return this.popConnection(this.dataSource.getUsername(), this.dataSource.getPassword()).getProxyConnection();
}

‍

popConnection部分源码如下：

private PooledConnection popConnection(String username, String password) throws SQLException {
    //.......
    while(conn == null) {
    this.lock.lock();

    try {
        if (!this.state.idleConnections.isEmpty()) {
            conn = (PooledConnection)this.state.idleConnections.remove(0);
            if (log.isDebugEnabled()) {
                log.debug("Checked out connection " + conn.getRealHashCode() 
               + " from pool.");
            }
        } else if (this.state.activeConnections.size() < this.poolMaximumActiveConnections) {
            conn = new PooledConnection(this.dataSource.getConnection(), this);
            if (log.isDebugEnabled()) {
                log.debug("Created connection " + conn.getRealHashCode() + ".");
            }
          else {
              PooledConnection oldestActiveConnection = (PooledConnection)this.state.activeConnections.get(0);
              long longestCheckoutTime = oldestActiveConnection.getCheckoutTime();
            // ........
          }
                

第3行，如果connection为null，则锁住（lock.lock();​，为了保证线程安全），然后第7行则判断是否有空余的连接（idle有空闲的意思），有则取出第一个，idleConnections.remove(0)​。在Mybatis中，共有两个连接池，一个是空闲的连接池，一个是活动的连接池。

如果没有在空闲的连接池中，没有空闲的连接了，则会判断 活动的连接池数量 是否小于设定的最大值（第13行），是则创建一个新的连接。

如果第13行判断不成立，说明活动连接池的连接数量，还未达到最大值，则获取最早创建的连接（第19行，oldestActiveConnection），并做一些设置，使其可以作为一个全新的connection使用。

这是Mybatis自己实现的连接池思想，不是用到了其他框架（例如C3P0）。

‍

而idleConnections​，就是一个数组而已：

public class PoolState {
    protected final List<PooledConnection> idleConnections = new ArrayList();
}

‍

‍

Mybatis中的事务

关于事务的概念，在学习数据库的时候大家应该都学过了，这里不再赘述。在Mybatis中，我们是通过SqlSession对象的 commit 方法和 rollback 方法实现事务的提交和回滚的，其底层也是调用 java.sql.Connection​的commit和 rollback方法，Mybatis部分源码如下：

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

public class JdbcTransaction implements Transaction {
    protected Connection connection;

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

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

}

‍

我们获取SqlSession对象的时候，默认是开启了事务的，也就是 Setting autocommit to false​，不自动提交；我们可以在获取SqlSession的时候，传入一个参数true，表明开启自动提交：

SqlSession session = factory.openSession(true);

‍

之后就不用再调用session.commit()​ 方法提交了。但我们一般不会设置自动提交，因为我们一般会多次与数据库交互，如果每次交互都提交，相当于没有事务控制。

‍