sql - 具有绑定参数的用户定义函数

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)