有没有办法让@Bean创建的bean可以区分?

时间:2018-02-05 16:20:56

标签: java spring spring-annotations spring-ioc spring-config

这是一个充满异国情调的用例,因此需要耐心去理解,并且可能需要特殊的解决方案。

  

上下文

我正在创建一个与Spring一起使用的库,它对上下文中存在的特定bean实例执行自动操作,由@Bean方法而不是@ComponentScan创建。如果可能的话,bean应该不是按类型区分,而是通过其他方式区分,最好是工厂方法上的注释

这是理想的情况。例如。让我们说有两种产豆方法:

@Bean
public SomeType makeSome() {...}

@Bean
@Special
public SomeOtherType makeOther() {...}

这里,第二个bean是 special ,因为创建它的方法上有@Special注释。但任何使其可区分的机制都是一个选项

然后,我想以某种方式只获得特殊 bean。

  

警告

我知道如果所有bean都实现相同的接口,我可以按类型注入它们。但这应该尽可能透明地工作,尽可能少地改变现有应用程序。

  

潜在的方法

这是我想到的两种广泛方法:

1)Jack进入注册bean的过程,并透明地将bean实例包装到某种容器中(我非常确定这部分是可行的)。 E.g。

public void registerBean(Object bean, ApplicationContext ctx) {
   ctx.register(bean); //do the usual
   ctx.register(new Wrapper(bean); //register it wrapped as well
}

然后,注入所有类型为Wrapper的bean。这里的问题显然是重复......或者我可以动态生成一个实现Wrapper接口的代理实例,因此它可以同时充当原始的bean 作为包装器。我确实说我也可以使用异国情调的解决方案,不是吗?

2)Spring已经将bean 候选者与实际注册的bean区分开来(例如@ComponentScan可以按包,注释等过滤候选者)。我希望可以插入这个过程并获得一些候选描述符,这些描述符仍然包含一些有用的元数据(比如它们的工厂方法),这些元数据可以让我以后区分这些bean实例。 / p>

3 个答案:

答案 0 :(得分:2)

似乎你需要使用@Qualifier它提供区分bean的功能:

@Bean
@Qualifier("special")
class MyBean {}

@Bean
class OtherBean {
    @Qualifier("special")
    private MyBean bean;
}

您可以在那里阅读更多内容:https://spring.io/blog/2014/11/04/a-quality-qualifier

UPD(明白你在说什么:)) 您可能需要查看BeanDefinitionRegistryPostProcessor

这是一个用法示例:

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.AnnotatedGenericBeanDefinition;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanDefinitionRegistryPostProcessor;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import static java.util.Collections.unmodifiableMap;

/**
 * This is hack to collect all beans of some type in single map without eager initialization of those beans
 *
 * Usage:
 * 1. Register new bean of type {@link ServiceTrackingBeanPostProcessor} parametrized with class of
 *    beans you want to collect
 * 2. Now you can inject {@link ServiceTracker} parametrized with your type anywhere
 *
 * @param <T> Located type
 */
public class ServiceTrackingBeanPostProcessor<T> implements BeanPostProcessor, BeanDefinitionRegistryPostProcessor {
    private final ConcurrentMap<String, T> registeredBeans = new ConcurrentHashMap<>();
    private final Class<T> clazz;
    private final String beanName;

    public ServiceTrackingBeanPostProcessor(Class<T> clazz) {
        this.clazz = clazz;
        beanName = "locatorFor" + clazz.getCanonicalName().replace('.', '_');
    }

    @Override
    public Object postProcessBeforeInitialization(Object o, String s) throws BeansException {
        return o;
    }

    @Override
    public Object postProcessAfterInitialization(Object o, String s) throws BeansException {
        if (!clazz.isInstance(o)) {
            return o;
        }
        registeredBeans.putIfAbsent(s, (T) o);
        return o;
    }

    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {
        AnnotatedGenericBeanDefinition def = new AnnotatedGenericBeanDefinition(Wrapper.class);
        registry.registerBeanDefinition(beanName, def);
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        beanFactory.registerSingleton(beanName, new Wrapper(unmodifiableMap(registeredBeans)));
    }

    private class Wrapper extends AbstractServiceTracker<T> {
        public Wrapper(Map<String, T> services) {
            super(services);
        }
    }
}

您只需要通过bean名称将检查条件从clazz.isInstance更改为从应用程序上下文获取beanDefinition,您可以获得有关实例化和注释的几乎任何信息

答案 1 :(得分:1)

以下是获取@Special注释的所有bean的一种方法。

Map<String,Object> beans = applicationContext.getBeansWithAnnotation(Special.class);

参考:https://stackoverflow.com/a/14236573/1490322

编辑:上述答案似乎仅在使用@Special注释类时才有效,因此它不适用于您的场景。但同一问题this other answer可能有效。它使用ConfigurableListableBeanFactory中的元数据来标识其方法使用特定注释进行注释的任何bean。

答案 2 :(得分:0)

我认为如上所述@Qualifier注释是一种方法。但我会展示不同的例子:

@Bean
public SomeType makeSome() {...}

@Bean
public SomeType makeSomeOther() {...}

在你的组件(@Service)中你可以使用这个bean:

@Autowired
@Qualifier("makeSome")
SomeType makeSomeBean;

如您所见,可以通过bean名称(Bean分配与@Bean注释方法名称相同的名称)来区分具有相同类型的两个bean