postgres:更新冲突插入行并返回旧值

时间:2019-03-16 17:32:03

标签: sql postgresql race-condition insert-update select-for-update

我需要查询来更新表中的行,但是如果id不存在,它将插入默认值。另外,它还必须避免线程竞争条件。

我在这里找到了一个应该没问题的答案 https://stackoverflow.com/a/7927957/8372336

使用此查询:

UPDATE tbl x
SET    tbl_id = 24
     , name = 'New Gal'
FROM  (SELECT tbl_id, name FROM tbl WHERE tbl_id = 4 FOR UPDATE) y 
WHERE  x.tbl_id = y.tbl_id
RETURNING y.tbl_id AS old_id, y.name AS old_name, x.tbl_id, x.name;

所以我认为它应该在更新后返回旧值,并且应该防止线程争用情况。

但是如果行不存在,我需要添加一个插入,并且这次还返回插入的值(旧值没有意义,因为它们不存在)。

所以基本上我需要做类似的事情

INSERT INTO tbl 
    (...) VALUES (...) 
    RETURNING ..., ... 
ON CONFLICT DO
UPDATE tbl x
SET    tbl_id = 24
     , name = 'New Gal'
FROM  (SELECT tbl_id, name FROM tbl WHERE tbl_id = 4 FOR UPDATE) y 
WHERE  x.tbl_id = y.tbl_id
RETURNING y.tbl_id AS old_id, y.name AS old_name, x.tbl_id, x.name;

但是我不确定是否可以使用这种方法。我如何使它工作并确保比赛条件?

1 个答案:

答案 0 :(得分:2)

以某种方式,这似乎可行:

insert into t (x)
    values ('a0'), ('b')
    on conflict (x) do update
        set x = excluded.x || '0'
    returning i, x, (select x from t t2 where t2.i = t.i);

我很惊讶,因为t在子查询中,而excluded在子查询中。嗯。 。 。可能是因为它不是on conflict子句的一部分,而是整个insert的一部分。这开始变得有意义。

Here是此版本的db <>小提琴。

我认为您的代码如下:

INSERT INTO tbl (...)
    VALUES (...) 
ON CONFLICT DO
UPDATE tbl x
    SET tbl_id = 24,
        name = 'New Gal'
RETURNING (SELECT t2.tbl_id FROM tbl t2 WHERE t2.tbl_id = tbl.tbl_id) AS old_id, 
          (SELECT t2.name FROM tbl t2 WHERE t2.tbl_id = tbl.tbl_id) AS old_name, 
          x.tbl_id, x.name;