Spring - 在运行时使用自定义限定符获取 bean

Question

我创建了一个自定义 Spring@Qualifier注释：

@Target({

        ElementType.FIELD,

        ElementType.METHOD,

        ElementType.PARAMETER, 

        ElementType.TYPE,

        ElementType.ANNOTATION_TYPE

})

@Retention(RetentionPolicy.RUNTIME)

@Qualifier

public @interface Database {

    String value() default "";

}

然后我将此注释应用于各种实现 Bean：

@Repository

@Database("mysql")

class MySqlActionRepository implements ActionRepository {}

@Repository

@Database("oracle")

class OracleActionRepository implements ActionRepository {}

@Repository

@Database("sqlserver")

class SqlServerActionRepository implements ActionRepository {}

现在，在运行时，只有这些 Bean 中的一个必须可用于注入，我创建了一个@PrimaryBean 方法。

@Bean

@Primary

ActionRepository actionRepository(

        final ApplicationContext applicationContext,

        final Configuration configuration) {

    final var database = configuration.getString("...");

    return BeanFactoryAnnotationUtils.qualifiedBeanOfType(

            applicationContext,

            ActionRepository.class,

            database

    );

}

但是，此解决方案不适用于我的自定义注释。它仅在使用标准时有效@Qualifier。

知道如何解决这个问题吗？

Answer 1

从这里看来，BeanFactoryAnnotationUtils不支持你的情况。但是我们可以结合 ApplicationContext'sgetBeansOfType()和findAnnotationOnBean()来达到同样的目的：@Bean@PrimaryActionRepository actionRepository(final ApplicationContext applicationContext,&nbsp; &nbsp; &nbsp; &nbsp; final Configuration configuration) {&nbsp; &nbsp; final var database = configuration.getString("...");&nbsp; &nbsp; Map<String, ActionRepository> beanMap = context.getBeansOfType(ActionRepository.class);&nbsp; &nbsp; for (Map.Entry<String, ActionRepository> entry : beanMap.entrySet()) {&nbsp; &nbsp; &nbsp; &nbsp; Database db = context.findAnnotationOnBean(entry.getKey(), Database.class);&nbsp; &nbsp; &nbsp; &nbsp; if (db != null && db.value().equals(database)) {&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; return entry.getValue();&nbsp; &nbsp; &nbsp; &nbsp; }&nbsp; &nbsp; }&nbsp; &nbsp; throw new RuntimeException("Cannot find the bean...");}