当前位置：首页 > news >正文

CommandLineRunner 和 ApplicationRunner

news 来源：原创 2024/10/2 3:36:03

CommandLineRunner 和 ApplicationRunner

背景：

项目启动之前，预先加载数据。比如，权限容器、特殊用户数据等。通常我们可以使用监听器、事件来操作。但是，springboot提供了一个简单的方式来实现此类需求，即，CommandLineRunner 和 ApplicationRunner。

话不多说，先看源码：

CommandLineRunner 和 ApplicationRunner 都是两个接口，继承了 Runner 接口。

CommandLineRunner

帮大家翻译一下：接口的实现必须包含在bean中，多个实现了CommandLineRunner的bean可以用@Order注解标识执行顺序。这里还说了，如果需要访问原始 ApplicationArguments 而不是字符串数组，可以使用ApplicationRunner。因为CommandLineRunner中run()方法的参数是字符串数组。

package org.springframework.boot;import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;/*** Interface used to indicate that a bean should <em>run</em> when it is contained within* a {@link SpringApplication}. Multiple {@link CommandLineRunner} beans can be defined* within the same application context and can be ordered using the {@link Ordered}* interface or {@link Order @Order} annotation.* <p>* If you need access to {@link ApplicationArguments} instead of the raw String array* consider using {@link ApplicationRunner}.** @author Dave Syer* @since 1.0.0* @see ApplicationRunner*/
@FunctionalInterface
public interface CommandLineRunner extends Runner {/*** Callback used to run the bean.* @param args incoming main method arguments* @throws Exception on error*/void run(String... args) throws Exception;}

ApplicationRunner

这里的注释与上面CommandLineRunner的注释差不多，就不介绍了。

package org.springframework.boot;import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;/*** 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 extends Runner {/*** Callback used to run the bean.* @param args incoming application arguments* @throws Exception on error*/void run(ApplicationArguments args) throws Exception;}

两者的使用：

CommandLineRunner

/*** @Author: wangrongyi* @Date: 2024/9/30 10:46* @Description:*/
@Component
@Slf4j
public class CommandInitConfig implements CommandLineRunner {@Overridepublic void run(String... args) throws Exception {log.info("CommandInitConfig执行了");for (String arg : args) {log.info("arg = {}", arg);}}
}

项目启动之后，接着就会执行所有实现了CommandLineRunner接口的run方法，如果有多个，执行顺序跟bean的注入顺序相同。

如果有携带的参数，直接跟在启动命令后面

java -jar xxx.jar args1 args2 args3

ApplicationRunner

/*** @Author: wangrongyi* @Date: 2024/9/30 11:13* @Description:*/
@Component
public class ApplicationInitConfig implements ApplicationRunner {@Overridepublic void run(ApplicationArguments args) throws Exception {System.out.println("应用启动完成，正在执行初始化任务...");// 获取非选项参数String[] nonOptionArgs = args.getNonOptionArgs().toArray(new String[0]);for (String arg : nonOptionArgs) {System.out.println("非选项参数: " + arg);}// 获取选项参数String[] optionNames = args.getOptionNames().toArray(new String[0]);for (String optionName : optionNames) {System.out.println("选项名称: " + optionName);if (args.containsOption(optionName)) {System.out.println("包含选项: " + optionName);String[] optionValues = args.getOptionValues(optionName).toArray(new String[0]);for (String value : optionValues) {System.out.println("选项值: " + value);}}}}
}

其中 ApplicationArguments 是 run()函数的参数，简单介绍一下这个接口：

ApplicationArguments 是 Spring Boot 提供的一个接口，用于处理命令行参数。它提供了一种更加结构化的方式来处理命令行参数，包括非选项参数（non-option arguments）和选项参数（option arguments）。该接口提供了几个获取参数的方法：

getNonOptionArgs()：获取非选项参数列表。

获取所有非选项参数，即不带 - 或 – 的参数
getOptionNames()：获取所有选项名称。

获取所有选项名称，即带 - 或 – 的参数。
containsOption(String name)：判断是否包含指定选项。

判断是否包含指定选项。
getOptionValues(String name)：获取指定选项的所有值。

获取指定选项的所有值。

ApplicationRunner 适合参数复杂的情况，打包后通过命令启动：

java -jar xxx.jar --option1=value1 --option2=value2 arg1 arg2

总结：

两者都可以在程序启动后执行操作，且都支持获取启动参数并进行处理。

使用 ApplicationArguments：通过 ApplicationRunner 和 ApplicationArguments 来处理参数。
使用 CommandLineRunner：通过 run(String… args) 方法来处理参数。

那么实现了ApplicationRunner 和 CommandLineRunner接口的run方法是如何调用的呢？

下面我们跟随源码解读：

进入主程序的run方法。

找到callRunners()方法
点进去之后会发现首选获取所有实现了Runner接口的bean，Runner接口只有两个子接口，所有也就是找到了所有实现了ApplicationRunner 和 CommandLineRunner接口的bean，然后遍历每个bean，调用callRunner方法执行。
callRunner方法中区分了bean的类型，执行不同的代码。