Spring Boot环境变量使用技巧

在Spring Boot中，环境变量是一种非常有用的方式来配置应用程序的不同环境。以下是一些使用Spring Boot环境变量的技巧：

1. 使用application.properties或application.yml

Spring Boot支持通过application.properties或application.yml文件来配置环境变量。你可以在这些文件中定义环境变量，并在代码中使用它们。

application.properties示例：

app.name=MyApp
app.version=1.0.0
app.database.url=${DB_URL}
app.database.username=${DB_USERNAME}
app.database.password=${DB_PASSWORD}

application.yml示例：

app:
  name: MyApp
  version: 1.0.0
  database:
    url: ${DB_URL}
    username: ${DB_USERNAME}
    password: ${DB_PASSWORD}

2. 使用命令行参数

你可以在启动应用程序时通过命令行参数传递环境变量。

示例：

java -jar myapp.jar --spring.datasource.url=jdbc:mysql://localhost:3306/mydb --spring.datasource.username=user --spring.datasource.password=password

3. 使用操作系统环境变量

你可以通过操作系统的环境变量来配置Spring Boot应用程序。这些环境变量可以在application.properties或application.yml文件中使用。

application.properties示例：

app.name=${APP_NAME}
app.version=${APP_VERSION}
app.database.url=${DB_URL}
app.database.username=${DB_USERNAME}
app.database.password=${DB_PASSWORD}

application.yml示例：

app:
  name: ${APP_NAME}
  version: ${APP_VERSION}
  database:
    url: ${DB_URL}
    username: ${DB_USERNAME}
    password: ${DB_PASSWORD}

4. 使用Spring Boot的@Value注解

你可以使用Spring Boot的@Value注解来注入环境变量到Java类中。

示例：

import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

@Component
public class AppConfig {

    @Value("${app.name}")
    private String appName;

    @Value("${app.version}")
    private String appVersion;

    @Value("${app.database.url}")
    private String dbUrl;

    @Value("${app.database.username}")
    private String dbUsername;

    @Value("${app.database.password}")
    private String dbPassword;

    // Getters and setters
}

5. 使用Spring Boot的EnvironmentPostProcessor

你可以创建一个自定义的EnvironmentPostProcessor来在应用程序启动之前修改环境变量。

示例：

import org.springframework.boot.SpringApplication;
import org.springframework.boot.env.EnvironmentPostProcessor;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.PropertiesPropertySource;
import org.springframework.stereotype.Component;

import java.util.Properties;

@Component
public class CustomEnvironmentPostProcessor implements EnvironmentPostProcessor {

    @Override
    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
        Properties customProperties = new Properties();
        customProperties.put("app.name", "MyCustomApp");
        customProperties.put("app.version", "1.0.0");
        environment.getPropertySources().addLast(new PropertiesPropertySource("customProperties", customProperties));
    }
}

6. 使用Spring Boot的@Profile注解

你可以使用@Profile注解来根据不同的环境激活不同的配置。

示例：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;

@Configuration
public class DatabaseConfig {

    @Bean
    @Profile("dev")
    public DataSource devDataSource() {
        // Dev database configuration
    }

    @Bean
    @Profile("prod")
    public DataSource prodDataSource() {
        // Prod database configuration
    }
}

总结

通过以上技巧，你可以在Spring Boot应用程序中灵活地使用环境变量来配置不同的环境和需求。这些方法可以帮助你更好地管理和部署应用程序。