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

时间:2016-01-28 19:47:04

标签: java hibernate jpa orm spring-data-jpa

我正在开发一个在集群环境中运行的项目,其中有许多节点和一个数据库。该项目使用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. 查找&gt; Act - 服务方法将使用存储库查看条目是否已存在,然后根据需要更新找到的条目或保存新条目。这不起作用。
  2. 尝试插入&gt;如果失败则更新 - 服务方法将尝试插入,由于唯一约束而捕获异常,然后执行更新。这不起作用,因为事务已经处于回滚状态,并且不能在其中进行进一步的操作。
  3. 原生查询&#34; INSERT INTO ... WHERE NOT EXISTS ...&#34; * - 存储库有一个新的原生查询:

    @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似乎首先执行SELECT子句所使用的WHERE - 这意味着最终会尝试多次插入,从而导致唯一的约束违规

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

    编辑2016-02-02

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

4 个答案:

答案 0 :(得分:0)

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

答案 1 :(得分:0)

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

答案 2 :(得分:0)

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

为什么不起作用?

您是否考虑过“乐观锁定”?

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

答案 3 :(得分:0)

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