如何让ContextCustomizer访问Spring环境属性？求替代方案

一、读取application.yaml配置 #

在MyContextCustomizer的customizeContext方法中，可直接通过ConfigurableApplicationContext获取已加载的环境配置：

static class MyContextCustomizer implements ContextCustomizer {

    @Override
    public void customizeContext(ConfigurableApplicationContext context, 
                                 MergedContextConfiguration mergedConfig) {
        // 从Spring环境中读取application.yaml里的配置项
        Environment env = context.getEnvironment();
        String dbUsername = env.getProperty("spring.datasource.username");
        String dbPassword = env.getProperty("spring.datasource.password");
        // 假设配置文件里存的是URL模板，比如jdbc:postgresql://%s:%d/testdb
        String dbUrlTemplate = env.getProperty("spring.datasource.url-template");

        // 执行启动阶段操作（比如启动Postgres TestContainer）
        // 示例：模拟容器启动后得到的地址和端口
        String containerHost = "localhost";
        int containerPort = 5432;

        // 生成实际的数据源URL
        String actualDbUrl = String.format(dbUrlTemplate, containerHost, containerPort);

        // 将动态配置注入Spring环境
        injectDynamicDbProperties(context, actualDbUrl, dbUsername, dbPassword);
    }

    private void injectDynamicDbProperties(ConfigurableApplicationContext context, 
                                          String dbUrl, String username, String password) {
        // 获取可修改的属性源集合
        MutablePropertySources propertySources = ((ConfigurableEnvironment) context.getEnvironment()).getPropertySources();
        
        // 构造动态属性Map
        Map<String, Object> dynamicDbProps = new HashMap<>();
        dynamicDbProps.put("spring.datasource.url", dbUrl);
        dynamicDbProps.put("spring.datasource.username", username);
        dynamicDbProps.put("spring.datasource.password", password);
        
        // 将自定义属性源添加到最前面，确保优先级高于application.yaml
        propertySources.addFirst(new MapPropertySource("postgres-test-container-props", dynamicDbProps));
    }
}

二、替代DynamicPropertySource的方案 #

核心是直接操作Spring的Environment属性源：

• 通过ConfigurableEnvironment获取MutablePropertySources，它允许我们修改环境中的属性
• 创建MapPropertySource存储动态生成的数据源配置
• 调用addFirst方法让自定义属性源优先级最高，覆盖原配置文件中的对应项

三、完善ContextCustomizerFactory逻辑 #

在工厂类中，仅当测试类标注了自定义注解时才创建自定义ContextCustomizer：

@Override
public ContextCustomizer createContextCustomizer(Class<?> testClass,
                                                 List<ContextConfigurationAttributes> configAttributes) {
    // 判断测试类是否启用了Postgres TestContainer注解
    if (testClass.isAnnotationPresent(EnabledPostgresTestContainer.class)) {
        return new MyContextCustomizer();
    }
    return null; // 未标注则不应用自定义逻辑
}

关键注意点 #

• customizeContext阶段，Spring已经完成了环境初始化，包括加载application.yaml等配置文件，因此可以直接从Environment读取属性
• 用addFirst保证动态配置优先级最高，避免被原配置覆盖

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