如何用Spring Boot解決專案啟動時初始化資源？

穿格子衣上班發表於2019-04-28
Spring Boot

在我們實際工作中，總會遇到這樣需求，在專案啟動的時候需要做一些初始化的操作，比如初始化執行緒池，提前載入好加密證書等。今天就給大家介紹一個 Spring Boot 神器，專門幫助大家解決專案啟動初始化資源操作。

這個神器就是CommandLineRunner，CommandLineRunner介面的 Component會在所有 Spring Beans 都初始化之後，SpringApplication.run() 之前執行，非常適合在應用程式啟動之初進行一些資料初始化的工作。

接下來我們就運用案例測試它如何使用，在測試之前在啟動類加兩行列印提示，方便我們識別CommandLineRunner的執行時機。

@SpringBootApplication
public class CommandLineRunnerApplication {
	public static void main(String[] args) {
		System.out.println("The service to start.");
		SpringApplication.run(CommandLineRunnerApplication.class, args);
		System.out.println("The service has started.");
	}
}
複製程式碼

接下來我們直接建立一個類繼承CommandLineRunner，並實現它的run()方法。

@Component
public class Runner implements CommandLineRunner {
    @Override
    public void run(String... args) throws Exception {
        System.out.println("The Runner start to initialize ...");
    }
}
複製程式碼

我們在run()方法中列印了一些引數來看出它的執行時機。完成之後啟動專案進行測試：

...
The service to start.

  .   ____          _            __ _ _
 /\\ / ___'_ __ _ _(_)_ __  __ _ \ \ \ \
( ( )\___ | '_ | '_| | '_ \/ _` | \ \ \ \
 \\/  ___)| |_)| | | | | || (_| |  ) ) ) )
  '  |____| .__|_| |_|_| |_\__, | / / / /
 =========|_|==============|___/=/_/_/_/
 :: Spring Boot ::        (v2.0.0.RELEASE)
...
2018-04-21 22:21:34.706  INFO 27016 --- [           main] o.s.b.w.embedded.tomcat.TomcatWebServer  : Tomcat started on port(s): 8080 (http) with context path ''
2018-04-21 22:21:34.710  INFO 27016 --- [           main] com.neo.CommandLineRunnerApplication     : Started CommandLineRunnerApplication in 3.796 seconds (JVM running for 5.128)
The Runner start to initialize ...
The service has started.
複製程式碼

根據控制檯的列印資訊我們可以看出CommandLineRunner中的方法會在 Spring Boot 容器載入之後執行，執行完成後專案啟動完成。

如果我們在啟動容器的時候需要初始化很多資源，並且初始化資源相互之間有序，那如何保證不同的 CommandLineRunner的執行順序呢？Spring Boot 也給出瞭解決方案。那就是使用 @Orderv 註解。

我們建立兩個CommandLineRunner的實現類來進行測試：

第一個實現類：

@Component
@Order(1)
public class OrderRunner1 implements CommandLineRunner {
    @Override
    public void run(String... args) throws Exception {
        System.out.println("The OrderRunner1 start to initialize ...");
    }
}
複製程式碼

第二個實現類：

@Component
@Order(2)
public class OrderRunner2 implements CommandLineRunner {
    @Override
    public void run(String... args) throws Exception {
        System.out.println("The OrderRunner2 start to initialize ...");
    }
}
複製程式碼

新增完成之後重新啟動，觀察執行順序：

...
The service to start.

  .   ____          _            __ _ _
 /\\ / ___'_ __ _ _(_)_ __  __ _ \ \ \ \
( ( )\___ | '_ | '_| | '_ \/ _` | \ \ \ \
 \\/  ___)| |_)| | | | | || (_| |  ) ) ) )
  '  |____| .__|_| |_|_| |_\__, | / / / /
 =========|_|==============|___/=/_/_/_/
 :: Spring Boot ::        (v2.0.0.RELEASE)
...
2018-04-21 22:21:34.706  INFO 27016 --- [           main] o.s.b.w.embedded.tomcat.TomcatWebServer  : Tomcat started on port(s): 8080 (http) with context path ''
2018-04-21 22:21:34.710  INFO 27016 --- [           main] com.neo.CommandLineRunnerApplication     : Started CommandLineRunnerApplication in 3.796 seconds (JVM running for 5.128)
The OrderRunner1 start to initialize ...
The OrderRunner2 start to initialize ...
The Runner start to initialize ...
The service has started.
複製程式碼

通過控制檯的輸出我們發現，新增@Order註解的實現類最先執行，並且@Order() 裡面的值越小啟動越早。

在實踐中，使用ApplicationRunner也可以達到相同的目的，兩著差別不大。看來使用 Spring Boot 解決初始化資源的問題非常簡單。

相關文章