在Spring中基于自定义注释的方法调用?

时间:2016-06-12 14:00:33

标签: java spring spring-mvc spring-annotations java-annotations

我有自定义注释:

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Controller {
    EventType[] events() default EventType.MESSAGE;
}

B 中的方法使用它们如下:

@Controller(events = {EventType.MESSAGE, EventType.DIRECT_MESSAGE})
public void onMessage(Message msg) {    }

@Controller(events = {EventType.STAR_ADDED})
public void onStarAdded(Message msg) {    }

现在,我想根据另一个类 A 中的注释events值调用上述方法。换句话说,当类 A 收到类型为STAR_ADDED的事件时,我想要使用注释@Controller(events = {EventType.STAR_ADDED})调用类 B 中的所有方法。

我知道如何在Java中执行此操作,但 Spring 是否提供了任何API来执行此操作?如果是,代码片段也会有所帮助。

1 个答案:

答案 0 :(得分:3)

解决方案1:

您也可以这样做:

enum EventType {
    MESSAGE {
        @Override
        public void handleMessage(Service service, Message message) {
            service.onMessage(message);
        }
    },
    STAR_ADDED {
        @Override
        public void handleMessage(Service service, Message message) {
            service.onStarAdded(message);
        }

        public abstract void handleMessage(Service service, Message message);
    }
}

在您的其他课程中,您知道什么是"活跃"事件:

yourEvent.handleMessage(service, message);

解决方案2:

我不知道春天是否有精确的东西,否则你也可以使用反射。这是一个使用反射的例子(我更喜欢上面的解决方案=>没有反射的枚举):

for(Method method: Service.class.getDeclaredMethods()){
    Controller annotation = m.getAnnotation(Controller.class);
    for(EventType event: annotation.events()){
        if(event.equals(yourActiveEventType)){
            method.invoke(service, message);
        }
        return ...
    }
}

提示(不是解决方案)3:

我真的不认为以下内容适用于您的场景,但我想我会提到它... Spring AOP允许您在调用带注释的方法时触发一些代码(它是' s与你的情景相反),检查这个答案,但值得你阅读:aspectj-pointcut-for-all-methods-of-a-class-with-specific-annotation

@Around("execution(@Controller * com.exemple.YourService.*(..))")
public Object aroundServiceMethodAdvice(final ProceedingJoinPoint pjp)
   throws Throwable {
   // perform actions before

   return pjp.proceed();

   // perform actions after
}

解决方案4 :(在评论后添加)

使用org.reflections

<dependency>
    <groupId>org.reflections</groupId>
    <artifactId>reflections</artifactId>
    <version>0.9.10</version>
</dependency>

示例:

Service service = ...;
Message message = ...;

Set<Method> methods = 
        ReflectionUtils.getMethods(Service.class, ReflectionUtils.withAnnotation(Controller.class),ReflectionUtils.withParametersAssignableTo(Message.class));

for(Method m: methods){
    Controller controller = m.getAnnotation(Controller.class);
    for(EventType eventType: controller.value()){
        if(EventType.MESSAGE.equals(eventType)){
            m.invoke(service, message);
        }
    }
}

这假设您已经拥有对服务对象的引用(您的方法所在的位置)。

由于您使用的是Spring,如果您的服务是&#39;如果是春季管理,你可以从春天的背景中获得实例,你必须自己尝试一下,因为这有点与你的设计有关:

@Autowired
private ApplicationContext appContext;

Reflections r  = new Reflections(new MethodAnnotationsScanner(), "com.your.package");
Set<Method> methods = r.getMethodsAnnotatedWith(Controller.class);
for(Method m: methods){
    Controller controller = m.getAnnotation(Controller.class);
    for(EventType eventType: controller.value()){
        if(EventType.MESSAGE.equals(eventType)){
            String className = m.getDeclaringClass().getSimpleName();
            className = className.replaceFirst(className.substring(0,1), className.substring(0,1).toLowerCase());
            Object service = appContext.getBean(className);
            m.invoke(service, message);
        }
    }
}

如果您的类是Spring管理的,并且使用其默认的camelcase名称添加到上下文中,则此方法有效。

您可以简化逻辑,但我相信主要元素在那里。