实现一个简单的 AOP 功能

最新推荐文章于 2022-10-20 16:51:26 发布

不会编程的程序员li

最新推荐文章于 2022-10-20 16:51:26 发布

阅读量148

点赞数

分类专栏：学习笔记文章标签： java spring

本文链接：https://blog.csdn.net/qq_30087395/article/details/126075002

版权

学习笔记专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用JDK和Cglib动态代理在Java中实现事务管理。通过创建TransactionProxyFactory和TransactionCglibProxyFactory，将事务处理代码与业务逻辑分离，确保在出现异常时能够正确回滚事务。示例展示了如何在方法调用前后分别开始和提交事务，以及在发生异常时进行回滚。

摘要由CSDN通过智能技术生成

通过动态代理，将横切代码块与业务代码分离。

JDK 动态代理方式

public class TransactionProxyFactory<T> implements InvocationHandler {

    private final T t;

    public TransactionProxyFactory(T t) {
        this.t = t;
    }

    public <E extends T> T getTarget() {
        return (E) Proxy.newProxyInstance(t.getClass().getClassLoader(), t.getClass().getInterfaces(), this);
    }
    
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        try {
            TransactionManager.beginTransaction();

            Object result = method.invoke(t, args);

            TransactionManager.commit();
            return result;
        } catch (Exception e) {
            e.printStackTrace();
            TransactionManager.rollback();
            throw e;
        }
    }
}

Cglib 动态代理方式

public class TransactionCglibProxyFactory<T> implements MethodInterceptor {

    private final Class<? extends T> clazz;

    public TransactionCglibProxyFactory(Class<? extends T> clazz) {
        this.clazz = clazz;
    }

    public <E extends T> T getTarget() {
        Enhancer enhancer = new Enhancer();
        enhancer.setSuperclass(clazz);
        enhancer.setCallback(this);
        return (E) enhancer.create();
    }

    @Override
    public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
        try {
            TransactionManager.beginTransaction();

            Object result = proxy.invokeSuper(obj, args);

            TransactionManager.commit();
            return result;
        } catch (Exception e) {
            e.printStackTrace();
            TransactionManager.rollback();
            throw e;
        }
    }
}

使用时调用代理对象

结合了之前写的 IoC 容器。

private final TransferService transferService = new TransactionProxyFactory<>(
        (TransferService) BeanFactory.getBean("transferService")).getTarget();