我如何找到所有带有自定义注释的bean，并将其注入bean-manager的映射？

Question

我有一些这样的豆子：

@MyAnnotation(type = TYPE1)
@Component
public class First implements Handler {

@MyAnnotation(type = TYPE2)
@Component
public class Second implements Handler {

@MyAnnotation(type = TYPE3)
@Component
public class Third implements Handler {

@Component
public class BeanManager {

    private Map<Type, Handler> handlers = new HashMap<>();

    @Override
    public Handler getHandler(Type type) {
        Handler handler = map.get(type);
        if (handler == null)
            throw new IllegalArgumentException("Invalid handler type: " + type);
        return handler ;
    }
}

public BeanManager(First first, Second second, Third third){
  handlers.put(Type.TYPE1, first);
  handlers.put(Type.TYPE2, second);
  handlers.put(Type.TYPE3, third);
 }

    @PostConstruct
    public void init(){
       Map<String, Object> beansWithAnnotation = context.getBeansWithAnnotation(MyAnnotation.class);
       //get type from annotation
       ...
      //add to map
      handlers.put(type, bean);
     }

3）移动将在bins中搜索注释移动到BeanPostProcessor，并将setter添加到BeanManager:

@Component
public class MyAnnotationBeanPostProcessor implements BeanPostProcessor {

    private final BeanManager beanManager;

    public HandlerInAnnotationBeanPostProcessor(BeanManager beanManager) {
        this.beanManager = beanManager;
    }

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        Annotation[] annotations = bean.getClass().getAnnotations();
        for (Annotation annotation : annotations) {
            if (annotation instanceof MyAnnotation) {
                Type[] types = ((MyAnnotation) annotation).type();
                for (Type type: types) {
                    beanManager.setHandler(type, (Handler) bean);
                }
            }
        }
        return bean;
    }
}

但是在这个解决方案中，我不喜欢Sethandler方法。

Answer 1

让我们稍微改变一下

@Component
public class BeanManager {

    private Map<Type, Handler> handlers = new HashMap<>();

    //haven't tested, just an idea : )
    @Autowired
    public void setHandler(List<Handler> handlersList) {
        for(Handler handler : handlers) {
            Type type = handler.getClass().getAnnotation(MyAnnotation.class).type();
            handlers.put(type, handler);
        }
    }
}

如果你需要更多的灵感，请查看这个链接