Spring Data与Hibernate中可用于JOIN的复合主键实现问题

问题根源 #

核心问题是双向关联中，关联关系的tenant_id列与实体自身的主键tenantId列重复映射，导致Hibernate生成SQL时重复添加该列，进而出现参数未绑定的错误。另外，你的EntityC存在映射错误（关联对象应为EntityB而非自身）。

修正步骤 #

1. 修正关联映射的重复列问题 #

在@ManyToOne的@JoinColumns中，将关联tenant_id的列标记为insertable = false, updatable = false——因为该列已经作为实体的主键字段被维护，不需要由关联关系来处理插入和更新。

2. 修正EntityC的映射错误 #

你当前的EntityC中错误地关联了自身，应改为关联EntityB。

修正后的实体代码 #

BaseTenantId（保持不变）

@Data
public class BaseTenantId implements Serializable {
    private UUID id;
    private Long tenantId;
}

EntityA（保持不变）

@IdClass(BaseTenantId.class)
@Entity
@Data
@Table("table_a")
public class EntityA {
    @Id
    @GeneratedValue
    private UUID id;
    @Id
    @Column(name = "tenant_id")
    private Long tenantId;
    @OneToMany(mappedBy = "entityA", cascade = CascadeType.ALL)
    private List<EntityB> entityBList = new ArrayList<>();
}

EntityB（修正关联列）

@IdClass(BaseTenantId.class)
@Entity
@Data
@Table("table_b")
public class EntityB {
    @Id
    @GeneratedValue
    private UUID id;
    @Id
    @Column(name = "tenant_id")
    private Long tenantId;
    
    @ManyToOne
    @JoinColumns(
        value = {
            @JoinColumn(referencedColumnName = "id", name = "entity_a_id"),
            // 标记tenant_id列不由关联关系维护
            @JoinColumn(referencedColumnName = "tenant_id", name = "tenant_id", insertable = false, updatable = false)
        })
    private EntityA entityA;   
    
    @OneToMany(mappedBy = "entityB", cascade = CascadeType.ALL)
    private List<EntityC> entityCList = new ArrayList<>();
}

EntityC（修正关联对象+关联列）

@IdClass(BaseTenantId.class)
@Entity
@Data
@Table("table_c")
public class EntityC {
    @Id
    @GeneratedValue
    private UUID id;
    @Id
    @Column(name = "tenant_id")
    private Long tenantId;
    
    @ManyToOne
    @JoinColumns(
        value = {
            @JoinColumn(referencedColumnName = "id", name = "entity_b_id"),
            // 标记tenant_id列不由关联关系维护
            @JoinColumn(referencedColumnName = "tenant_id", name = "tenant_id", insertable = false, updatable = false)
        })
    private EntityB entityB; 
}

3. 持久化时的注意事项 #

保存实体时，必须保证关联实体的tenantId与父实体一致（符合复合主键约束），同时维护双向关联的一致性：

// 示例保存逻辑
EntityA entityA = new EntityA();
entityA.setTenantId(1L);

EntityB entityB = new EntityB();
entityB.setTenantId(1L); // 必须与EntityA的tenantId一致
entityB.setEntityA(entityA);
entityA.getEntityBList().add(entityB);

EntityC entityC = new EntityC();
entityC.setTenantId(1L); // 必须与EntityB的tenantId一致
entityC.setEntityB(entityB);
entityB.getEntityCList().add(entityC);

entityARepository.save(entityA);

额外说明 #

• 该方案适配Postgres分区表：Hibernate在JOIN查询时会自动带上tenant_id作为关联条件，满足分区表的路由要求。
• 若后续需要更灵活的复合主键管理，可考虑替换@IdClass为@EmbeddedId，核心解决思路一致。
• 自定义Repository查询方法时，需确保所有查询都包含tenantId条件，避免跨租户数据访问。

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