Spring中的@Conditional注解

代码语言：javascript复制

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE, ElementType.METHOD})
public @interface Conditional {
    Class<? extends Condition>[] value(); // 这里需要实现Condition接口
}

代码语言：javascript复制

public class UCloudConditional implements Condition {
    private Logger log = LoggerFactory.getLogger(UCloudConditional.class);
    private Properties properties;

    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) {
        properties = new Properties();
        InputStream is = null;
        try {
            is = conditionContext.getResourceLoader().getResource("classpath:xxx.properties").getInputStream();
            properties.load(is);
            return StringUtils.isEmpty(properties.getProperty("sms.used"))
                    ? false
                    : SupplierConstant.UCLOUD.equalsIgnoreCase(properties.getProperty("sms.used"));
        } catch (Exception e) {
            log.error("UCloudConditional#matches, load properties fail,e:", e);
            return false;
        } finally {
            if (is != null) {
                try {
                    is.close();
                } catch (IOException e) {
                    log.error("UCloudConditional#matches, load properties fail,e:", e);
                }
            }
        }

    }

代码语言：javascript复制

public class ClConditional implements Condition {
    private Properties properties;

    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata annotatedTypeMetadata) {
        properties = new Properties();
        InputStream is = null;
        try {
            is = conditionContext.getResourceLoader().getResource("classpath:xxx.properties").getInputStream();
            properties.load(is);
            return StringUtils.isEmpty(properties.getProperty("sms.used"))
                    ? true
                    : SupplierConstant.CHUANG_LAN.equalsIgnoreCase(properties.getProperty("sms.used"));
        } catch (Exception e) {
            log.error("ClConditional#matches, load properties fail,e:", e);
            return true; // CL作为默认使用的短信服务商
        } finally {
            if (is != null) {
                try {
                    is.close();
                } catch (IOException e) {
                    log.error("ClConditional#matches, load properties fail,e:", e);
                }
            }
        }

    }

代码语言：javascript复制

@Configuration
public class SMSMessageConfiguration {

    @Bean
    @Conditional(UCloudConditional.class)
    public ISendMessageTemplate createUCloudTemplate() {
        return new UCloudSendMessageTemplate();
    }

    @Bean
    @Conditional(ClConditional.class)
    public ISendMessageTemplate createClTemplate() {
        return new ClSendMessageTemplate();
    }
}

代码语言：javascript复制

public interface ISendMessageTemplate {
    BaseResponse sendMessage(String phone, String content);

}

代码语言：javascript复制

@Service
public class SendMessageAdapter {
    private ISendMessageTemplate sendMessageTemplate;

    @Autowired
    public SendMessageAdapter(ISendMessageTemplate sendMessageTemplate) {
        this.sendMessageTemplate = sendMessageTemplate;
    }

    /**
     * 短信发送
     *
     * @param phone
     * @param content
     * @return
     */
    public BaseResponse sendMessage(String phone, String content) {
        return this.sendMessageTemplate.sendMessage(phone, content);
    }
}

#SpringBoot中的做法

代码语言：javascript复制

/**使用配置化的方式参与感更强，但需要更多额外的配置来处理，
 * 而使用Boot的方式提供一种开箱即用的机制相对来说更加灵活，只需要注入相关依赖或者
 * 取消相关依赖即可
 **/
 @Primary
    @Bean
    @ConditionalOnClass(value = xxx.xx.x.SendUSMSMessageParam.class)
    public ISendMessageTemplate createUCloudTemplate() {
        return new UCloudSendMessageTemplate();
    }


    @ConditionalOnMissingClass(value = "xxx.xx.x.SendUSMSMessageParam")
    @Bean
    public ISendMessageTemplate createClTemplate() {
        return new ClSendMessageTemplate();
    }

spring 注解

0 人点赞