PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Insérer, lors d'une mise à jour en double dans PostgreSQL ?

PostgreSQL depuis la version 9.5 a la syntaxe UPSERT, avec la clause ON CONFLICT. avec la syntaxe suivante (similaire à 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;

La recherche de "upsert" dans les archives des groupes de messagerie de postgresql conduit à trouver un exemple de ce que vous voulez éventuellement faire, dans le manuel :

Exemple 38-2. Exceptions avec UPDATE/INSERT

Cet exemple utilise la gestion des exceptions pour effectuer UPDATE ou INSERT, selon le cas :

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');

Il existe peut-être un exemple de la façon de procéder en masse, en utilisant les CTE dans la version 9.1 et supérieure, dans la liste de diffusion des pirates :

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;

Voir la réponse de a_horse_with_no_name pour un exemple plus clair.