Spring Boot-加载初始数据

Question

我想知道在应用程序启动之前加载初始数据库数据的最佳方法是什么？我正在寻找的是一些东西，将填补我的H2数据库与数据。

例如，我有一个域模型“user”我可以通过转到/users来访问用户，但最初数据库中没有任何用户，所以我必须创建它们。是否存在自动填充数据库的数据？

@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"));
    }
}

但我很怀疑这是最好的方法。还是真的？

Answer 1

您可以在src/main/resources文件夹中创建一个data.sql文件，它将在启动时自动执行。在这个文件中，您可以添加一些insert语句，例如：

INSERT INTO users (username, firstname, lastname) VALUES
  ('lala', 'lala', 'lala'),
  ('lolo', 'lolo', 'lolo');

类似地，您也可以创建schema.sql文件（或schema-h2.sql)来创建您的模式：

CREATE TABLE task (
  id          INTEGER PRIMARY KEY,
  description VARCHAR(64) NOT NULL,
  completed   BIT NOT NULL);

虽然通常不需要这样做，因为Spring boot已经将Hibernate配置为基于内存数据库的实体创建模式。如果您真的想要使用schema.sql，则必须通过将其添加到应用程序中来禁用此功能。属性：

spring.jpa.hibernate.ddl-auto=none

spring.datasource.initialization-mode=always

为此，您必须配置spring.datasource.platform属性：

spring.datasource.platform=h2