如何基于环境变量有条件替换Spring Boot的YML配置属性？

Spring Boot支持两种方式实现你的需求，分别是配置文件内的SpEL表达式，以及编程式的环境处理器，具体如下：

直接在application.yml中使用SpEL表达式 #

Spring Boot的配置文件支持Spring表达式语言（SpEL），可以通过#{}包裹表达式实现条件判断。你原来的写法存在语法问题，正确的实现需要结合systemEnvironment读取环境变量，示例如下：

spring:
  datasource:
    username: "#{systemEnvironment['ENV'] == 'PROD' ? systemEnvironment['USER_PROD'] : systemEnvironment['USER_TEST']}"
    password: "#{systemEnvironment['ENV'] == 'PROD' ? systemEnvironment['PWD_PROD'] : systemEnvironment['PWD_TEST']}"

注意事项：

• 表达式必须用双引号包裹，避免YAML解析时出现语法错误
• 确保所有用到的环境变量（ENV、USER_PROD、USER_TEST等）已正确设置，否则会抛出属性未找到异常
• 可以添加默认值避免异常，比如：systemEnvironment.get('USER_TEST', 'default_test_user')

编程方式实现（EnvironmentPostProcessor） #

如果需要更复杂的逻辑，或者不想在配置文件中写入复杂表达式，可以通过实现EnvironmentPostProcessor接口，在应用启动前动态修改配置：

1. 创建自定义环境处理器类：

import org.springframework.boot.SpringApplication;
import org.springframework.boot.env.EnvironmentPostProcessor;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.MapPropertySource;
import java.util.HashMap;
import java.util.Map;

public class DatasourceConfigPostProcessor implements EnvironmentPostProcessor {
    @Override
    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
        // 默认使用TEST环境
        String env = environment.getProperty("ENV", "TEST");
        Map<String, Object> datasourceProps = new HashMap<>();
        
        if ("PROD".equals(env)) {
            datasourceProps.put("spring.datasource.username", environment.getProperty("USER_PROD"));
            datasourceProps.put("spring.datasource.password", environment.getProperty("PWD_PROD"));
        } else {
            datasourceProps.put("spring.datasource.username", environment.getProperty("USER_TEST"));
            datasourceProps.put("spring.datasource.password", environment.getProperty("PWD_TEST"));
        }
        
        // 将自定义属性添加到环境中，优先级高于配置文件
        environment.getPropertySources().addFirst(new MapPropertySource("customDatasourceProps", datasourceProps));
    }
}

2. 在src/main/resources/META-INF/spring.factories中注册处理器：

org.springframework.boot.env.EnvironmentPostProcessor=com.yourpackage.DatasourceConfigPostProcessor

这种方式逻辑更清晰，便于扩展复杂的条件判断，且不需要修改原有application.yml的结构。

内容的提问来源于stack exchange，提问作者Mohamed MOUFAHIM