美文网首页程序员
Java事务处理全解析(六)——使用动态代理

Java事务处理全解析(六)——使用动态代理

作者: 无知者云 | 来源:发表于2017-04-06 22:26 被阅读0次

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

    这是一个关于Java事务处理的系列文章,请通过以下方式下载github源代码:

    git clone https://github.com/davenkin/java_transaction_workshop.git
    

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

    首先定义一个TransactionEnabledProxyManager类:

    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即可:

    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中通常是通过配置实现的):

    public class AnnotationBankServiceTest extends BankFixture
    {
        @Test
        public void transferSuccess() throws SQLException
        {
            TransactionManager transactionManager = new TransactionManager(dataSource);
            TransactionEnabledAnnotationProxyManager transactionEnabledProxyManager = new TransactionEnabledAnnotationProxyManager(transactionManager);
            BankService bankService = new AnnotationBankService(dataSource);
            BankService proxyBankService = (BankService) transactionEnabledProxyManager.proxyFor(bankService);
            proxyBankService.transfer(1111, 2222, 200);
    
            assertEquals(800, getBankAmount(1111));
            assertEquals(1200, getInsuranceAmount(2222));
        }
    
        @Test
        public void transferFailure() throws SQLException
        {
            TransactionEnabledAnnotationProxyManager transactionEnabledAnnotationProxyManager = new TransactionEnabledAnnotationProxyManager(new TransactionManager(dataSource));
            BankService bankService = new AnnotationBankService(dataSource);
            BankService proxyBankService = (BankService) transactionEnabledAnnotationProxyManager.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注解一样。

    相关文章

      网友评论

        本文标题:Java事务处理全解析(六)——使用动态代理

        本文链接:https://www.haomeiwen.com/subject/sohcattx.html