快捷導(dǎo)航

詳解SpringBoot程序啟動(dòng)時(shí)執(zhí)行初始化代碼

更新時(shí)間：2018年09月05日 09:32:16 作者：stonesingsong

這篇文章主要介紹了詳解SpringBoot程序啟動(dòng)時(shí)執(zhí)行初始化代碼，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧

因項(xiàng)目集成了Redis緩存部分?jǐn)?shù)據(jù)，需要在程序啟動(dòng)時(shí)將數(shù)據(jù)加載到Redis中，即初始化數(shù)據(jù)到Redis。

在SpringBoot項(xiàng)目下，即在容器初始化完畢后執(zhí)行我們自己的初始化代碼。

第一步：創(chuàng)建實(shí)現(xiàn)ApplicationListener接口的類

package com.stone;

import com.stone.service.IPermissionService;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;

/**
 * @author Stone Yuan
 * @create 2017-12-02 21:54
 * @description
 */
public class ApplicationStartup implements ApplicationListener<ContextRefreshedEvent> {

  @Override
  public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {
    IPermissionService service = contextRefreshedEvent.getApplicationContext().getBean(IPermissionService.class);
    service.loadUserPermissionIntoRedis();
  }
}

注意：

1、我們自己的初始化代碼寫(xiě)在onApplicationEvent里；

2、ContextRefreshedEvent是Spring的ApplicationContextEvent一個(gè)實(shí)現(xiàn)，在容器初始化完成后調(diào)用；

3、以注解的方式注入我們需要的bean，會(huì)報(bào)空指針異常，因此需要以代碼中的方式獲取我們要的bean

第二步：在SpringBootApplication中注冊(cè)我們剛創(chuàng)建的類

package com.stone;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class YwythApplication {

  public static void main(String[] args) {
    SpringApplication springApplication = new SpringApplication(YwythApplication.class);
    springApplication.addListeners(new ApplicationStartup());
    springApplication.run(args);
  }
}

利用CommandLineRunner、EnvironmentAware在Spring boot啟動(dòng)時(shí)執(zhí)行初始化代碼

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.context.EnvironmentAware;
import org.springframework.core.annotation.Order;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;

import java.util.List;

@Component
//如果有多個(gè)這樣的類時(shí)，可以通過(guò)Order指定執(zhí)行順序，數(shù)值越小執(zhí)行優(yōu)先級(jí)越高
@Order(value = 0)
public class InitSystemConfig implements CommandLineRunner, EnvironmentAware {


  /*
   * 在服務(wù)啟動(dòng)后執(zhí)行，會(huì)在@Bean實(shí)例化之后執(zhí)行，故如果@Bean需要依賴這里的話會(huì)出問(wèn)題
   */
  @Override
  public void run(String... args) {

    //這里可以根據(jù)數(shù)據(jù)庫(kù)返回結(jié)果創(chuàng)建一些對(duì)象、啟動(dòng)一些線程等
    

  }

  /*
   * 在SystemConfigDao實(shí)例化之后、@Bean實(shí)例化之前執(zhí)行
   * 常用于讀取數(shù)據(jù)庫(kù)配置以供其它bean使用
   * environment對(duì)象可以獲取配置文件的配置，也可以把配置設(shè)置到該對(duì)象中
   */
  @Override
  public void setEnvironment(Environment environment) {

  }
}

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

您可能感興趣的文章: