如何基于三个JPA实体类为PostgreSQL数据库填充示例数据

嘿，我来分享几个实用的方案，帮你给这三个JPA实体（顺便提一句，Java类名建议首字母大写哦，比如Dog、Cat、Shelter，符合编码规范）填充示例数据，每种方法各有适用场景，你可以按需选择：

方案1：直接执行SQL脚本 #

这是最直接的方式，适合初始化基础静态数据，你可以手动在PostgreSQL客户端（比如psql、pgAdmin）执行，或者配置项目启动时自动执行。

假设JPA默认生成的表名是dog、cat、shelter，且Shelter的@ManyToOne关联的是Dog（补全后应该是@ManyToOne private Dog dog;，对应表字段dog_id），示例SQL如下：

-- 先插入Dog数据
INSERT INTO dog (name, age) VALUES
('旺财', 3),
('小白', 2),
('大黄', 5);

-- 插入Cat数据
INSERT INTO cat (name, sex) VALUES
('咪咪', '母'),
('橘胖', '公'),
('小黑', '母');

-- 插入Shelter数据（关联已有的Dog）
INSERT INTO shelter (dog_id) VALUES
(1),
(2),
(3);

方案2：用JPA EntityManager手动插入 #

如果需要在Java代码里动态生成并插入数据，可以借助EntityManager来操作，比如在一个服务类或者初始化类中：

import jakarta.persistence.EntityManager;
import jakarta.persistence.PersistenceContext;
import org.springframework.stereotype.Component;
import jakarta.transaction.Transactional;

@Component
public class DataInitializer {

    @PersistenceContext
    private EntityManager entityManager;

    @Transactional
    public void initSampleData() {
        // 创建Dog实例
        Dog dog1 = new Dog();
        dog1.setName("旺财");
        dog1.setAge(3);
        entityManager.persist(dog1);

        Dog dog2 = new Dog();
        dog2.setName("小白");
        dog2.setAge(2);
        entityManager.persist(dog2);

        // 创建Cat实例
        Cat cat1 = new Cat();
        cat1.setName("咪咪");
        cat1.setSex("母");
        entityManager.persist(cat1);

        Cat cat2 = new Cat();
        cat2.setName("橘胖");
        cat2.setSex("公");
        entityManager.persist(cat2);

        // 创建Shelter实例并关联Dog
        Shelter shelter1 = new Shelter();
        shelter1.setDog(dog1);
        entityManager.persist(shelter1);

        Shelter shelter2 = new Shelter();
        shelter2.setDog(dog2);
        entityManager.persist(shelter2);
    }
}

方案3：Spring Boot项目用CommandLineRunner自动填充 #

如果是Spring Boot项目，你可以实现CommandLineRunner或ApplicationRunner接口，让项目启动时自动执行数据填充逻辑：

import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;
import jakarta.persistence.EntityManager;
import jakarta.persistence.PersistenceContext;
import jakarta.transaction.Transactional;

@Component
public class SampleDataLoader implements CommandLineRunner {

    @PersistenceContext
    private EntityManager entityManager;

    @Override
    @Transactional
    public void run(String... args) throws Exception {
        // 这里重复方案2里的实体创建和persist逻辑即可
        Dog dog = new Dog();
        dog.setName("大黄");
        dog.setAge(5);
        entityManager.persist(dog);

        Cat cat = new Cat();
        cat.setName("小黑");
        cat.setSex("母");
        entityManager.persist(cat);

        Shelter shelter = new Shelter();
        shelter.setDog(dog);
        entityManager.persist(shelter);
    }
}

方案4：用数据库迁移工具（Flyway/Liquibase） #

如果是生产环境，推荐用数据库迁移工具来管理数据初始化，这样数据变更可以和代码版本绑定，便于维护。以Flyway为例：

1. 在Spring Boot项目中引入Flyway依赖
2. 在src/main/resources/db/migration目录下创建SQL脚本，命名为V1__insert_sample_data.sql
3. 把方案1里的SQL内容写入这个脚本，项目启动时Flyway会自动执行这个脚本完成数据填充

方案5：测试用例中填充数据 #

如果是为测试场景准备数据，可以在测试类中用@Sql注解直接执行SQL脚本，或者在@BeforeEach方法中用JPA插入：

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
import jakarta.persistence.EntityManager;

@DataJpaTest
public class EntityTest {

    @Autowired
    private EntityManager entityManager;

    @BeforeEach
    void setUp() {
        // 测试前插入示例数据
        Dog dog = new Dog();
        dog.setName("测试旺财");
        dog.setAge(3);
        entityManager.persist(dog);
    }

    @Test
    void testEntityOperations() {
        // 测试逻辑
    }
}

小提示 #

记得补全Shelter类的@ManyToOne关联代码哦，比如：

@Entity
public class Shelter {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;
    
    @ManyToOne
    private Dog dog; // 或者关联Cat，根据你的业务需求
    
    // getter、setter省略
}

内容的提问来源于stack exchange，提问作者Makaroniasz