PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Inserir, em atualização duplicada no PostgreSQL?


PostgreSQL desde a versão 9.5 tem sintaxe UPSERT, com cláusula ON CONFLICT. com a seguinte sintaxe (semelhante ao MySQL)
INSERT INTO the_table (id, column_1, column_2) 
VALUES (1, 'A', 'X'), (2, 'B', 'Y'), (3, 'C', 'Z')
ON CONFLICT (id) DO UPDATE 
  SET column_1 = excluded.column_1, 
      column_2 = excluded.column_2;

Pesquisar nos arquivos do grupo de e-mail do postgresql por "upsert" leva a encontrar um exemplo de como fazer o que você possivelmente deseja fazer, no manual:

Exemplo 38-2. Exceções com UPDATE/INSERT

Este exemplo usa manipulação de exceção para executar UPDATE ou INSERT, conforme apropriado:
CREATE TABLE db (a INT PRIMARY KEY, b TEXT);

CREATE FUNCTION merge_db(key INT, data TEXT) RETURNS VOID AS
$$
BEGIN
    LOOP
        -- first try to update the key
        -- note that "a" must be unique
        UPDATE db SET b = data WHERE a = key;
        IF found THEN
            RETURN;
        END IF;
        -- not there, so try to insert the key
        -- if someone else inserts the same key concurrently,
        -- we could get a unique-key failure
        BEGIN
            INSERT INTO db(a,b) VALUES (key, data);
            RETURN;
        EXCEPTION WHEN unique_violation THEN
            -- do nothing, and loop to try the UPDATE again
        END;
    END LOOP;
END;
$$
LANGUAGE plpgsql;

SELECT merge_db(1, 'david');
SELECT merge_db(1, 'dennis');

Há possivelmente um exemplo de como fazer isso em massa, usando CTEs na versão 9.1 e superior, na lista de discussão de hackers:
WITH foos AS (SELECT (UNNEST(%foo[])).*)
updated as (UPDATE foo SET foo.a = foos.a ... RETURNING foo.id)
INSERT INTO foo SELECT foos.* FROM foos LEFT JOIN updated USING(id)
WHERE updated.id IS NULL;

Veja a resposta de a_horse_with_no_name para um exemplo mais claro.