一次动态代理的填坑之旅

作者: 美团Java | 来源:发表于2018-07-21 23:42 被阅读486次

    占小狼,转载请注明原创出处,谢谢!

    背景

    想在现有的接口加上熔断降级或者限流的功能,比较好的方式是通过注解的方式,并基于动态代理进行实现,下面代码是Rhino的实现

    @Rhino
    public class ServiceImpl {
    
        @Degrade(rhinoKey = "syncMethod-0",  fallBackMethod = "fallbackMethod")
        public void method() throws Exception {
            int i = 1 / 0;
        }
    
        private String fallbackMethod() throws Exception {
            return "fallback";
        }
    }
    

    通过在方法添加@Degrade注解,很方便的赋予了method方法熔断降级功能,在该方法的失败率达到阈值时,就自动熔断,并调用降级方法。

    这里的动态代理并没有使用Spring的AOP,而是自己实现了BeanPostProcessorBeanFactoryPostProcessor接口,另外也实现了PriorityOrdered接口。

    在生成动态代理对象的时候,根据类是否有实现接口,选择使用JDK的Proxy还是使用Cglib。

    /**
     * @param clazz
     * @param origin
     * @return
     */
    private static Object createProxyService(Class clazz, Object origin) {
        Object proxy;
        if (clazz.getInterfaces().length > 0) {
            proxy = Proxy.newProxyInstance(clazz.getClassLoader(), clazz.getInterfaces(), new RhinoInvocationHandler(origin, clazz));
        } else {
            Enhancer enhancer = new Enhancer();
            enhancer.setSuperclass(clazz);
            enhancer.setCallback(new RhinoInvocationHandler(origin, clazz));
            proxy = enhancer.create();
        }
        return proxy;
    }
    

    这样的实现,在99%的情况下,是没有问题的,直到有一天,有两个业务同时反馈了一个问题,使用了熔断降级的注解之后,发现自身的Spring AOP注解失效了或者是直接启动异常,WTF,这个问题一直都没考虑过好吗...

    为什么会这样?Spring AOP到底干了什么,或者是自己的注解到底有什么问题?

    复现问题

    为了快速定位问题,并解决问题,最好的办法就是复现该问题,写了一个简单的Spring AOP的例子。

    @Component  //加入到IoC容器
    @Aspect  //指定当前类为切面类
    public class Aop {
    
        @Pointcut("execution(* com.dianping.rhino.aop.*.*(..))")
        public void pointCut(){
        }
    
        @Before("pointCut()")
        public void begin(){
            System.out.println("begin");
        }
    
        @After("pointCut()")
        public void close(){
            System.out.println("close");
        }
    
        @Around(value = "@annotation(MethodLog)")
        public void around() {
            System.out.println(" MethodLog ");
        }
    }
    

    这里简短的解释下Spring AOP各个注解的作用
    @Aspect
    切面,标识该类是一个切面类
    @Pointcut
    切入点,用来标识哪些方法是需要被添加切面的
    @Before
    在切入点,执行方法之前进行增强
    @After
    在切入点,执行方法之后进行增强
    @Around
    在该例子中,只有添加了@MethodLog注解的方法才会被增强

    一切准备就绪,开启DEBUG之旅,Spring的内部逻辑有点复杂,整个过程需要一点耐心。

    通过Debug发现,Rhino的代理对象Processor排在Spring AOP的Processor,意味着Rhino生成的代理对象,会传给Spring AOP的Processor再做一层代理,在Spring AOP生成代理对象的内部逻辑中,有这么一段判断逻辑。

    在createProxy方法中,会根据传入的beanClass,即上一个Processor处理过的对象,判断是否有实现接口。

    回到Rhino的实现,因为ServiceImpl类没有实现接口,所以内部会采用CGLIB的方式创建代理对象,我们来看下这个对象的接口。

    好家伙,默默的给加了一个Factory接口,这样在Spring AOP的处理中就当做有接口的情况进行实现了。这里最大的问题是,最终生成的代理对象是Factory类型的对象,在赋值给ServiceImpl变量时就会抛异常了。

    解决问题

    终于找到了问题的所在,那么改如何解决呢?
    无法修改Spring AOP的逻辑,但是可以控制Rhino的逻辑,只需要把Rhino的Processor移到Spring AOP的Processor之后,这样就可以在Rhino的Proccessor中处理经过Spring AOP的代理过的beanClass对象了,有效的避免这个问题。

    如何有效调整Processor的处理顺序?
    通过分析发现Spring AOP的内部实现基于Ordered接口,而Rhino的实现是基于PriorityOrdered接口,而且处理器在初始化完成后会进行排序,实现PriorityOrdered接口的会放到前面,相同类型的再根据设置的order进行排序。很显然,Rhino的Processor被放在前面,找到了问题,解决方法也很简单,Rhino也换成Order接口,并且order设置成LOWEST_PRECEDENCE,即排在最后面。

    结论

    经过上面的调整之前,这个问题确实被有效的解决了,以前遇到Spring的异常,都是一脸懵逼。所以,遇到类似的问题,最好通过DEBUG源码去发现问题,并解决问题,这样可以有效的防止后续的继续挖坑。

    相关文章

      网友评论

      • 68号小喇叭:Spring会根据自己的命名规则判断是否cglib代理,可以试试用Spring自带的enhancer和命名policy是否可行,这样可以实现动态代理的叠加

      本文标题:一次动态代理的填坑之旅

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