使用 Spring Boot 的 CommandLineRunner 初始化数据

原创

mb64ba3e25405fa 2023-07-21 16:14:02 ©著作权

©著作权归作者所有：来自51CTO博客作者mb64ba3e25405fa的原创作品，请联系作者获取转载授权，否则将追究法律责任

Spring Boot 提供了 CommandLineRunner 接口，可以用于在应用程序启动时执行一些初始化操作，例如初始化数据库数据、加载配置文件、预热缓存等。通过实现 CommandLineRunner 接口，您可以很方便地在应用程序启动后执行所需的初始化任务。以下是如何使用 CommandLineRunner 初始化数据的示例：

1. 创建初始化类

首先，创建一个实现 CommandLineRunner 接口的初始化类，并在其中实现初始化逻辑。

import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

@Component
public class MyDataInitializer implements CommandLineRunner {

    @Override
    public void run(String... args) throws Exception {
        // 在此处编写初始化数据的逻辑
        System.out.println("Initializing data...");
        // 例如：初始化数据库数据、加载配置文件、预热缓存等
    }
}

2. 初始化逻辑

在 run 方法中编写初始化逻辑。这里可以执行各种初始化任务，例如读取文件、调用其他服务、初始化数据库等。

@Override
public void run(String... args) throws Exception {
    // 初始化数据库数据
    userRepository.save(new User("John Doe", "john@example.com", "password"));
    userRepository.save(new User("Jane Smith", "jane@example.com", "password"));

    // 加载配置文件
    String apiBaseUrl = configService.getApiBaseUrl();
    System.out.println("API Base URL: " + apiBaseUrl);

    // 预热缓存
    cacheService.preheatCache();
}