详解SpringBoot程序启动时执行初始化代码

因项目集成了Redis缓存部分数据，需要在程序启动时将数据加载到Redis中，即初始化数据到Redis。

在SpringBoot项目下，即在容器初始化完毕后执行我们自己的初始化代码。

第一步：创建实现ApplicationListener接口的类

package com.stone;

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


public class ApplicationStartup implements ApplicationListener {

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

注意：

1、我们自己的初始化代码写在onApplicationEvent里；

2、ContextRefreshedEvent是Spring的ApplicationContextEvent一个实现，在容器初始化完成后调用；

3、以注解的方式注入我们需要的bean，会报空指针异常，因此需要以代码中的方式获取我们要的bean

第二步：在SpringBootApplication中注册我们刚创建的类

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启动时执行初始化代码

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指定执行顺序，数值越小执行优先级越高
@Order(value = 0)
public class InitSystemConfig implements CommandLineRunner, EnvironmentAware {


  
  @Override
  public void run(String... args) {

    //这里可以根据数据库返回结果创建一些对象、启动一些线程等
    

  }

  
  @Override
  public void setEnvironment(Environment environment) {

  }
}

 以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持考高分网。