Spring事务使用ThreadLocal绑定连接并通过动态代理增强

Spring默认的事务是自动开启的，我么们没执行一次DAO就是获取了一次连接，这样无法实现多条操作统一提交，比如转账，这时我们一种解决思路就是将connection连接与当前线程绑定，这样我们这么线程在操作时，获取的都是同一个连接。

创建连接工具类

/**
 * 连接的工具类，它用于从数据源中获取一个连接，并且实现和线程的绑定
 */
public class ConnectionUtils {
    private ThreadLocal<Connection> tl = new ThreadLocal<Connection>();

    //提供一个set方法，在xml中让Spring帮我们注入
    private DataSource dataSource;

    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }
    /**
     * 获取当前线程上的连接
     *
     * @return
     */
    public Connection getThreadConnection() {
        //1、先从ThreadLocal上获取
        Connection conn = tl.get();
        try {
            //2、判断当前线程上是否有连接
            if (conn == null) {
                //3、从数据源中获取一个连接，并且存入ThreadLocal中
                conn = dataSource.getConnection();
                //绑定!!!!!!!!!要不每次都重新获取，不一样的连接对象，无法统一管理!
                tl.set(conn);
            }
            //4 返回当前线程上的连接
            return conn;
        } catch (SQLException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    /**
     * 把连接和线程解绑
     */
    public void removeConnection(){
        tl.remove();
    }
}

创建事务的工具类

在事务的工具类中，我们可以获取到线程上的这个唯一的连接，注意开启事务的时候需要将Spring自动的事务管理关闭

/**
 * 和事务管理相关的工具类，它包含了：
 * 1、开启事务 2、 提交事务 3、 回滚事务 4、 释放连接
 */
public class TransactionManager {

    //提供一个set方法，让Spring帮我们注入进来
    private ConnectionUtils connectionUtils;

    public void setConnectionUtils(ConnectionUtils connectionUtils) {
        this.connectionUtils = connectionUtils;
    }

    //开启事务（从池子中拿）
    public void beginTransaction() {
        try {
            connectionUtils.getThreadConnection().setAutoCommit(false);//关闭Spring 的自动事务
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    //提交事务
    public void commit() {
        try {
            connectionUtils.getThreadConnection().commit();//链式编程
            System.out.println("==========" + connectionUtils.getThreadConnection());
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    //回滚事务
    public void rollback() {
        try {
            connectionUtils.getThreadConnection().rollback();
            System.out.println(connectionUtils.getThreadConnection());
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    //释放连接（归还给池子中）
    public void release() {
        try {
            connectionUtils.getThreadConnection().close();//归还到池子中
            connectionUtils.removeConnection();//连接和线程解绑
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

为了将我们的业务层的代码简化，我们可以使用动态代理的方式，去增强我们的Service层的所有方法，在增强的时候，开启事务并提交或回滚事务。
这里我们使用Proxy类去代理我们的类，它是通过newProxyInstance获取的代理对象。
其参数说明：

参数名	作用
ClassLoader	它是用于加载代理对象的字节码的，和代理对象使用相同的类加载器，固定写法)
Class[]	他是用于让代理对象和被代理对象有相同的方法，所以我们传入被代理对象的实现的接口，就可以获得所有相同的方法，固定写法
InvocationHandler	它是让我们写如何代理，它是一个接口，我们一般写一个该接口的实现类，通常情况下都是匿名内部类，但也不是必须

/**
 * 使用动态代理支持事务
 */
public class BeanFactory {
    //由Spring在创建BeanFactory对象时注入
    private IAccountService accountService;
    //由Spring在创建BeanFactory对象时注入
    private TransactionManager txManager;

    //使用动态代理
    public IAccountService getProxyAccountService() {
        return (IAccountService) Proxy.newProxyInstance(accountService.getClass().getClassLoader(),
                accountService.getClass().getInterfaces(), new InvocationHandler() {
                    @Override
                    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                        Object returnValue = null; //返回值
                        try {
                            txManager.beginTransaction();//开启事务
                            //method就是动态代理对象调用什么，就是什么方法
                            returnValue = method.invoke(accountService, args);//不需要改变参数，只是开启事务
                            txManager.commit();//无异常就提交事务
                        } catch (Exception e) {
                            e.printStackTrace();
                            txManager.rollback();//回滚
                        } finally {
                            txManager.release();//归还连接对象，连接和线程解绑
                        }
                        return returnValue;
                    }
                });
    }

    //提供set方法，方便注入accountService
    public void setAccountService(IAccountService accountService) {
        this.accountService = accountService;
    }
    //提供set方法，方便注入txManager
    public void setTxManager(TransactionManager txManager) {
        this.txManager = txManager;
    }
}

使用xml的方式配置，并注入依赖

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xsi:schemaLocation="http://www.springframework.org/schema/beans
        http://www.springframework.org/schema/beans/spring-beans.xsd">

    <!--    配置BeanFactory-->
    <bean id="beanFactory" class="edu.xidian.factory.BeanFactory">
        <property name="txManager" ref="txManager"></property>
        <property name="accountService" ref="accountService"></property>
    </bean>
    <!--    通过工厂模式创建出代理对象-->
    <bean id="proxyAccountService" factory-bean="beanFactory" factory-method="getProxyAccountService"></bean>

    <!--    数据源-->
    <bean id="dataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource">
        <property name="driverClass" value="com.mysql.jdbc.Driver"></property>
        <property name="jdbcUrl" value="jdbc:mysql:///springtest"></property>
        <property name="user" value="root"></property>
        <property name="password" value="123456"></property>
    </bean>
    <!--queryRunner-->
    <bean id="queryRunner" class="org.apache.commons.dbutils.QueryRunner" scope="prototype">
        <!-- 不再需要注入连接了-->
    </bean>
    <!--accountService-->
    <bean id="accountService" class="edu.xidian.service.impl.AccountServiceImpl">
        <property name="accountDao" ref="accountDao"></property>
    </bean>
    <!--accountDao-->
    <bean id="accountDao" class="edu.xidian.dao.impl.AccountDaoImpl">
        <property name="queryRunner" ref="queryRunner"></property>
        <property name="connectionUtils" ref="connectionUtils"></property>
    </bean>
    <!--ConnectionUtils-->
    <bean id="connectionUtils" class="edu.xidian.utils.ConnectionUtils">
        <property name="dataSource" ref="dataSource"></property>
    </bean>
    <!--TransactionManager-->
    <bean id="txManager" class="edu.xidian.utils.TransactionManager">
        <property name="connectionUtils" ref="connectionUtils"></property>
    </bean>
</beans>