Spring動態(tài)替換Properties配置變量

有如下需求，在Spring啟動的時候，我們需要修改某個指定的配置的值，從而達到動態(tài)加載變量的效果：

    @Value("${destination}")
    public String destination;

1.實現(xiàn)方法有如下：

• 在對應(yīng)的Spring Bean實例化之后，通過反射動態(tài)的修改實例的參數(shù)值
• 動態(tài)的修改PropertySource

具體的類如下：

public interface HelloService {

    /**
     * say hello
     */
    void sayHello();
}


@Service
public class HelloServiceImpl implements HelloService{

    @Value("${destination}")
    private String destination;

    @Override
    public void sayHello() {
        System.out.println("Hello, let us go to " + destination);
    }
}

第一種方法實現(xiàn)

方式1：
定義一個應(yīng)用事件，監(jiān)聽?wèi)?yīng)用上下文刷新事件，然后我們可以通過這個事件拿到ApplicationContext上下文，進而去獲取對應(yīng)的SpringBean，然后進行反射賦值操作。

public class LocalApplicationListener implements ApplicationListener<ContextRefreshedEvent> {

    @Override
    public void onApplicationEvent(ContextRefreshedEvent event) {
        Object helloService = event.getApplicationContext().getBean(HelloService.class);
        if (helloService instanceof HelloServiceImpl) {
            try {
                Field field = HelloServiceImpl.class.getDeclaredField("destination");
                field.setAccessible(true);
                ReflectionUtils.setField(field, helloService, "廣東深圳");
            } catch (NoSuchFieldException e) {
                e.printStackTrace();
            }
        }
    }
}

有了這個監(jiān)聽事件，我們需要把這個監(jiān)聽器添加到Spring的監(jiān)聽器集合中，可以通過啟動類的啟動方法進行添加：

@SpringBootApplication
public class DemoApplication {

    public static void main(String[] args) {
        SpringApplication app = new SpringApplication(DemoApplication.class);
        app.addListeners(new LocalApplicationListener());
        app.run(args);
    }
}

方式2：
定義一個類實現(xiàn)BeanPostProcessor接口，在接口中進行反射操作。BeanPostProcessor是Spring提供的一個Bean擴展接口，可以通過該接口實現(xiàn)一些Bean創(chuàng)建之前和創(chuàng)建之后的操作。

@Component
public class LocalBeanProcessor implements BeanPostProcessor {

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

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if (bean instanceof HelloServiceImpl) {
            try {
                Field field = HelloServiceImpl.class.getDeclaredField("destination");
                field.setAccessible(true);
                ReflectionUtils.setField(field, bean, "廣東廣州");
            } catch (NoSuchFieldException e) {
                e.printStackTrace();
            }
        }
        return bean;
    }
}

第二種實現(xiàn)方法：
通過了解Spring的加載流程，我們得知Spring在上下文準(zhǔn)備完畢(配置信息解析完畢并創(chuàng)建好了應(yīng)用上下文)之后，會調(diào)用推送一個ApplicationEnvironmentPreparedEvent，從這個事件中，我們可以獲取到ConfigurableEnvironment對象，而ConfigurableEnvironment對象可以獲取到MutablePropertySources。對于MutablePropertySources，它包含了Spring的所有 的配置信息，包括我們啟動應(yīng)用的application.properties文件的配置信息。具體代碼如下：

public class LocalEnvironmentPrepareEventListener implements ApplicationListener<ApplicationEnvironmentPreparedEvent> {

    @Override
    public void onApplicationEvent(ApplicationEnvironmentPreparedEvent event) {
        MutablePropertySources propertySources = event.getEnvironment().getPropertySources();
        for (PropertySource<?> propertySource : propertySources) {
            boolean applicationConfig = propertySource.getName().contains("applicationConfig");
            if (!applicationConfig) {
               continue;
            }
            Object property = propertySource.getProperty("destination");
            Map<String, OriginTrackedValue> source = (Map<String, OriginTrackedValue>) propertySource.getSource();
            OriginTrackedValue originTrackedValue = source.get("destination");
            OriginTrackedValue newOriginTrackedValue = OriginTrackedValue.of("中國香港", originTrackedValue.getOrigin());
            source.put("destination", newOriginTrackedValue);
            System.out.println(property);
        }
    }
}

同樣的我們需要把該監(jiān)聽器添加到Spring中：

@SpringBootApplication
public class DemoApplication {

    public static void main(String[] args) {
        SpringApplication app = new SpringApplication(DemoApplication.class);
        app.addListeners(new LocalEnvironmentPrepareEventListener());
        app.run(args);
    }
}

知名的分布式配置管理工具Apollo就是通過反射來實現(xiàn)配置參數(shù)的動態(tài)修改的，Apollo實現(xiàn)了BeanPostProcessor接口，這樣它就可以把所有的Bean和@Value的key的關(guān)系保存起來，類似于Map<String, List<Bean>>，當(dāng)配置中心的配置被改動的時候，就發(fā)一個通知給對應(yīng)的服務(wù)，然后由服務(wù)自己去拉取配置參數(shù)，重新賦值。