在 Postgres 9.5 中使用带有 WITH 子句的 upsert 时如何检索插入行的 id？

Question

我正在尝试使用“WITH”在 Postgres 9.5 中进行不安查询

with s as (
            select id
            from products
            where product_key = 'test123'
        ), i as (
            insert into products (product_key, count_parts)
            select 'test123', 33
            where not exists (select 1 from s)
            returning id                       
        )
        update products
        set product_key='test123', count_parts=33
        where id = (select id from s) 
        returning id

显然，我只在更新时检索 id，即使我知道插入成功，也没有在插入时得到任何结果。

我需要修改此查询，以便在插入和更新时都能获取 id。

谢谢！

Answer 1

我不清楚你为什么在 WITH first SELECT 时这样做，但你只返回 UPDATE id 的原因是因为你没有选择 INSERT return。

如 cmets 中所述（和链接），Postgres 9.5 支持 INSERT ON CONFLICT 子句，这是一种更简洁的使用方式。

还有一些9.5前后的例子：

9.5之前：WITH常用方式

WITH    u AS (
            UPDATE      products
            SET         product_key='test123', count_parts=33
            WHERE       product_key = 'test123'
            RETURNING   id
        ),i AS (
            INSERT
            INTO        products ( product_key, count_parts )
            SELECT      'test123', 33
            WHERE       NOT EXISTS( SELECT 1 FROM u )
            RETURNING   id
        )
SELECT  *
FROM    (       SELECT id FROM u
        UNION   SELECT id FROM i
        ) r;

9.5 之后：使用 INSERT .. ON CONFLICT

INSERT INTO products ( product_key, count_parts )
VALUES      ( 'test123', 33 )
ON CONFLICT ( product_key ) DO
UPDATE
SET         product_key='test123', count_parts=33
RETURNING   id;

更新：

正如评论中所暗示的，使用INSERT .. ON CONFLICT 方式可能会有轻微的缺点。
如果表使用自动增量并且这个查询经常发生，那么WITH 可能是一个更好的选择。
查看更多：https://stackoverflow.com/a/39000072/1161463