JPA/Hibernate嵌入式复合主键处理Null ID生成错误的最佳实践

@Embeddable
@Data // Lombok注解，用于生成getter/setter, equals, hashCode等
public class BlockAttributeID implements Serializable {

    @Column(name = "block_id")
    Long blockID; // 仅包含Block的ID

    String attribute;

    // equals 和 hashCode 方法的实现需要注意，尤其是当blockID可能为null时
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof BlockAttributeID)) return false;
        BlockAttributeID that = (BlockAttributeID) o;
        return Objects.equals(blockID, that.blockID) && Objects.equals(attribute, that.attribute);
    }

    @Override
    public int hashCode() {
        return Objects.hash(blockID, attribute);
    }
}

@Data
@Table(name = "block_attribute")
@Entity
public class BlockAttribute {

    @EmbeddedId
    BlockAttributeID blockAttributeID;

    // 冗余的ManyToOne关联，与EmbeddedId中的blockID形成冲突或混淆
    @ManyToOne(fetch = FetchType.LAZY)
    @JsonIgnore
    @JoinColumn(name = "block_id") // 这个@JoinColumn通常会导致问题
    Block block; // 这里又有一个Block实体引用

    String label;
    // ... 其他字段

    // equals 和 hashCode 同样需要基于复合主键进行正确实现
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof BlockAttribute)) return false;
        BlockAttribute that = (BlockAttribute) o;
        return Objects.equals(blockAttributeID, that.blockAttributeID);
    }

    @Override
    public int hashCode() {
        return Objects.hash(blockAttributeID);
    }
}

@Table(name = "block")
@Entity
@Data
public class Block {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name = "block_id")
    Long blockID; // Block的主键

    // ... 其他字段和关联

    @OneToMany(mappedBy = "block", fetch = FetchType.LAZY, cascade = CascadeType.ALL, orphanRemoval = true)
    Set<BlockAttribute> blockAttributes = new HashSet<>();

    // ... 其他方法
}

// 1. 保存父Block实体，生成其blockID
block = blockRepository.save(block);

// 2. 设置BlockAttribute的block字段
blockAttribute.setBlock(block); // 此时blockAttributeID中的blockID并未被设置

// 3. 尝试保存BlockAttribute
blockAttributeRepository.save(blockAttribute); // 抛出Null ID错误

import com.fasterxml.jackson.annotation.JsonIgnore;
import lombok.Data; // 推荐使用Lombok简化代码
import javax.persistence.*;
import java.io.Serializable;
import java.util.Objects;

@Embeddable
@Data // 确保生成了getter/setter以及默认的equals/hashCode，但需手动优化
public class BlockAttributeID implements Serializable {

    // 将ManyToOne关联直接嵌入到复合主键类中
    @ManyToOne(fetch = FetchType.LAZY)
    @JsonIgnore // 通常在嵌入式ID中，避免序列化Block实体，防止循环引用
    @JoinColumn(name = "block_id", referencedColumnName = "block_id") // 明确指定关联列
    Block block; // 现在直接持有Block实体引用

    String attribute;

    // 构造函数，方便创建复合主键实例
    public BlockAttributeID(Block block, String attribute) {
        this.block = block;
        this.attribute = attribute;
    }

    // JPA规范要求存在无参构造函数
    public BlockAttributeID() {
    }

    // 优化后的equals方法：基于Block的ID和attribute进行比较
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof BlockAttributeID)) return false;
        BlockAttributeID that = (BlockAttributeID) o;
        // 比较Block实体时，应比较其主键ID，而不是整个实体对象，以避免代理问题
        return Objects.equals(
            this.block != null ? this.block.getBlockID() : null,
            that.block != null ? that.block.getBlockID() : null
        ) && Objects.equals(this.attribute, that.attribute);
    }

    // 优化后的hashCode方法：基于Block的ID和attribute生成
    @Override
    public int hashCode() {
        return Objects.hash(
            this.block != null ? this.block.getBlockID() : null,
            this.attribute
        );
    }
}

import lombok.Data;
import javax.persistence.*;
import java.util.Objects;

@Data
@Table(name = "block_attribute")
@Entity
public class BlockAttribute {

    @EmbeddedId
    BlockAttributeID blockAttributeID; // 复合主键，现在包含了Block的引用

    // 移除冗余的Block字段，因为它已经包含在blockAttributeID中
    // @ManyToOne(fetch = FetchType.LAZY)
    // @JsonIgnore
    // @JoinColumn(name = "block_id")
    // Block block;

    String label;

    @Enumerated(EnumType.STRING)
    Type type;

    @Enumerated(EnumType.STRING)
    Unit unit;

    String value;

    // equals 和 hashCode 应该基于 @EmbeddedId
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof BlockAttribute)) return false;
        BlockAttribute that = (BlockAttribute) o;
        return Objects.equals(blockAttributeID, that.blockAttributeID);
    }

    @Override
    public int hashCode() {
        return Objects.hash(blockAttributeID);
    }
}

// 1. 首先保存父Block实体，确保其主键（blockID）已生成
Block savedBlock = blockRepository.save(block);

// 2. 创建BlockAttributeID实例，传入已保存的Block实体和attribute值
// 此时savedBlock已经拥有了数据库生成的主键ID
BlockAttributeID blockAttributeID = new BlockAttributeID(savedBlock, completeBlockDTO.getBlockAttributeDTO().getAttribute());

// 3. 将创建好的BlockAttributeID设置到BlockAttribute实体中
blockAttribute.setBlockAttributeID(blockAttributeID);

// 4. 保存BlockAttribute实体
blockAttributeRepository.save(blockAttribute);

// 对于其他依赖于Block的子实体（如BlockBoundary），如果其关联方式是ManyToOne，
// 则可以直接设置Block实体引用，因为它的ID是独立的，不作为其复合主键的一部分。
// blockBoundary.setBlock(savedBlock);
// blockBoundaryRepository.save(blockBoundary);