Spring Boot - 加载初始数据

Question

Spring Boot - 加载初始数据

springspring-bootspring-data

300

我想知道在应用程序启动之前加载初始数据库数据的最佳方法是什么？我正在寻找的是一种可以填充我的H2数据库中数据的方法。

例如，我有一个领域模型“用户”，可以通过访问 /users 来访问用户，但最初数据库中将没有任何用户，因此我必须创建他们。有没有办法自动填充数据库中的数据？

目前，我有一个由容器实例化的Bean，为我创建用户。

示例：

@Component
public class DataLoader {

    private UserRepository userRepository;

    @Autowired
    public DataLoader(UserRepository userRepository) {
        this.userRepository = userRepository;
        LoadUsers();
    }

    private void LoadUsers() {
        userRepository.save(new User("lala", "lala", "lala"));
    }
}

但我非常怀疑这是否是最好的做法。难道不是吗？

- Lithicas

9

可以这样做，或者简单地将 data.sql 和/或 schema.sql 添加到初始化数据中。所有这些都在参考指南中有记录，建议您阅读该指南。 - M. Deinum

如果这有帮助到您，请标记正确的答案。 - Malakai

22个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Grigory Kislin · Answer 1

对于动态数据，最紧凑的方法是将@mathias-dpunkt的解决方案放入MainApp中（使用Lombok @AllArgsConstructor）：

@SpringBootApplication
@AllArgsConstructor
public class RestaurantVotingApplication implements ApplicationRunner {
  private final VoteRepository voteRepository;
  private final UserRepository userRepository;

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

  @Override
  public void run(ApplicationArguments args) {
    voteRepository.save(new Vote(userRepository.getOne(1), LocalDate.now(), LocalTime.now()));
  }
}

- W Kenny · Answer 2

对于使用MysqlDriver的人，我尝试使用@bean注释的Init属性，它可以正常工作。

在路径resources\Scripts中创建模式和数据SQL文件后，

请在application.properties中添加以下行。

spring.jpa.hibernate.ddl-auto=none

编辑应用程序内容：

package com.spring_mvaen.demo;

import org.springframework.boot.CommandLineRunner;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.jdbc.datasource.init.DatabasePopulator;
import org.springframework.jdbc.datasource.init.DatabasePopulatorUtils;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;

@SpringBootApplication
public class DemoApplication implements CommandLineRunner {

  public static void main(String[] args) {
    SpringApplication.run(DemoApplication.class, args);
  }
  @Override
  public void run(String... arg0) throws Exception {
    System.out.println("Hello world from Command Line Runner");
  }

  @Bean(name = "dataSource")
  public DriverManagerDataSource dataSource() {
    DriverManagerDataSource dataSource = new DriverManagerDataSource();
    dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
    dataSource.setUrl("jdbc:mysql://localhost:3306/db_spring_rest?useUnicode=true&useLegacyDatetimeCode=fa    lse&serverTimezone=UTC&createDatabaseIfNotExist=true&allowPublicKeyRetrieval=true&useSSL=false");
    dataSource.setUsername("root");
    dataSource.setPassword("root");

    // schema init
    Resource initSchema = new ClassPathResource("scripts/schema.sql");
    Resource initData = new ClassPathResource("scripts/data.sql");
    DatabasePopulator databasePopulator = new ResourceDatabasePopulator(initSchema, initData);
    DatabasePopulatorUtils.execute(databasePopulator, dataSource);

    return dataSource;
  }


}