spring - 如何仅在两个配置文件处于活动状态时触发 bean 配置

Question

我有一个只有在两个配置文件“演示”和“本地”都处于活动状态时才应该创建的 bean。在基于 java 的 Spring 配置中实现这一目标的最佳方法是什么。

到目前为止，我想出的是，创建如下 bean：

@Profile("demo")
@Bean("isDemoActive")
public Boolean isDemoActive(){ return true;}

并将那些注入到 bean 创建方法中并对这些 bean 执行 if 条件。

有没有更好/更简单的方法来做这种事情？

Answer 1

根据我上面的评论，这是我的建议：

import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.core.type.AnnotatedTypeMetadata;

public class DoubleProfilesCondition implements Condition {
    public boolean matches(ConditionContext context,AnnotatedTypeMetadata metadata) {

        String[] activeProfiles = context.getEnvironment().getActiveProfiles();

        int counter = 0;
        for (int i = 0; i < activeProfiles.length; i++) {
            String profile = activeProfiles[i];
            if (profile.equals("profile1") || profile.equals("profile2")) {
                counter++;
            }
        }

        if (counter == 2)
            return true;
        return false;
   }
}

以及指示创建哪些 bean 的类：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Conditional;
import org.springframework.context.annotation.Configuration;

@Configuration
@Conditional(DoubleProfilesCondition.class)
public class MyConfig {

    public @Bean
    ExampleService service() {
        ExampleService service = new ExampleService();
        service.setMessage("hello, success!");
        return service;
    }
}

Answer 2

我编写了一个@ProfileAll与标准注释非常相似的注释，但是如果所有@Profile给定的配置文件当前都处于活动状态，则注释的配置、方法……仅由 spring 处理：

ProfileAll.java（注解）

import org.springframework.context.annotation.Conditional;

@Retention(RetentionPolicy.RUNTIME)
@Target({ ElementType.TYPE, ElementType.METHOD })
@Documented
@Conditional(ProfileAllCondition.class)
public @interface ProfileAll {

    /** The set of profiles for which the annotated component should be registered. */
    String[] value();

}

ProfileAllCondition.java（条件）

import org.springframework.context.annotation.Condition;
import org.springframework.context.annotation.ConditionContext;
import org.springframework.core.type.AnnotatedTypeMetadata;
import org.springframework.util.MultiValueMap;

class ProfileAllCondition implements Condition {

    @Override
    public boolean matches(ConditionContext context, AnnotatedTypeMetadata metadata) {
        if (context.getEnvironment() != null) {
            MultiValueMap<String, Object> attrs = metadata.getAllAnnotationAttributes(ProfileAll.class.getName());
            if (attrs != null) {
                LinkedList<String> profilesActive = new LinkedList<>();
                LinkedList<String> profilesNotActive = new LinkedList<>();
                List<Object> values = attrs.get("value");
                int count = 0;
                for (Object value : values) {
                    for (String profile : ((String[]) value)) {
                        count++;
                        if (context.getEnvironment().acceptsProfiles(profile)) {
                            profilesActive.add(profile);
                        } else {
                            profilesNotActive.add(profile);
                        }
                    }
                }
                if (profilesActive.size() == count) {
                    return true;
                } else {
                    return false;
                }
            }
        }
        return true;
    }
}

如下使用它，例如：

@Profile({ "mysql", "cloud", "special" })
@Configuration
public class MySqlConfig {
...

只有当所有三个配置文件在启动时都处于活动状态时，才会处理 MySqlConfig。