Postgres函数更新旧数据并插入新数据

编程问答 2022-06-07

问题描述

我正在尝试编写一个函数，当我的脚本从AD中提取信息时，它会插入不存在的新条目，然后更新应有的条目。新条目将具有来自序列的ID。

CREATE OR REPLACE FUNCTION "SCHEMA".fn_name()
    returns trigger AS 
$$
    begin
        if new.dn_id is null then
            NEW.dn_id = nextval("SCHEMA".seq_name); 
        else if (TG_OP = 'UPDATE') then 
            insert into "SCHEMA"."TABLE_NAME"(dn,dn_name,search,user_cat) values (new.dn,new.dn_name,new.search,new.user_cat) where old.dn_name = new.dn_name;
    RETURN NEW;
    end;
$$

LANGUAGE 'plpgsql';\

create trigger trg_name
before insert or update
on "SCHEMA"."TABLE"
for each row
execute procedure "SCHEMA".fn_name();

我对Postgresql很陌生。我知道在oracle中，您可以按以下方式在触发器中使用参考NEW作为新的Old； 

create or replace TRIGGER trg_name
    BEFORE INSERT
    ON table
    REFERENCING NEW AS new OLD AS old
    FOR EACH ROW
BEGIN
    SELECT seq_name.NEXTVAL INTO :new.dn_id FROM DUAL;
END trg_name;

如何获得与Postgres中的oracle具有的功能相同的功能？

解决方法

合并

似乎MERGE尚未在Postgres中实现。

方法1-INSERT ... ON CONFLICT

了解更多here with theory & examples。

这种方式需要您知道的名称约束（PRIMARY KEY或任何UNIQUE约束）（不会自动生成）。例如，名为pk_known_languages的约束。

INSERT INTO public.known_languages(id,"comment",translations)
 VALUES ('en','English','Translations here'),('ru','Русский','Тут переводы')
ON CONFLICT ON CONSTRAINT pk_known_languages
  DO UPDATE
  SET "comment" = EXCLUDED."comment",translations = EXCLUDED.translations;

注意：要DELETE记录某些记录，您需要方法2。

方法2-使用CTE

进行大查询

CTE-Common Table Expresson -是常规SQL命令（SELECT,INSERT,UPDATE,DELETE甚至VALUES），在附加主SQL命令之前执行。了解更多here with theory & examples。

CTE的示例：

    WITH cte_name AS (
        SELECT *
        FROM public.my_table_1
    )
    INSERT INTO public.my_table_2
    SELECT *
    FROM cte_name;

这是一个大查询，cte_name的结果充当临时表，可在主命令中使用。

由同一id合并的解决方案：

（这将更新具有相同id的记录，并插入具有新NULL或id的记录。）

WITH data(id,translations) AS (
  VALUES ('en','Тут переводы')
),upd AS (
   UPDATE public.known_languages
     SET
       "comment" = data."comment",translations = data.translations
     FROM data
     WHERE known_languages.id = data.id
     RETURNING data.id   -- return all mathced ids
 )
 INSERT INTO public.known_languages
   SELECT *
   FROM data
   WHERE data.id NOT IN (SELECT id
                         FROM upd);

下一个CTE可以使用所有先前CTE返回的数据。 首先CTE，data返回您要插入/更新的数据。 第二个CTE，upd对具有相同id的行进行更新，并返回匹配的id，以便在后续插入时忽略它们。 第三，常规INSERT命令使用upd的结果从id中排除这样的INSERT。

如果所有新记录始终没有ID：

（此变体仅用于提高速度，在这种情况下，您仍然可以使用以前的查询。）

WITH data(id,translations) AS (
  VALUES (NULL,translations = data.translations
     FROM data
     WHERE known_languages.id = data.id
       AND data.id IS NOT NULL
 )
 INSERT INTO public.known_languages
   SELECT *
   FROM data
   WHERE data.id IS NULL;

注意：

此查询还可以扩展到DELETE中不存在的data记录（或任何其他条件）。只需将INSERT转换为第三CTE并添加DELETE语句作为主命令即可。

functionfunctionfunctionpostgresqlpostgresql-9.6triggers