在使用spring boot 構建應用啟動時,我們在工作中都是通過命令列來啟動應用,有時候會需要一些特定的引數以在應用啟動時,做一些初始化的操作。
spring boot 提供了 CommandLineRunner
和 ApplicationRunner
這兩個介面供使用者使用。
1. CommandLineRunner
1.1 宣告:
@FunctionalInterface
public interface CommandLineRunner {
/**
* Callback used to run the bean.
* @param args incoming main method arguments
* @throws Exception on error
*/
void run(String... args) throws Exception;
}
複製程式碼
1.2 使用:
package com.example.consoleapplication;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;
@Component
public class TestRunner implements CommandLineRunner {
@Override
public void run(String... args) {
// Do something...
for(String arg: args){
System.out.println(arg);
}
System.out.print("test command runner");
}
}
複製程式碼
1.3 執行結果
執行: java -jar build/libs/consoleapplication-0.0.1-SNAPSHOT.jar -sdfsaf sdfas
,
結果如下:
2019-03-16 17:31:56.544 INFO 18679 --- [ main] c.e.consoleapplication.DemoApplication : No active profile set, falling back to default profiles: default
2019-03-16 17:31:57.195 INFO 18679 --- [ main] c.e.consoleapplication.DemoApplication : Started DemoApplication in 16.172 seconds (JVM running for 16.65)
-sdfsaf
sdfas
test command runner%
複製程式碼
2. ApplicationRunner
2.1 宣告
/**
* Interface used to indicate that a bean should <em>run</em> when it is contained within
* a {@link SpringApplication}. Multiple {@link ApplicationRunner} beans can be defined
* within the same application context and can be ordered using the {@link Ordered}
* interface or {@link Order @Order} annotation.
*
* @author Phillip Webb
* @since 1.3.0
* @see CommandLineRunner
*/
@FunctionalInterface
public interface ApplicationRunner {
/**
* Callback used to run the bean.
* @param args incoming application arguments
* @throws Exception on error
*/
void run(ApplicationArguments args) throws Exception;
}
複製程式碼
2.2 使用
ApplicationRunner
和 CommandLineRunner
的使用是有差別的:
CommandLineRunner
的使用,只是把引數根據空格分割。ApplicationRunner
會根據 是否匹配--key=value
來解析引數,- 能匹配,則為
optional
引數, 可用getOptionValues
獲取引數值。 - 不匹配則是
non optional
引數。
- 能匹配,則為
package com.example.consoleapplication;
import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component;
import org.springframework.boot.ApplicationArguments;
@Component
public class TestApplicationRunner implements ApplicationRunner {
@Override
public void run(ApplicationArguments args) throws Exception {
// Do something...
System.out.println("option arg names" + args.getOptionNames());
System.out.println("non option+" + args.getNonOptionArgs());
}
}
複製程式碼
2.3 執行結果
執行命令 java -jar build/libs/consoleapplication-0.0.1-SNAPSHOT.jar -non1 non2 --option=1
, 結果為:
2019-03-16 18:08:08.528 INFO 19778 --- [ main] c.e.consoleapplication.DemoApplication : No active profile set, falling back to default profiles: default
2019-03-16 18:08:09.166 INFO 19778 --- [ main] c.e.consoleapplication.DemoApplication : Started DemoApplication in 16.059 seconds (JVM running for 16.56)
test
option arg names[option]
non option+[-non1, non2]-non1
non2
--option=1
test%
複製程式碼
可以看到, optional
引數名有 option
, non optional
引數有 -non1
和 non2
3. 小結
CommandLineRunner
和 ApplicationRunner
都能實現命令列應用啟動時根據引數獲取我們需要的值,做特殊的邏輯。但兩者有所不同,推薦使用 ApplicationRunner
的 optional
引數, 方便擴充套件。