在BeanDefinitionRegistryPostProcessor中访问弹簧加载的属性

时间:2019-02-14 12:56:19

标签: spring spring-bean property-placeholder

如何访问<context:property-placeholder>BeanDefinitionRegistryPostProcessor.postProcessBeanDefinitionRegistry加载的属性。

当我使用@Value时,它不会加载属性。

1 个答案:

答案 0 :(得分:1)

仅在对@Value进行后期处理之后,才设置带有BeanDefinitionRegistry注释的字段的值,这意味着它们在初始化过程的此阶段不可用。

但是,您可以显式扫描配置环境并从那里读取相关属性的值,然后在动态bean定义中使用它们。

要获得对配置环境的访问权限,您可以使用带有BeanDefinitionRegistryPostProcessor作为参数的,用@Bean注释的方法来创建ConfigurableEnvironment

请参见以下示例:

package com.sample.spring;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanDefinitionRegistryPostProcessor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.EnumerablePropertySource;
import org.springframework.core.env.PropertySource;

@Configuration
public class DynamicBeanConfig {

    private static final String PROPERTY_KEY = "somename"; 

    @Bean
    public BeanDefinitionRegistryPostProcessor beanPostProcessor(ConfigurableEnvironment environment) {
        return new PostProcessor(environment);
    }

    class PostProcessor implements BeanDefinitionRegistryPostProcessor {

        private String propertyValue;

        /*
         * Reads property value from the configuration, then stores it
         */
        public PostProcessor(ConfigurableEnvironment environment) {
            propertyValue = readProperty(environment);
        }

        @Override
        public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {}

        /*
         * Creates the bean definition dynamically (using the configuration value), then registers it
         */
        @Override
        public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {
            BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(SampleDynamicBean.class);
            builder.addPropertyValue("property", propertyValue);
            registry.registerBeanDefinition("sampleDynamicBean", builder.getBeanDefinition());
        }

        /*
         * Iterates over all configuration sources, looking for the property value.
         * As Spring orders the property sources by relevance, the value of the first 
         * encountered property with the correct name is read and returned.
         */
        private String readProperty(ConfigurableEnvironment environment) {
            for (PropertySource<?> source : environment.getPropertySources()) {
                if (source instanceof EnumerablePropertySource) {
                    EnumerablePropertySource<?> propertySource = (EnumerablePropertySource<?>) source;
                    for (String property : propertySource.getPropertyNames()) {
                        if (PROPERTY_KEY.equals(property))
                        {
                            return (String)propertySource.getProperty(PROPERTY_KEY);
                        }
                    }
                }
            }
            throw new IllegalStateException("Unable to determine value of property " + PROPERTY_KEY);
        }

    }

    class SampleDynamicBean {
        private String property;

        public void setProperty(String property)
        {
            this.property = property;
        }

        public String getMessage()
        {
            return "This message is produced by a dynamic bean, it includes " + property;
        }

    }

}

示例代码改编自该博客文章https://scanningpages.wordpress.com/2017/07/28/spring-dynamic-beans/