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 params。

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)

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

1 个答案: