JPA 应用层级参照完整性：高效检查子记录以避免父实体删除错误

import jakarta.persistence.*;
import com.example.demo.listener.DepartmentEntityListener; // 导入监听器

@Entity
@Table(name = "departments")
@EntityListeners(DepartmentEntityListener.class) // 注册实体监听器
public class Department {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @Column(nullable = false, unique = true)
    private String name;

    // Getter和Setter方法
    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}

import jakarta.persistence.*;

@Entity
@Table(name = "employees")
public class Employee {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @Column(nullable = false)
    private String name;

    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "department_id", nullable = false)
    private Department department;

    // Getter和Setter方法
    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public Department getDepartment() {
        return department;
    }

    public void setDepartment(Department department) {
        this.department = department;
    }
}

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;

@Repository
public interface EmployeeRepository extends JpaRepository<Employee, Long> {

    /**
     * 检查是否存在属于指定部门的员工。
     * 推荐使用 existsBy，因为它直接返回布尔值，底层通常使用 EXISTS 优化查询。
     *
     * @param departmentId 部门ID
     * @return 如果存在员工，则返回 true；否则返回 false。
     */
    boolean existsByDepartmentId(Long departmentId);

    /**
     * 查找属于指定部门的第一位员工。
     * 也可以用于检查存在性，如果返回 null 则表示不存在。
     *
     * @param departmentId 部门ID
     * @return 找到的第一位员工，如果没有则返回 null。
     */
    Employee findFirstByDepartmentId(Long departmentId);
}

package com.example.demo.listener; // 确保包名与你的项目结构匹配

import com.example.demo.entity.Department;
import com.example.demo.repository.EmployeeRepository;
import jakarta.persistence.PreRemove;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component // 标记为Spring组件，以便可以注入依赖
public class DepartmentEntityListener {

    // 使用静态变量和 @Autowired 进行注入，因为JPA实例化监听器时不会通过Spring容器
    // 这种方式是Spring Boot中在JPA监听器中注入依赖的常用模式。
    private static EmployeeRepository employeeRepository;

    @Autowired
    public void setEmployeeRepository(EmployeeRepository employeeRepository) {
        DepartmentEntityListener.employeeRepository = employeeRepository;
    }

    @PreRemove // 在部门实体被删除之前调用
    public void preRemoveDepartment(Department department) {
        if (department.getId() == null) {
            // 如果ID为空，可能是新建的实体，或未持久化的实体，无需检查子记录
            return;
        }

        // 使用 existsByDepartmentId 高效检查是否存在关联员工
        boolean hasEmployees = employeeRepository.existsByDepartmentId(department.getId());

        if (hasEmployees) {
            // 如果存在关联员工，则抛出异常，阻止删除操作
            throw new IllegalStateException("无法删除部门，因为该部门下存在员工记录。");
            // 也可以自定义一个更具体的业务异常，例如：
            // throw new ReferentialIntegrityException("无法删除部门，因为该部门下存在员工记录。");
        }
    }
}