在一個初春的下午，甲跟我說，要在Spring Boot啟動服務(wù)的時候，設(shè)置表自增的起始值。

于是我用屁股想了一下，不就是在main方法里折騰嘛。

后來實際操作了一把，發(fā)現(xiàn)屁股被打了。

于是乎，找到官方文檔(以2.1.4為例)，找到這一段：

如果你需要在啟動SpringApplication后執(zhí)行一些具體的代碼，你可以實現(xiàn)ApplicaitonRunner或者CommandLineRunner接口。兩個接口都實現(xiàn)了一個工作方式相同的run方法，該方法僅會在SpringApplication.run(...)前執(zhí)行。

唯一不同的是實現(xiàn)CommandLineRunner接口的run方法參數(shù)為String類型，而實現(xiàn)ApplicaitonRunner的run方法的參數(shù)則是需要ApplicationArguments。官方文檔中有個例子供參考。

如果有多個ApplicaitonRunner或者CommandLineRunner接口的實現(xiàn)存在啟動順序，則可以使用org.springframework.core.Ordered接口或者org.springframework.core.annotation.Order注解的形式來給他們排序。

由于我沒有參數(shù)類型等的限制，所以用哪個接口都一樣，寫個跟官方不一樣的，于是代碼大概長這樣:

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;

@Component
public class InstructionStart implements ApplicationRunner {

  private Logger logger = LoggerFactory.getLogger(this.getClass());

  @Autowired
  private JdbcTemplate template;

  @Override
  public void run(ApplicationArguments applicationArguments) throws Exception {
    String increment = "0";
    logger.info("初始化遞增起始值為：{}", increment);
    template.execute("ALTER TABLE `table` AUTO_INCREMENT = " + increment);
  }
}

深刻的意識到腦子和屁股一樣重要。

寫完啟動項，那么再把退出也說一下：

每一個SpringApplication都應(yīng)該向JVM注冊一個鉤子函數(shù)來確保ApplicationContext能優(yōu)雅地關(guān)閉。使所有的標(biāo)準(zhǔn)Spring生命周期回調(diào)（例如DisposableBean接口和@PreDestroy注解）都可用。

此外，如果你希望beans在調(diào)用SpringApplication.exit()時返回特定的退出代碼，則可以實現(xiàn)org.springframework.boot.ExitCodeGenerator接口，這些退出代碼會被傳給System.exit()作為返回的狀態(tài)碼。官方還給了個例子，就是下面這個。

@SpringBootApplication
public class ExitCodeApplication {

  @Bean
  public ExitCodeGenerator exitCodeGenerator() {
    return () -> 42;
  }

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

當(dāng)然，ExitCodeGenerator也可以由異常來實現(xiàn)，當(dāng)遇到一個這樣的異常時，Sprin Boot會返回實現(xiàn)了getExitCode()方法的退出代碼。

后面退出部分翻譯地磕磕碰碰的，有不對的地方歡迎指正。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

最新評論