使用Spring的 Environment 和 ConfigurableEnvironment 来在springboot应用启动过程中对属性进行修改。

1.修改yml文件的问题

我在上一篇文章中介绍了使用《java启动springboot项目前根据环境变量动态改编yaml文件的变量值》但是在FC启动后发现存在问题

• springboot初始化时需要初始化数据库，这时就会存在数据库连接在yml文件修改之前，这就会导致链接数据库失败。
  解决方案想的是在springboot项目启动前，对变量进行修改

2.结合spring的Environment和ConfigurableEnvironment对环境变量进行修改

如果你希望在Spring Boot应用启动前动态更改数据库配置，可以使用Spring的 Environment 和 ConfigurableEnvironment 来在应用启动过程中对属性进行修改。具体步骤如下：

2.1 创建一个类实现 ApplicationContextInitializer

ApplicationContextInitializer 是一个接口，可以在Spring上下文刷新之前对其进行进一步的配置。我们可以在这里进行数据库配置的动态修改。

import org.springframework.context.ApplicationContextInitializer;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.MapPropertySource;

import java.util.HashMap;
import java.util.Map;

public class CustomDatabaseInitializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {
    
    @Override
    public void initialize(ConfigurableApplicationContext applicationContext) {
        ConfigurableEnvironment env = applicationContext.getEnvironment();

        // 获取新的数据库配置
        String newUrl = "jdbc:mysql://newhost:3306/newdb";
        String newUsername = "newuser";
        String newPassword = "newpass";
        String newDriverClassName = "com.mysql.cj.jdbc.Driver";

        // 用新的配置更新环境变量
        Map<String, Object> map = new HashMap<>();
        map.put("spring.datasource.url", newUrl);
        map.put("spring.datasource.username", newUsername);
        map.put("spring.datasource.password", newPassword);
        map.put("spring.datasource.driver-class-name", newDriverClassName);

        env.getPropertySources().addFirst(new MapPropertySource("customDatabaseConfig", map));
    }
}

2.2 配置Spring Boot应用使用 ApplicationContextInitializer

在Spring Boot的启动类中配置该 ApplicationContextInitializer。

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class Application {

    public static void main(String[] args) {
        SpringApplication app = new SpringApplication(Application.class);
        app.addInitializers(new CustomDatabaseInitializer());
        app.run(args);
    }
}

2.3 配置文件 application.yml

即使你已经在代码中动态设置了数据库配置，也建议在 application.yml 文件中保留默认的数据库配置，以便项目在初始开发和测试过程中使用。

spring:
  datasource:
    url: jdbc:mysql://localhost:3306/initialdb
    username: initialuser
    password: initialpass
    driver-class-name: com.mysql.cj.jdbc.Driver

2.4 验证

启动应用后，Spring Boot会首先执行 CustomDatabaseInitializer 类中的 initialize 方法，动态更新 Environment 中的数据库配置。

3.方案优点

• 灵活性：可以在代码中灵活调整数据库配置。
• 易于管理：可以通过版本控制管理数据库配置的变化。
• 高效：在应用启动之前完成配置修改，不需要在应用启动后通过HTTP请求等方式动态刷新配置。
  通过上述方法，可以在Spring Boot应用启动前动态更改数据库配置。这时我们可以把我们需要由用户自己填写的变量通通放到CustomDatabaseInitializer进行初始化，由Spring Boot的启动类加载，而用户只需在FC中配置环境变量，系统就会自动注入到springboot项目中。ok这个逻辑完善了。