时间戳

首页 > 解决方案 > 基于命令行参数注入实现

java - 基于命令行参数注入实现

问题描述

有没有办法根据 Spring Boot 中的命令行参数注入特定的接口实现？

我有一个数据加载应用程序，我需要根据命令行参数加载特定类型的数据。

这是我的主要课程和CommandLineRunner：

@SpringBootApplication
public class DataLoadersApplication implements CommandLineRunner {

    private Type1LoadProcess type1LoadProcess;
    private Type2LoadProcess type2LoadProcess;

    public DataLoadersApplication(Type1LoadProcess type1LoadProcess,
      Type2LoadProcess type2LoadProcess) {
        this.type1LoadProcess = type1LoadProcess;
        this.type2LoadProcess = type2LoadProcess;
    }

    public static void main(String[] args) {
        SpringApplication.run(DataLoadersApplication.class, args);
    }

    @Override
    public void run(String... args) {
        if (args[0].equalsIgnoreCase("load-type1")) {
            type1LoadProcess.process();
        } else if (args[0].equalsIgnoreCase("load-type2")) {
            type2LoadProcess.process();
        } 
    }
}

有没有一种方法可以创建一个DataLoadeProcess具有两个实现的接口Type1DataLoadProcess并Type2DataLoadProcess根据命令行参数在主类中注入实现？

标签： javaspring-bootcommand-lineinterfaceautowired

解决方案

您可以使用 Spring 配置文件来实现您的目标： https ://docs.spring.io/spring-boot/docs/current/reference/html/boot-features-profiles.html

创建接口DataLoadProcess

然后上课：

@Component
@Profile("type1")
public class Type1LoadProcess implements DataLoadProcess {
}

@Component
@Profile("type2")
public class Type2LoadProcess implements DataLoadProcess {
}

然后你可以注入接口类型，如：

@Autowired 
DataLoadProcess dataLoadProcessor;

现在您可以使用其中一个配置文件启动您的应用程序，例如使用系统属性集：

-Dspring.profiles.active=type1

推荐阅读