Spring 代理 Bean 中获取不到原始 Bean 对象注解的解决方法

2022-04-27 510

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 我在接受 mq 消息的时候，需要做一个重试次数限制，如果超过 maxNum 就发邮件告警，不再重试。所以我需要对 consumer 对象进行代理，然后如果超过异常次数，我直接返回成功，并且发送成功消息，但是我获取 consumer handler 方法的方式是通过 method.getAnnotation(XXClient.class) 方式，那么就会返回 null。

问题描述

我在接受 mq 消息的时候，需要做一个重试次数限制，如果超过 maxNum 就发邮件告警，不再重试。所以我需要对 consumer 对象进行代理，然后如果超过异常次数，我直接返回成功，并且发送成功消息，但是我获取 consumer handler 方法的方式是通过 method.getAnnotation(XXClient.class) 方式，那么就会返回 null。

问题示例代码

目标类, 我这里就之定义一个 test 方法，里面做一些个简单的打印。

@Component
public class TestBean {
    @Anno
    public void test() {
        System.out.println("test .....");
    }
}

代理逻辑逻辑处理, 主要就是做一个 @Around 的方法覆盖，保证在调用目标方法之前，先输出我插入的逻辑。

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface Anno {
    String key() default "100%";
}
@Aspect
@Component
public class AnnoAspect {
    @Around("@annotation(anno)")
    public Object anno(ProceedingJoinPoint point, Anno anno) throws Throwable {
        System.out.println("anno invoke!!!!!!");
        return point.proceed();
    }
}

调用点，通过 AnnotationConfigApplicationContext 获取 bean. 然后通过 getMethods() 获取所有的方法，最后查找 Anno 注解的 Method 对象。

AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext(BeanStart.class);
    TestBean bean = applicationContext.getBean(TestBean.class);
    Class<? extends TestBean> classz = bean.getClass();
    Method[] methods = classz.getMethods();
    for (Method m : methods) {
        Anno annotation = m.getAnnotation(Anno.class);
        if (annotation != null) {
            System.out.println(" ============= invoke test ===========");
            m.invoke(bean, new Object());
        }
    }

由于 m.getAnnotaion(Anno.class) 无法获取到注解信息，所以执行 test 方法失败，到此问题还原完毕，我们再来看看如何解决。

解决方案

通过 Anno ao = AnnotationUtils.findAnnotation(method, Anno.class); 方法获取即可。

有的代码是这样写的 :

String name = classz.getName();
boolean isSpringProxy = name.indexOf("SpringCGLIB$$") >= 0;
Method[] methods;
if (isSpringProxy) {
    methods = ReflectionUtils.getAllDeclaredMethods(AopUtils.getTargetClass(bean));
} else {
    methods = classz.getMethods();
}
// 省略部分代码
if (isSpringProxy) {
    annotation = AnnotationUtils.findAnnotation(method, MqClient.class);
} else {
    annotation = method.getAnnotation(Anno.class);
}