带弹簧 JPA / 休眠的条件插入

2022-09-03 16:13:54

我正在处理一个在集群环境中运行的项目,其中有许多节点和单个数据库。该项目使用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);

    //...
}

此示例的想法是,系统的各个部分可以插入行,以计划定期运行的内容,在所述运行之间任意次数。当某些东西实际运行时,它不必担心对同一事物进行两次操作。

如何编写有条件地插入到此表中的服务方法?我尝试过的一些不起作用的事情是:

  1. 查找>行为 - 服务方法将使用存储库来查看某个条目是否已存在,然后根据需要更新找到的条目或保存新条目。这不起作用。
  2. 尝试插入>如果失败则更新 - 服务方法将尝试插入,捕获由于唯一约束而导致的异常,然后改为执行更新。这不起作用,因为事务将已处于回滚状态,并且无法在其中执行其他操作。
  3. 带有“插入到...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似乎首先执行子句使用 - 这意味着最终尝试多个插入,导致唯一的约束冲突。SELECTWHERE

我绝对不知道JTA,JPA或Hibernate的很多细节。关于如何跨多个 JVM 插入具有唯一约束(不仅仅是主键)的表的任何建议?

编辑 2016-02-02

使用Postgres(2.3)作为数据库,尝试使用隔离级别SERIALIZABLE - 可悲的是,这本身仍然会导致约束冲突异常。


答案 1

您正在尝试确保一次只有 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

答案 2

Hibernate 及其查询语言提供对插入语句的支持。因此,您实际上可以使用 HQL 编写该查询。有关详细信息,请参阅此处。http://docs.jboss.org/hibernate/orm/5.0/userguide/html_single/Hibernate_User_Guide.html#_hql_syntax_for_insert


推荐