如何在Spring Boot的postProcessEnvironment中获取命令行参数加载外部API配置

方法一：直接通过SpringApplication对象获取参数数组 #

SpringApplication实例在调用postProcessEnvironment时已经持有了启动时传入的完整命令行参数，直接调用application.getArgs()就能拿到和main方法接收完全一致的参数数组，适配遗留API的调用需求。

修改后的代码如下：

public class LegacyConfigReader implements EnvironmentPostProcessor {
    private static final String PROPERTY_SOURCE_NAME = "legacyResource";

    @Override
    public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
        System.out.println("inside postProcessEnvironment");

        // 获取启动时传入的完整命令行参数
        String[] args = application.getArgs();

        Map<String, String> legacyConfigMap = LegacyConfigUtils.readConfig(args);

        Map<String, Object> propertyResource = legacyConfigMap.entrySet()
                                        .stream()
                                        .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

        environment.getPropertySources().addFirst(new MapPropertySource(PROPERTY_SOURCE_NAME, propertyResource));
    }
}

方法二：从Environment中提取命令行参数源 #

如果需要从Environment中读取具体的命令行参数键值对，可遍历环境中的PropertySources，找到CommandLinePropertySource类型的源进行提取：

@Override
public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
    System.out.println("inside postProcessEnvironment");

    // 查找命令行参数源
    CommandLinePropertySource<?> commandLineSource = null;
    for (PropertySource<?> source : environment.getPropertySources()) {
        if (source instanceof CommandLinePropertySource) {
            commandLineSource = (CommandLinePropertySource<?>) source;
            break;
        }
    }

    if (commandLineSource != null) {
        // 提取所有命令行参数的键值对
        Map<String, Object> commandLineProps = new HashMap<>();
        for (String propName : commandLineSource.getPropertyNames()) {
            commandLineProps.put(propName, commandLineSource.getProperty(propName));
        }
        // 按需使用参数，或转换为数组传给遗留API
    }

    // 后续调用遗留API的逻辑...
}

注意事项 #

• 方法一获取的参数数组完全匹配main方法输入，无需额外转换，最适配遗留API的调用逻辑。
• Spring Boot 1.5+版本均支持SpringApplication.getArgs()方法，推荐使用2.x及以上稳定版本。
• 若命令行参数包含Spring Boot特有的--分隔符，getArgs()返回的是--之后的应用参数，和独立Java应用的参数逻辑保持一致。

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