如何在 Postgres 中使用 upsert

Question

我想將 Postgres 中的這段代碼轉換為更短的代碼，可以做同樣的事情。 我讀到了 upsert，但我不明白在我的代碼上實現它的好方法。 我寫的東西很好，但我想找到一種更優雅的方式來寫它。 希望這里有人可以幫助我：這是查詢：

CREATE OR REPLACE FUNCTION insert_table(
    in_guid    character varying,
    in_x_value character varying,
    in_y_value character varying
)
RETURNS TABLE(response boolean) LANGUAGE 'plpgsql'

DECLARE _id integer;

BEGIN
    -- guid exists and it's been 10 minutes from created_date:
    IF ((SELECT COUNT (*) FROM public.tbl_client_location WHERE guid = in_guid AND created_date < NOW() - INTERVAL '10 MINUTE') > 0) THEN
        RETURN QUERY (SELECT FALSE);
    
    -- guid exists but 10 minutes hasen't passed yet:
    ELSEIF ((SELECT COUNT (*) FROM public.tbl_client_location WHERE guid = in_guid) > 0) THEN
    
        UPDATE
            public.tbl_client_location
        SET
            x_value = in_x_value,
            y_value = in_y_value,
            updated_date = now()
        WHERE
            guid = in_guid;

        RETURN QUERY (SELECT TRUE);
    
    -- guid not exist:
    ELSE
       
        INSERT INTO public.tbl_client_location
            ( guid   , x_value   , y_value    )
        VALUES
            ( in_guid, in_x_value, in_y_value )
        RETURNING id INTO _id;  

        RETURN QUERY (SELECT TRUE);

    END IF;
END

Answer 1

這確實可以簡單得多：

CREATE OR REPLACE FUNCTION insert_table(in_guid text
                                      , in_x_value text
                                      , in_y_value text
                                      , OUT response bool)  -- ④
  -- RETURNS record  -- optional noise  -- ④
  LANGUAGE plpgsql AS  -- ①
$func$  -- ②
-- DECLARE
   -- _id integer;  -- what for?
BEGIN
   INSERT INTO tbl AS t
          (   guid,    x_value,    y_value)
   VALUES (in_guid, in_x_value, in_y_value)
   ON CONFLICT (guid) DO UPDATE  -- guid exists
   SET    (         x_value,          y_value, updated_date)
        = (EXCLUDED.x_value, EXCLUDED.y_value, now())  -- ⑤
   WHERE  t.created_date >= now() - interval '10 minutes'  -- ③ have not passed yet
   -- RETURNING id INTO _id  -- what for?
   ;
   response := FOUND;  -- ⑥
END
$func$;

假設guid定義為UNIQUE或PRIMARY KEY ，並且created_date定義為NOT NULL DEFAULT now() 。

① 語言名稱是一個標識符——最好不帶引號。

② 缺少 function 主體周圍的引號（無效命令）。 看：

• '$$' 在 PL/pgSQL 中用於什么

③ 僅在 10 分鍾未過時才UPDATE 。 請記住，時間戳是從相應事務開始的時間戳。 因此，請保持交易簡短明了。 看：

• now() 和 current_timestamp 的區別

④ 帶有OUT參數且沒有RETURNS子句的 function 自動返回單行（ record ）。 您的原始文件被聲明為 set-returning function （0-n 返回行），這沒有意義。 看：

• 使用 PL/pgSQL 在 PostgreSQL 中將多個字段作為記錄返回

⑤ 通常最好使用特殊的EXCLUDED行而不是再次拼出值。 看：

• 如何使這個 UPSERT 查詢更短？

⑤ 也使用短語法來更新多列。 看：

• 更新以特定字符串開頭的多個列

⑥ 要查看是否寫入了行，請使用特殊變量FOUND 。 細微的區別：與你原來的不同，你在事實之后得到true或false ，說一行實際上已經被寫入（或沒有）。 在您的原始文件中，觸發器或規則可能仍會跳過INSERT或UPDATE （不引發異常），並且 function 結果在這種情況下會產生誤導。 看：

• IS NOT NULL 對記錄的測試在設置變量時不返回 TRUE

進一步閱讀：

• Postgres ON CONFLICT ON CONSTRAINT 在錯誤日志中觸發錯誤
• 如何在 PostgreSQL 中使用 RETURNING 和 ON CONFLICT？

---

您可能只運行單個 SQL 語句，提供您的值一次：

INSERT INTO tbl AS t(guid, x_value,y_value)
VALUES ($in_guid, $in_x_value, $in_y_value)  -- your values here, once
ON CONFLICT (guid) DO UPDATE
SET    (x_value,y_value, updated_date)
     = (EXCLUDED.x_value, EXCLUDED.y_value, now())
WHERE  t.created_date >= now() - interval '10 minutes';

Answer 2

我終於解決了。 我制作了另一個 function 將被調用並檢查它是否已經存在以及時間，然后我可以毫無問題地進行 upsert。 這就是我最后所做的：

CREATE OR REPLACE FUNCTION fnc_check_table(
    in_guid character varying)
    RETURNS TABLE(response boolean) 
    LANGUAGE 'plpgsql'

    COST 100
    VOLATILE 
    ROWS 1000
AS $BODY$
BEGIN
IF EXISTS (SELECT FROM tbl WHERE guid = in_guid AND created_date < NOW() - INTERVAL '10 MINUTE' ) THEN
RETURN QUERY (SELECT FALSE);
ELSEIF EXISTS (SELECT FROM tbl WHERE guid = in_guid AND created_date > NOW() - INTERVAL '10 MINUTE') THEN
RETURN QUERY (SELECT TRUE);
ELSE
RETURN QUERY (SELECT TRUE);
END IF;

END
$BODY$;



CREATE OR REPLACE FUNCTION fnc_insert_table(
    in_guid character varying,
    in_x_value character varying,
    in_y_value character varying)
    RETURNS TABLE(response boolean) 
    LANGUAGE 'plpgsql'

    COST 100
    VOLATILE 
    ROWS 1000
AS $BODY$
BEGIN
IF (fnc_check_table(in_guid)) THEN
    INSERT INTO tbl (guid, x_value, y_value)
    VALUES (in_guid,in_x_value,in_y_value)
    ON CONFLICT (guid)
    DO UPDATE SET x_value=in_x_value, y_value=in_y_value, updated_date=now();
    RETURN QUERY (SELECT TRUE);
ELSE
    RETURN QUERY (SELECT FALSE);
END IF;
END
$BODY$;