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

Question

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

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

这是我的主类和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 接口(interface)，其中包含两个实现 Type1DataLoadProcess 和 Type2DataLoadProcess 并根据命令行参数？

Answer 1

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

创建接口(interface)DataLoadProcess

然后是类:

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

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

然后你可以像这样注入(inject)接口(interface)类型:

@Autowired 
DataLoadProcess dataLoadProcessor;

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

-Dspring.profiles.active=type1