使用 Java Config 覆盖 Autowire 设置器

Question

考虑以下类：

public class MyBean {
    private A a;

    @Autowired(required=true)
    public void setA(A a) {
        this.a = a;
    }

    public A getA() {
        return a;
    }
}

在某些情况下需要覆盖自动装配的注入，例如当 Spring 无法找到注入的单个候选者时。在 XML 中，我可以有以下示例：

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd">

    <bean id="first" class="my.pkg.AImpl"/>
    <bean id="second" class="my.pkg.AImpl"/>
    <bean id="myBeanFirst" class="my.pkg.MyBean">
        <property name="a" ref="first"/>
    </bean>
    <bean id="myBeanSecond" class="my.pkg.MyBean">
        <property name="a" ref="second"/>
    </bean>

</beans>

有没有办法用 Java Config 做同样的事情？以下内容不起作用（我理解为什么），因为 Spring 在从 myBean 方法返回后尝试自动装配该属性，但它因 NoUniqueBeanDefinitionException 而失败：

@Configuration
public class MyConfig {
    @Bean
    public A first() {
        return new AImpl();
    }

    @Bean
    public A second() {
        return new AImpl();
    }

    @Bean
    public MyBean myBeanFirst(A first) {
        MyBean myBean = new MyBean();
        myBean.setA(first);
        return myBean;
    }

    @Bean
    public MyBean myBeanSecond(A second) {
        MyBean myBean = new MyBean();
        myBean.setA(first);
        return myBean;
    }
}

修改 MyBean 类并不总是一种选择，例如因为它来自外部库。 这是我必须使用 XML 配置的情况吗？

谢谢， 安德烈亚·波尔奇

更新 感谢您提供两种解决方案（按名称注入和使用@Primary），但它们不能解决我的用例，所以我认为我需要更具体。

在我的用例中，MyBean 类来自外部库，因此无法对其进行任何更改。我还需要有多个 MyBean 实例，每个实例都注入不同的 A 接口实例。我已经更新了上面的代码以反映这一点（xml 和 java）。

有没有使用 java config 的解决方案？是否可以避免自动装配对 MyBean 的依赖？ （仅在该类的 bean 上，不会为上下文中的每个 bean 完全禁用自动装配）

Answer 1

好的，我相信这个答案会满足您的需求。

我们需要的是MergedBeanDefinitionPostProcessor 的实现，它将为MyBean 类的属性a 设置正确的值。这可以通过下面的类来完成

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.support.MergedBeanDefinitionPostProcessor;
import org.springframework.beans.factory.support.RootBeanDefinition;
import org.springframework.stereotype.Component;


@Component
public class MyBeanPostProcessor implements MergedBeanDefinitionPostProcessor {


    @Override
    public void postProcessMergedBeanDefinition(RootBeanDefinition beanDefinition, Class<?> beanType, String beanName) {
        if(beanName.equals("myBeanFirst")) {
            beanDefinition.getPropertyValues().add("a", getMyBeanFirstAImpl());
        }
        else if(beanName.equals("myBeanSecond")) {
            beanDefinition.getPropertyValues().add("a", getMyBeanSecondAImpl());
        }
    }

    private Object getMyBeanFirstAImpl() {
        return new AImpl();
    }

    private Object getMyBeanSecondAImpl() {
        return new AImpl();
    }

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }
}

正如您所见，此处的 bean 名称是硬编码的，但它们可以设置为静态 final String，以下代码中的 @Bean 注释也将使用这些名称

@Configuration
public class Configuration {



    @Bean
    public MyBean myBeanFirst() {
        return new MyBean();
    }

    @Bean
    public MyBean myBeanSecond() {
        return new MyBean();
    }

}

您会注意到在下面的代码中，在 MyBean 创建方法中没有调用 setA，因为无论我们设置什么值（或者在这种情况下不设置），当 Spring 执行 bean 后处理器时都会被覆盖。

如果您需要 A 的默认值（例如，如果您将其注入其他 bean），请继续在之前的配置中定义 @Bean

Answer 2

如果您有多个候选注入，您可以通过在要成为默认 AImpl 的 bean 上使用 @Primary 来指定默认一个。这样就不需要对 MyBean 进行任何更改

Answer 3

如果您希望首先使用 myBeanFirst，则只需调用 first()。 myBeanSecond 也一样，希望它使用第二个，然后在设置 A 时调用second()；

@Configuration
public class MyConfig {
    @Bean
    public A first() {
        return new AImpl();
    }

    @Bean
    public A second() {
        return new AImpl();
    }

    @Bean
    public MyBean myBeanFirst() {
        MyBean myBean = new MyBean();
        myBean.setA(first());
        return myBean;
    }

    @Bean
    public MyBean myBeanSecond() {
        MyBean myBean = new MyBean();
        myBean.setA(second());
        return myBean;
    }
}