sql - PostgreSQL:递增否则插入

Question

我有一些行经常以增量更新，但很少插入。是否可以切换新的 INSERT ... ON CONFLICT 语句的顺序以优化更新而不是插入？

现在我正在这样做:

INSERT INTO ?? (??) VALUES (?) ON CONFLICT(??) DO UPDATE SET ?? = ?? + 1 RETURNING ??

虽然这可行，但即使插入失败，它也会每次增加主键的序列。

是否有可能以第一个操作是更新的方式重写查询，并且只有在没有执行更新时才执行插入？

Answer 1

我不知道它的内置命令，但你可以为此编写一个存储过程:

CREATE OR REPLACE FUNCTION update_or_insert(in_parameter1 INTEGER, ...) RETURNING SETOF my_table AS $$
DECLARE
    result my_table%ROWTYPE;
BEGIN
    WITH updated_rows AS (
        UPDATE my_table SET ... WHERE ... RETURNING *
    )
    SELECT * INTO result FROM updated_rows;
    IF FOUND THEN
        RETURN NEXT result;
    ELSE
        WITH inserted_rows AS (
            INSERt INTO my_table (...) VALUES (...) RETURNING *
        )
        SELECT * INTO result FROM inserted_rows;
        RETURN NEXT result;
    END IF;
RETURN;
$$ LANGUAGE plpgsql;

您可以按如下方式调用此函数:

SELECT * FROM update_or_insert(123, ...);