如何在SpringBoot中基于命令行参数创建Bean并在ApplicationRunner中使用

1. 创建命令行参数封装类 #

先定义一个类来封装命令行参数，可根据需求选择存储原始参数数组，或解析后的键值对：

// 命令行参数封装类
public class CommandLineParams {
    // 存储原始命令行参数数组
    private final String[] sourceArgs;

    // 构造方法初始化参数
    public CommandLineParams(String[] sourceArgs) {
        this.sourceArgs = sourceArgs;
    }

    // Getter方法
    public String[] getSourceArgs() {
        return sourceArgs;
    }

    // 可选：添加选项参数解析逻辑（比如解析--file=xxx这类格式）
    public String getOptionValue(String optionName) {
        // 可在这里实现自定义解析，或在Bean初始化时直接解析好存入字段
        return null;
    }
}

2. 将参数封装类注册为Spring Bean #

在你的主类MySbApp中添加@Bean方法，借助ApplicationArguments获取命令行参数并生成Bean：

@SpringBootApplication
public class MySbApp implements ApplicationRunner {
    // 修正main方法启动类：必须是当前主类MySbApp.class，原代码的Myclass.class会导致启动失败
    public static void main(String[] args) {
        SpringApplication.run(MySbApp.class, args);
    }

    @Autowired
    private Myclass myclass;

    @Autowired
    private CommandLineParams commandLineParams; // 注入参数Bean

    // 注册命令行参数Bean
    @Bean
    public CommandLineParams commandLineParams(ApplicationArguments args) {
        String[] sourceArgs = args.getSourceArgs();
        return new CommandLineParams(sourceArgs);
    }

    @Override
    public void run(ApplicationArguments args) throws Exception {
        // 使用注入的参数Bean
        String[] arguments = commandLineParams.getSourceArgs();
        for (String arg : arguments) {
            System.out.println("HEYYYYYY" + arg);
        }
        // 将参数Bean传给Myclass使用
        myclass.someMethod(commandLineParams);
    }
}

3. 在Myclass中使用参数Bean #

修改Myclass，通过注入或方法传参的方式使用参数Bean：

@Component // 必须添加该注解，让Spring将Myclass扫描为Bean，才能被@Autowired注入
public class Myclass {
    // 方式1：直接注入参数Bean
    @Autowired
    private CommandLineParams commandLineParams;

    public void someMethod() {
        String[] args = commandLineParams.getSourceArgs();
        // 业务逻辑处理...
    }

    // 方式2：通过方法参数传入参数Bean
    public void someMethod(CommandLineParams params) {
        String[] args = params.getSourceArgs();
        // 业务逻辑处理...
    }
}

关键注意事项 #

• 如果你的命令行参数是键值对格式（如--input=file.txt --output=result.txt），可以在注册Bean时直接解析：

@Bean
public CommandLineParams commandLineParams(ApplicationArguments args) {
    String inputPath = args.getOptionValues("input") != null ? args.getOptionValues("input").get(0) : null;
    String outputPath = args.getOptionValues("output") != null ? args.getOptionValues("output").get(0) : null;
    // 对应的CommandLineParams需修改构造方法和字段，存储inputPath、outputPath
    return new CommandLineParams(inputPath, outputPath);
}

• Myclass必须被Spring管理（添加@Component/@Service等注解），否则无法被@Autowired注入。

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