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

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

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

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

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

第一步：創(chuàng)建實現(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、我們自己的初始化代碼寫在onApplicationEvent里；

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

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

第二步：在SpringBootApplication中注冊我們剛創(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啟動時執(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
//如果有多個這樣的類時，可以通過Order指定執(zhí)行順序，數(shù)值越小執(zhí)行優(yōu)先級越高
@Order(value = 0)
public class InitSystemConfig implements CommandLineRunner, EnvironmentAware {


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

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

  }

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

  }
}

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

您可能感興趣的文章: