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

Redirecionamento de inserção baseado em gatilho Postgres sem quebrar RETURNING


A única solução que encontrei é criar uma visão para a tabela base e usar INSTEAD OF gatilhos nessa visualização:
CREATE TABLE flags_base (
    id integer NOT NULL,
    flaggable_type character varying(255) NOT NULL,
    flaggable_id integer NOT NULL,
    body text
);

ALTER TABLE ONLY flags_base
    ADD CONSTRAINT flags_base_pkey PRIMARY KEY (id);

CREATE TABLE "comment_flags" (
 CHECK ("flaggable_type" = 'Comment'),
 PRIMARY KEY ("id")
) INHERITS ("flags_base");

CREATE TABLE "profile_flags" (
 CHECK ("flaggable_type" = 'Profile'),
 PRIMARY KEY ("id")
) INHERITS ("flags_base");

CREATE OR REPLACE VIEW flags AS SELECT * FROM flags_base;

CREATE OR REPLACE FUNCTION flag_insert_trigger_fun() RETURNS TRIGGER AS $BODY$
BEGIN
  IF (NEW."flaggable_type" = 'Comment') THEN
    INSERT INTO comment_flags VALUES (NEW.*);
  ELSIF (NEW."flaggable_type" = 'Profile') THEN
    INSERT INTO profile_flags VALUES (NEW.*);
  ELSE
    RAISE EXCEPTION 'Wrong "flaggable_type"="%", fix flag_insert_trigger_fun() function', NEW."flaggable_type";
  END IF;
  RETURN NEW;
END; $BODY$ LANGUAGE plpgsql;

CREATE TRIGGER flag_insert_trigger
  INSTEAD OF INSERT ON flags
  FOR EACH ROW EXECUTE PROCEDURE flag_insert_trigger_fun();

Mas desta forma você deve forneça o id campo em cada inserção (mesmo se flags_base a chave primária de 's tem um valor padrão / é um serial), então você deve preparar seu acionador de inserção para corrigir NEW.id se for um NULL .

ATUALIZAÇÃO :Parece que as colunas das visualizações também podem ter valores padrão, definidos com
ALTER VIEW [ IF EXISTS ] name ALTER [ COLUMN ] column_name SET DEFAULT expression

que é usado apenas em visualizações têm uma regra/trigger de inserção/atualização.

http://www.postgresql.org/docs/9.3/static/sql-alterview.html