Hibernate更新含分区键关联列时生成错误SQL引发异常

问题场景 #

定义了两个实体：

• SalesContact：account_id字段标注为@PartitionKey
• ContactEmail：通过account_id和contact_id双列关联SalesContact

执行更新代码：

SalesContact salesContact = salesContactRepository.findByIdAndAccountId(id, 1679911196L).get();
salesContact.setFirstName("demo-2x");
SalesContact saved = salesContactRepository.save(salesContact);

Hibernate生成错误的Update语句：

update contacts set first_name=? where id=? and account_id=? and account_id=?

最终抛出org.springframework.dao.InvalidDataAccessResourceUsageException异常，原因是第四个参数未赋值。

原因分析 #

核心问题是Hibernate在生成Update语句时重复添加了分区键account_id的条件，通常由以下情况导致：

• SalesContact的实体映射存在冲突：分区键account_id同时被识别为查询条件的多个来源（如主键关联、分区键注解、关联映射外键）
• 双列关联ContactEmail时，account_id的映射配置导致Hibernate将其重复纳入Update的WHERE条件
• 特定版本的Hibernate分区扩展存在bug，自动生成SQL时重复拼接分区键条件

解决方案 #

1. 修正实体映射配置 #

确保SalesContact的主键和分区键映射清晰，避免重复声明：

@Entity
@Table(name = "contacts")
public class SalesContact {
    @Id
    private Long id;
    @PartitionKey
    @Column(name = "account_id")
    private Long accountId;
    private String firstName;
    // getter、setter省略
}

调整ContactEmail的双列关联配置，明确关联列与实体字段的对应关系：

@Entity
public class ContactEmail {
    @Id
    private Long id;
    @ManyToOne
    @JoinColumns({
        @JoinColumn(name = "account_id", referencedColumnName = "account_id"),
        @JoinColumn(name = "contact_id", referencedColumnName = "id")
    })
    private SalesContact salesContact;
    // 其他字段及getter、setter
}

2. 自定义Update语句 #

通过@SQLUpdate注解强制指定正确的Update语句，绕过Hibernate的自动生成逻辑：

@SQLUpdate(sql = "update contacts set first_name = ? where id = ? and account_id = ?")
@Entity
@Table(name = "contacts")
public class SalesContact {
    // 实体字段配置
}

3. 升级Hibernate版本 #

如果使用较旧的Hibernate分区相关扩展（如Hibernate Shards或特定云数据库适配插件），升级到最新稳定版本，这类重复条件的bug通常会被修复。

4. 优化Repository操作逻辑 #

避免查询后直接调用save时的条件冲突，可通过自定义更新方法替代默认save：

public interface SalesContactRepository extends JpaRepository<SalesContact, Long> {
    @Modifying
    @Query("update SalesContact sc set sc.firstName = ?1 where sc.id = ?2 and sc.accountId = ?3")
    int updateFirstNameByIdAndAccountId(String firstName, Long id, Long accountId);
}

调用时直接使用自定义方法：

salesContactRepository.updateFirstNameByIdAndAccountId("demo-2x", id, 1679911196L);

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