解决JPA中嵌入式主键与多对一关系导致的Null ID生成错误

@Embeddable
@Data
public class BlockAttributeID implements Serializable {

    @Column(name = "block_id")
    Long blockID; // 引用Block的ID

    String attribute;

    // equals和hashCode方法省略，但必须正确实现
    @Override
    public boolean equals(Object o) { /* ... */ }
    @Override
    public int hashCode() { /* ... */ }
}

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

    @EmbeddedId
    BlockAttributeID blockAttributeID; // 嵌入式复合主键

    @ManyToOne(fetch = FetchType.LAZY)
    @JsonIgnore
    @JoinColumn(name = "block_id") // 再次声明block_id外键
    Block block; // 多对一关系

    String label;
    // ... 其他属性

    // equals和hashCode方法省略，但必须正确实现
    @Override
    public boolean equals(Object o) { /* ... */ }
    @Override
    public int hashCode() { /* ... */ }
}

@Embeddable
@Data // 或手动添加getter/setter
public class BlockAttributeID implements Serializable {

    // 直接在嵌入式ID中定义ManyToOne关系
    @ManyToOne(fetch = FetchType.LAZY)
    @JsonIgnore // 如果需要，忽略JSON序列化
    @JoinColumn(name = "block_id") // 指定外键列
    Block block; // 引用Block实体本身

    String attribute;

    // 构造函数用于方便创建ID实例
    public BlockAttributeID(Block block, String attribute) {
        this.block = block;
        this.attribute = attribute;
    }

    public BlockAttributeID() {
        // 默认构造函数，JPA要求
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof BlockAttributeID)) return false;
        // 比较时使用Block实体和attribute
        BlockAttributeID that = (BlockAttributeID) o;
        return block != null && attribute != null && block.equals(that.getBlock()) && attribute.equals(that.getAttribute());
    }

    @Override
    public int hashCode() {
        // 推荐使用Objects.hash(block, attribute) 或其他更健壮的哈希算法
        // 简单的实现可以基于block的ID和attribute的哈希值
        return getClass().hashCode(); // 也可以根据实际情况调整，例如 Objects.hash(block.getBlockID(), attribute);
    }
}

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

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

    // 原有的 @ManyToOne Block block 字段被移除

    String label;
    @Enumerated(EnumType.STRING)
    Type type;
    // ... 其他属性

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof BlockAttribute)) return false;
        BlockAttribute that = (BlockAttribute) o;
        return blockAttributeID != null && blockAttributeID.equals(that.getBlockAttributeID());
    }

    @Override
    public int hashCode() {
        return getClass().hashCode();
    }
}

// 假设 block 是一个已经持久化并具有有效ID的Block实体
// block = blockRepository.save(block); // 确保block已保存并获取其ID

// 创建BlockAttribute实例
BlockAttribute blockAttribute = new BlockAttribute();
// ... 设置blockAttribute的其他属性

// 手动创建并设置BlockAttributeID
// 确保这里的 'block' 是已经从数据库中加载或刚刚保存的，具有有效ID的Block实例
BlockAttributeID blockAttributeID = new BlockAttributeID(block, completeBlockDTO.getBlockAttributeDTO().getAttribute());
blockAttribute.setBlockAttributeID(blockAttributeID);

// 保存BlockAttribute
blockAttributeRepository.save(blockAttribute);