JPA应用层引用完整性：高效检查子记录以安全删除父实体

import jakarta.persistence.*;

// 父实体
@Entity
@Table(name = "parent_entity")
// 关联实体监听器
@EntityListeners(ParentEntityListener.class)
public class ParentEntity {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private String name;

    // Getter and 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 = "child_entity")
public class ChildEntity {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private String description;

    // 子实体通过外键关联父实体
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "parent_id")
    private ParentEntity parent;

    // Getter and Setter
    public Long getId() {
        return id;
    }

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

    public String getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public ParentEntity getParent() {
        return parent;
    }

    public void setParent(ParentEntity parent) {
        this.parent = parent;
    }
}

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

@Repository
public interface ChildRepository extends JpaRepository<ChildEntity, Long> {

    /**
     * 查询是否存在与给定父ID关联的子记录。
     * 使用 findFirstBy 确保只查询一条记录，提高效率。
     *
     * @param parentId 父实体的ID
     * @return 如果存在子记录则返回第一个子实体，否则返回 null。
     */
    ChildEntity findFirstByParentId(Long parentId);
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import jakarta.persistence.PreRemove;

// 将监听器声明为Spring组件，以便能够进行依赖注入
@Component
public class ParentEntityListener {

    // 自动注入ChildRepository
    // 注意：在非Spring管理的JPA环境中，这里需要通过某种方式获取Repository实例
    // 但在Spring Boot应用中，直接Autowired是推荐且便捷的方式
    private static ChildRepository childRepository;

    // Spring会在启动时调用此静态方法注入Repository
    @Autowired
    public void setChildRepository(ChildRepository childRepository) {
        ParentEntityListener.childRepository = childRepository;
    }

    /**
     * 在ParentEntity被删除之前调用此方法。
     * 检查是否存在关联的子记录。
     *
     * @param parent 要删除的ParentEntity实例
     * @throws ReferentialIntegrityException 如果存在子记录，则抛出此异常
     */
    @PreRemove
    public void preRemoveParent(ParentEntity parent) {
        // 使用findFirstByParentId方法高效检查子记录是否存在
        ChildEntity child = childRepository.findFirstByParentId(parent.getId());

        if (child != null) {
            // 如果存在子记录，则抛出自定义异常，阻止删除操作
            throw new ReferentialIntegrityException(
                "无法删除父实体 (ID: " + parent.getId() + ")，因为它仍有关联的子记录。"
            );
        }
    }
}

// 自定义引用完整性异常
public class ReferentialIntegrityException extends RuntimeException {
    public ReferentialIntegrityException(String message) {
        super(message);
    }
}