PostgreSQL
 sql >> Datenbank >  >> RDS >> PostgreSQL

Upsert mit einer Transaktion

Angenommen diese einfache Tabelle:

CREATE TABLE tbl(id int primary key, value int);

Das funktioniert fast 100 % sicher (siehe Kommentare) für gleichzeitige Transaktionen.:

CREATE OR REPLACE FUNCTION f_upsert(_id int, _value int)
  RETURNS void AS
$func$
BEGIN
LOOP
   UPDATE tbl SET value = _value WHERE  id = _id;

   EXIT WHEN FOUND;

   BEGIN
      INSERT INTO tbl (id, value)
      VALUES (_id, _value);

      RETURN;

   EXCEPTION WHEN UNIQUE_VIOLATION THEN     -- tbl.id has UNIQUE constraint.
      RAISE NOTICE 'It actually happened!'; -- hardly ever happens
   END;

END LOOP;
END
$func$ LANGUAGE plpgsql;

Aufruf:

SELECT f_upsert(2, 2);

Es ist diesem INSERT / SELECT sehr ähnlich Fall mit weiteren Erklärungen und Links:

  • Ist SELECT oder INSERT in einer Funktion anfällig für Race-Conditions?