PostgreSQL 创建分区表,使用JPA插入报错

1.遇到的问题:

JPA对 PostgreSQL 分区表进行数据插入报错:

org.hibernate.StaleStateException: Batch update returned unexpected row count from update [0]; actual row count: 0; expected: 1

2.根本原因:

JPA对 PostgreSQL 分区表进行数据插入的时候,因为数据实际插入的是分区表,主表没有数据,

所以主表返回 "Affected rows: 0",但是hibernate源码(JPA默认实现是hibernate) 中有一个方法: org.hibernate.jdbc.Expectations.checkBatched

这个方法会检测"Affected rows",不等于变量 expectedRowCount (默认值为1) 则会抛出异常.

3.解决方案

3.1 方案一: 把触发器返回值改成 NEW,并且 id 改成负数.

这样就会往主表里面插入一行id为负数的数据, "Affected rows"就会变成1.

然后插入之后再根据id<0 删掉这条数据即可.

SQL 触发器的 function代码:

CREATE OR REPLACE FUNCTION xxx()
RETURNS TRIGGER AS $$
BEGIN
//...
//...
    NEW.id = NEW.id * -1;
    RETURN NEW;
END;

Java代码:

//保存
repository.saveAll(entityList);

//删除触发器保存的id小于0的数据
repository.deleteByIdLessZero();

3.2 方案二: 重写 hibernate 的检查 "Affected rows" 的机制.

但是这个文章是2016年的,距今5,6年了,而且这个修改等于是使"Affected rows"检测机制失效,对全局有影响.所以我没有使用.

重写 hibernate 的检查 "Affected rows" 的机制 - 相关文章:

 Spring-JPA可以使用Postgres分区吗？

Custom Batch Builder/Batch in Spring-JPA

方案编写不易,如果对你有帮助,请点赞或收藏~~~