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

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);

这可能吗？我知道我可以通过连接字符串来做到这一点，但我想准备我的声明并使用绑定(bind)参数。

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)