乐趣区

Spring Boot 2 – 使用CommandLineRunner与ApplicationRunner

本篇文章我们将探讨 CommandLineRunner 和 ApplicationRunner 的使用。
在阅读本篇文章之前,你可以新建一个工程,写一些关于本篇内容代码,这样会加深你对本文内容的理解,关于如何快速创建新工程,可以参考我的这篇博客:
Spring Boot 2 – 创建新工程
概述
CommandLineRunner 和 ApplicationRunner 是 Spring Boot 所提供的接口,他们都有一个 run() 方法。所有实现他们的 Bean 都会在 Spring Boot 服务启动之后自动地被调用。
由于这个特性,它们是一个理想地方去做一些初始化的工作,或者写一些测试代码。
CommandLineRunner
使用 Application 实现
在我们新建好工程后,为了简单我们直接使用 Application 类实现 CommandLineRunner 接口,这个类的注解 @SpringBootApplication 会为我们自动配置。
package cn.examplecode.sb2runner;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class Sb2runnerApplication implements CommandLineRunner {

private static Logger logger = LoggerFactory.getLogger(Sb2runnerApplication.class);

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

@Override
public void run(String… args) throws Exception {
logger.info(“ 服务已启动,执行 command line runner。”);

for (int i = 0; i < args.length; ++i) {
logger.info(“args[{}]: {}”, i, args[i]);
}
}
}
接下来我们直接启动服务,查看日志如下,发现 run() 方法被正常地执行了:
Tomcat started on port(s): 8080 (http) with context path ”
Started Sb2runnerApplication in 2.204 seconds (JVM running for 3.161)
服务已启动,执行 command line runner。
参数传递
run() 方法有个可变参数 args,这个参数是用来接收命令行参数的,我们下面来加入参数来测试一下:
然后重启服务,观察日志,可以看到参数被正常地接收到了:
Tomcat started on port(s): 8080 (http) with context path ”
Started Sb2runnerApplication in 1.888 seconds (JVM running for 2.41)
服务已启动,执行 command line runner。
args[0]: –param=sth
命令行参数传递
之前我们说过使用 Spring Boot 的一大优势就是可以将工程直接打包成一个 jar 包而不需要单独部署。打包成 jar 包后可以直接执行该 jar 包进行服务的启动,这样在执行 jar 包时我们就可以传入命令行参数,让 CommandLineRunner 接收参数。
这种场景在服务器上特别常用。比如我们想执行某个操作,又不想对外部暴露,此时可以使用 CommandLineRunner 作为该操作的入口。
下面我们就打成 jar 包来演示一下。
进入终端界面,开始打包

打包完成后,执行该 jar 包,记得先把 IDE 的服务停掉。

可以从日志中看到我们也正常地获取到了参数。通过传递参数,在业务逻辑上我们可以根据不同的参数而执行不同的操作。
上面我们提到的只是一个 CommandLineRunner,如果我们有多个 CommandLineRunner 怎么办呢?怎么控制它们执行的顺序呢?
下面我们就来介绍如何指定执行的顺序。
指定执行顺序
Spring Boot 为我们提供了一个注解 ”@Order”,可以用来指定执行的顺序,比如我们工程里面有三个 CommandLineRunner:
@Component
@Order(1)
public class CommandRunner1 implements CommandLineRunner {

private static Logger logger = LoggerFactory.getLogger(CommandRunner1.class);

@Override
public void run(String… args) throws Exception {
logger.info(“ 执行第一个 command line runner…”);
}

}

@Component
@Order(2)
public class CommandRunner2 implements CommandLineRunner {

private static Logger logger = LoggerFactory.getLogger(CommandRunner2.class);

@Override
public void run(String… args) throws Exception {
logger.info(“ 执行第二个 command line runner…”);
}

}

@Component
@Order(3)
public class CommandRunner3 implements CommandLineRunner {

private static Logger logger = LoggerFactory.getLogger(CommandRunner3.class);

@Override
public void run(String… args) throws Exception {
logger.info(“ 执行第三个 command line runner…”);
}

}

我们可以在该类的上面直接加入 @Order 注解,然后 Spring Boot 就会按照我们注解指定的顺序从小到大的执行了。很简单,是不是?
Tomcat started on port(s): 8080 (http) with context path ”
Started Sb2runnerApplication in 1.764 seconds (JVM running for 2.292)
执行第一个 command line runner…
执行第二个 command line runner…
执行第三个 command line runner…
ApplicationRunner
ApplicationRunner 与 CommandLineRunner 做的事情是一样的,也是在服务启动之后其 run() 方法会被自动地调用,唯一不同的是 ApplicationRunner 会封装命令行参数,可以很方便地获取到命令行参数和参数值。
@Component
public class ApplicationRunner1 implements ApplicationRunner {

private static Logger logger = LoggerFactory.getLogger(ApplicationRunner1.class);

@Override
public void run(ApplicationArguments args) throws Exception {
logger.info(“ 执行 application runner…”);
logger.info(“ 获取到参数: ” + args.getOptionValues(“param”));
}
}

执行结果:
我们可以发现,通过 run() 方法的参数 ApplicationArguments 可以很方便地获取到命令行参数的值。
所以如果你的工程需要获取命令行参数的话,建议你使用 ApplicationRunner。
总结
无论是 CommandLineRunner 还是 ApplicationRunner,它们的目的都是在服务启动之后执行一些操作。如果需要获取命令行参数时则建议使用 ApplicationRunner。
另一种场景是我们在服务器上需要执行某个操作,比如修正数据库用户的数据,而又找不到合适的执行入口,那么这就是它们理想的使用场景了。
我的博客中其他关于 Spring Boot 的所有文章可以点击这里找到,欢迎关注!
如果有问题可以留言,或者给我发邮件 lloyd@examplecode.cn,期待我们共同学习与成长!

退出移动版