作者:Rosalind33 | 来源:互联网 | 2022-12-07 15:23
有没有办法在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
根据命令行arg在main类中注入实现?
1> Simon Martin..:
您可以使用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