使用 Spring JPA / Hibernate 进行条件插入

Posted 2023-02-26

【中文标题】使用 Spring JPA / Hibernate 进行条件插入

【英文标题】：Conditional insert with Spring JPA / Hibernate

【发布时间】：2016-05-06 08:52:28

【问题描述】：

我正在开发一个在集群环境中运行的项目，其中有许多节点和一个数据库。该项目使用 Spring-data-JPA (1.9.0) 和 Hibernate (5.0.1)。我无法解决如何防止重复行问题。

为了举例，这里有一个简单的表格

@Entity
@Table(name = "scheduled_updates")
public class ScheduledUpdateData 
    public enum UpdateType 
        TYPE_A,
        TYPE_B
    

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    @Column(name = "id")
    private UUID id;

    @Column(name = "type", nullable = false)
    @Enumerated(EnumType.STRING)
    private UpdateType type;

    @Column(name = "source", nullable = false)
    private UUID source;

重要的是有一个UNIQUE(type, source) 约束。

当然还有匹配的示例存储库：

@Repository
public class ScheduledUpdateRepository implements JpaRepository<ScheduledUpdateData, UUID> 
    ScheduledUpdateData findOneByTypeAndSource(final UpdateType type, final UUID source);

    //...

此示例的想法是，系统的某些部分可以插入行以安排定期运行的某些内容，在所述运行之间可多次运行。无论什么东西实际运行，它都不必担心在同一东西上运行两次。

如何编写一个有条件地插入该表的服务方法？我尝试过的一些不起作用的方法是：

Find > Act - 服务方法将使用存储库来查看条目是否已存在，然后根据需要更新找到的条目或保存新条目。这不起作用。 尝试插入 > 如果失败则更新 - 服务方法将尝试插入，捕获由于唯一约束引起的异常，然后进行更新。这不起作用，因为事务已经处于回滚状态，并且无法在其中进行进一步的操作。

带有“INSERT INTO ... WHERE NOT EXISTS ...”的本机查询* - 存储库有一个新的本机查询：

@Repository
public class ScheduledUpdateRepository implements JpaRepository<ScheduledUpdateData, UUID> 
    // ...

    @Modifying
    @Query(nativeQuery = true, value = "INSERT INTO scheduled_updates (type, source)" +
                                       " SELECT :type, :src" +
                                       " WHERE NOT EXISTS (SELECT * FROM scheduled_updates WHERE type = :type AND source = :src)")
    void insertUniquely(@Param("type") final String type, @Param("src") final UUID source);

不幸的是，这也不起作用，因为 Hibernate 似乎首先执行WHERE 子句使用的SELECT - 这意味着最终会尝试多次插入，从而导致违反唯一约束。

我绝对不知道 JTA、JPA 或 Hibernate 的许多细节。关于如何跨多个 JVM 插入具有唯一约束（不仅仅是主键）的表中的任何建议？

编辑 2016-02-02

使用 Postgres (2.3) 作为数据库，尝试使用 隔离级别 SERIALIZABLE - 遗憾的是，这仍然会导致违反约束的异常。

【问题讨论】：

尝试将@version注解添加到实体和版本字段到表。

重试整个事务（第二次更新而不是插入）。

@sky_light 不幸的是，@Version 没有帮助，因为当两个事务认为该行尚不存在时会发生错误。

@DraganBozanovic 在整个项目中，单个事务可能会更新多个表，其中任何一个或所有表都可能具有独特的约束，因此重试整个事务听起来需要引入一个全新的框架层在现有的 @Services 之上 - 这似乎不正确

如果serviceA在一个事务中运行，并调用serviceB在自己的事务中运行（即使用REQUIRES_NEW），那么您可以在serviceA中捕获serviceB抛出的异常，并且不会回滚serviceA。

【参考方案1】：

Hibernate 及其查询语言支持插入语句。因此，您实际上可以使用 HQL 编写该查询。浏览此处获取更多信息。 http://docs.jboss.org/hibernate/orm/5.0/userguide/html_single/Hibernate_User_Guide.html#_hql_syntax_for_insert

【讨论】：

虽然这在理论上可以回答这个问题，it would be preferable 在此处包含答案的基本部分，并提供链接以供参考。

【参考方案2】：

这听起来像是一个 upsert 案例，可以按照 here 的建议处理。

【参考方案3】：

Find > Act - 服务方法将使用存储库查看条目是否已存在，然后根据需要更新找到的条目或保存新条目。这不起作用。

为什么这不起作用？

你考虑过“乐观锁”吗？

这两个帖子可能会有所帮助：

https://www.baeldung.com/jpa-optimistic-locking

https://www.baeldung.com/java-jpa-transaction-locks

【讨论】：

这些都不能解决 update-if-exists-insert-otherwise 问题。锁定仅适用于单个记录。如果在初始查询时该记录尚不存在，则无需锁定任何内容。锁定整个表是可能的，但这会造成严重的性能瓶颈。更好的方法是在数据库中创建一个单记录“锁定”表，并将其用作信号量。详情见我的回复。

【参考方案4】：

您正在尝试确保一次只有 1 个节点可以执行此操作。 最好的（或至少与数据库无关的）方法是使用“锁定”表。 该表将只有一行，并将充当信号量以确保串行访问。

确保此方法包含在事务中

// this line will block if any other thread already has a lock
// until that thread's transaction commits
Lock lock = entityManager.find(Lock.class, Lock.ID, LockModeType.PESSIMISTIC_WRITE);

// just some change to the row, it doesn't matter what
lock.setDateUpdated(new Timestamp(System.currentTimeMillis()));  
entityManager.merge(lock);
entityManager.flush();

// find your entity by unique constraint
// if it exists, update it
// if it doesn't, insert it