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> {@Overridepublic 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/initialdbusername: initialuserpassword: initialpassdriver-class-name: com.mysql.cj.jdbc.Driver
2.4 验证
启动应用后,Spring Boot会首先执行 CustomDatabaseInitializer
类中的 initialize
方法,动态更新 Environment
中的数据库配置。
3.方案优点
- 灵活性:可以在代码中灵活调整数据库配置。
- 易于管理:可以通过版本控制管理数据库配置的变化。
- 高效:在应用启动之前完成配置修改,不需要在应用启动后通过HTTP请求等方式动态刷新配置。
通过上述方法,可以在Spring Boot应用启动前动态更改数据库配置。这时我们可以把我们需要由用户自己填写的变量通通放到CustomDatabaseInitializer进行初始化,由Spring Boot的启动类加载,而用户只需在FC中配置环境变量,系统就会自动注入到springboot项目中。ok这个逻辑完善了。