带绑定参数的用户定义函数

Question

我正在使用Postgres upsert示例。我可以让它像示例中所示那样工作，但我需要使函数调用变为动态。该功能是

CREATE TABLE db (a INT PRIMARY KEY, b TEXT); 

CREATE FUNCTION merge_db(key INT, data TEXT) RETURNS VOID AS 
$$ 
BEGIN 
LOOP 
    UPDATE db SET b = data WHERE a = key; 
    IF found THEN 
     RETURN; 
    END IF; 
    BEGIN 
     INSERT INTO db(a,b) VALUES (key, data); 
     RETURN; 
    EXCEPTION WHEN unique_violation THEN 
    END; 
END LOOP; 
END; 
$$ 
LANGUAGE plpgsql; 

我能得到它的工作是这样的：

SELECT merge_db(1, 'david'); 
SELECT merge_db(1, 'dennis'); 

但我想这样做：

SELECT merge_db($1,$2); 

这可能吗？我知道我可以通过串联字符串来实现这一点，但我想准备我的声明并使用绑定参数。

Answer 1

不知道为什么我没有想到之前尝试这一点，但这里的答案：

CREATE TABLE db (a INT PRIMARY KEY, b TEXT); 

CREATE FUNCTION merge_db(INT, TEXT) RETURNS VOID AS 
$$ 
BEGIN 
LOOP 
    UPDATE db SET b = $2 WHERE a = $1; 
    IF found THEN 
     RETURN; 
    END IF; 
    BEGIN 
     INSERT INTO db(a,b) VALUES ($1, $2); 
     RETURN; 
    EXCEPTION WHEN unique_violation THEN 
    END; 
END LOOP; 
END; 
$$ 
LANGUAGE plpgsql; 

这样，您可以使用

SELECT merge_db($1, $2)