Java事务处理全解析（六）—— 使用动态代理（Dynamic Proxy）完成事务

在本系列的上一篇文章中，我们讲到了使用Template模式进行事务管理，这固然是一种很好的方法，但是不那么完美的地方在于我们依然需要在service层中编写和事务处理相关的代码，即我们需要在service层中声明一个TransactionTemplate。在本篇文章中，我们将使用Java提供的动态代理（Dynamic Proxy）功能来完成事务处理，你将看到无论是在service层还是DAO层都不会有事务处理代码，即他们根本就意识不到事务处理的存在。使用动态代理完成事务处理也是AOP的一种典型应用。

Java动态代理的基本原理为：被代理对象需要实现某个接口（这是前提），代理对象会拦截对被代理对象的方法调用，在其中可以全然抛弃被代理对象的方法实现而完成另外的功能，也可以在被代理对象方法调用的前后增加一些额外的功能。在本篇文章中，我们将拦截service层的transfer方法，在其调用之前加入事务准备工作，然后调用原来的transfer方法，之后根据transfer方法是否执行成功决定commit还是rollback。

首先定义一个TransactionEnabledProxyManager类：

package davenkin.step5_transaction_proxy;

import davenkin.step3_connection_holder.TransactionManager;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

public class TransactionEnabledProxyManager
{
    private TransactionManager transactionManager;

    public TransactionEnabledProxyManager(TransactionManager transactionManager)
    {

        this.transactionManager = transactionManager;
    }

    public Object proxyFor(Object object)
    {
        return Proxy.newProxyInstance(object.getClass().getClassLoader(), object.getClass().getInterfaces(), new TransactionInvocationHandler(object, transactionManager));
    }
}

class TransactionInvocationHandler implements InvocationHandler
{
    private Object proxy;
    private TransactionManager transactionManager;

    TransactionInvocationHandler(Object object, TransactionManager transactionManager)
    {
        this.proxy = object;
        this.transactionManager = transactionManager;
    }

    public Object invoke(Object o, Method method, Object[] objects) throws Throwable
    {
        transactionManager.start();
        Object result = null;
        try
        {
            result = method.invoke(proxy, objects);
            transactionManager.commit();
        } catch (Exception e)
        {
            transactionManager.rollback();
        } finally
        {
            transactionManager.close();
        }
        return result;
    }
}

通过调用该类的proxyFor方法，传入需要被代理的对象（本例中为service对象），返回一个代理对象。此后，在调用代理对象的transfer方法时，会自动调用TransactionIvocationHandler的invoke方法，在该方法中，我们首先开始事务，然后执行：

 result = method.invoke(proxy, objects);

上面一行代码执行的是原service层的transfer方法，如果方法执行成功则commit，否则rollback事务。



由于与事务处理相关的代码都被转移到了代理对象中，在service层中我们只需调用DAO即可：

package davenkin.step5_transaction_proxy;

import davenkin.BankService;
import davenkin.step3_connection_holder.ConnectionHolderBankDao;
import davenkin.step3_connection_holder.ConnectionHolderInsuranceDao;
import javax.sql.DataSource;

public class BareBankService implements BankService
{
    private ConnectionHolderBankDao connectionHolderBankDao;
    private ConnectionHolderInsuranceDao connectionHolderInsuranceDao;

    public BareBankService(DataSource dataSource)
    {
        connectionHolderBankDao = new ConnectionHolderBankDao(dataSource);
        connectionHolderInsuranceDao = new ConnectionHolderInsuranceDao(dataSource);
    }

    public void transfer(final int fromId, final int toId, final int amount)
    {
        try
        {
            connectionHolderBankDao.withdraw(fromId, amount);
            connectionHolderInsuranceDao.deposit(toId, amount);
        } catch (Exception e)
        {
            throw new RuntimeException();
        }
    }
}

如何，上面的BareBankService中没有任何事务处理的影子，我们只需关注核心业务逻辑即可。



然后在客户代码中，我们需要先创建代理对象（这在Spring中通常是通过配置实现的）：

@Test
    public void transferFailure() throws SQLException
    {
        TransactionEnabledProxyManager transactionEnabledProxyManager = new TransactionEnabledProxyManager(new TransactionManager(dataSource));
        BankService bankService = new BareBankService(dataSource);
        BankService proxyBankService = (BankService) transactionEnabledProxyManager.proxyFor(bankService);

        int toNonExistId = 3333;
        proxyBankService.transfer(1111, toNonExistId, 200);

        assertEquals(1000, getBankAmount(1111));
        assertEquals(1000, getInsuranceAmount(2222));
    }

在上面的测试代码中，我们首先创建一个BareBankService对象，然后调用transactionEnabledProxyManager的proxyFor方法生成对原BareBankService对象的代理对象，最后在代理对象上调用transfer方法，测试运行成功。



可以看到，通过以上动态代理实现，BareBankService中的所有public方法都被代理了，即他们都被加入到事务中。这对于service层中的所有方法都需要和数据库打交道的情况是可以的，本例即如此（有且只有一个transfer方法），然而对于service层中不需要和数据库打交道的public方法，这样做虽然也不会出错，但是却显得多余。在下一篇文章中，我们将讲到使用Java注解（annotation）的方式来声明一个方法是否需要事务，就像Spring中的Transactional注解一样。