15

我想在会话/事务中记录用户的 id，使用SET，所以我可以稍后在触发函数中访问它，使用current_setting。基本上，我正在尝试以前发布的一张非常相似的票中的选项 n2 ，不同之处在于我使用的是 PG 10.1 。

我一直在尝试 3 种方法来设置变量：

  • SET local myvars.user_id = 4，从而在事务中本地设置它；
  • SET myvars.user_id = 4，从而在会话中设置它；
  • SELECT set_config('myvars.user_id', '4', false)，取决于最后一个参数，将是前两个选项的快捷方式。

它们都不能在触发器中使用，触发器NULL在通过 获取变量时接收current_setting。这是我设计的用于对其进行故障排除的脚本（可以很容易地与 postgres docker 映像一起使用）：

database=$POSTGRES_DB
user=$POSTGRES_USER
[ -z "$user" ] && user="postgres"

psql -v ON_ERROR_STOP=1 --username "$user" $database <<-EOSQL
    DROP TRIGGER IF EXISTS add_transition1 ON houses;
    CREATE TABLE IF NOT EXISTS houses (
        id SERIAL NOT NULL,
        name VARCHAR(80),
        created_at TIMESTAMP WITHOUT TIME ZONE DEFAULT now(),
        PRIMARY KEY(id)
    );

    CREATE TABLE IF NOT EXISTS transitions1 (
        id SERIAL NOT NULL,
        house_id INTEGER,
        user_id INTEGER,
        created_at TIMESTAMP WITHOUT TIME ZONE DEFAULT now(),
        PRIMARY KEY(id),
        FOREIGN KEY(house_id) REFERENCES houses (id) ON DELETE CASCADE

    );

    CREATE OR REPLACE FUNCTION add_transition1() RETURNS TRIGGER AS \$\$
        DECLARE
            user_id integer;
        BEGIN
            user_id := current_setting('myvars.user_id')::integer || NULL;
            INSERT INTO transitions1 (user_id, house_id) VALUES (user_id, NEW.id);
            RETURN NULL;
        END;
    \$\$ LANGUAGE plpgsql;

    CREATE TRIGGER add_transition1 AFTER INSERT OR UPDATE ON houses FOR EACH ROW EXECUTE PROCEDURE add_transition1();

    BEGIN;
    %1% SELECT current_setting('myvars.user_id');
    %2% SELECT set_config('myvars.user_id', '55', false);
    %3% SELECT current_setting('myvars.user_id');
    INSERT INTO houses (name) VALUES ('HOUSE PARTY') RETURNING houses.id;
    SELECT * from houses;
    SELECT * from transitions1;
    COMMIT;
    DROP TRIGGER IF EXISTS add_transition1 ON houses;
    DROP FUNCTION IF EXISTS add_transition1;
    DROP TABLE transitions1;
        DROP TABLE houses;
EOSQL

我得出的结论是，该函数是在不同的事务和不同的（？）会话中触发的。这是一个可以配置的东西，以便所有事情都在同一个上下文中发生吗？

4

3 回答 3

7

目前尚不清楚您为什么要尝试连接NULL，user_id但这显然是问题的原因。摆脱它：

CREATE OR REPLACE FUNCTION add_transition1() RETURNS TRIGGER AS $$
    DECLARE
        user_id integer;
    BEGIN
        user_id := current_setting('myvars.user_id')::integer;
        INSERT INTO transitions1 (user_id, house_id) VALUES (user_id, NEW.id);
        RETURN NULL;
    END;
$$ LANGUAGE plpgsql;

注意

SELECT 55 || NULL

总是给NULL。

于 2018-08-22T14:00:03.873 回答
7

正确处理自定义选项的所有可能情况：

  1. 选项尚未设置

    所有对它的引用都会引发异常，包括current_setting()除非使用第二个参数调用missing_ok。手册：

    如果没有命名设置setting_name，则current_setting抛出错误，除非missing_ok提供和 is true。

  2. 选项设置为有效的整数文字

  3. 选项设置为无效的整数文字

  4. 选项重置（烧毁为3.的特殊情况）

    例如，如果您使用SET LOCAL或设置自定义选项set_config('myvars.user_id3', '55', true)，则选项值会在交易结束时重置。它仍然存在，可以被引用，但它现在返回一个空字符串 ( '') - 不能转换为integer.

除了演示中的明显错误之外，您需要为所有 4 种情况做好准备。所以：

CREATE OR REPLACE FUNCTION add_transition1()
  RETURNS trigger AS
$func$
DECLARE
   _user_id text := current_setting('myvars.user_id', true);  -- see 1.
BEGIN
   IF _user_id ~ '^\d+$' THEN  -- one or more digits?

      INSERT INTO transitions1 (user_id, house_id)
      VALUES (_user_id::int, NEW.id);  -- valid int, cast is safe

   ELSE

      INSERT INTO transitions1 (user_id, house_id)
      VALUES (NULL, NEW.id);           -- use NULL instead

      RAISE WARNING 'Invalid user_id % for house_id % was reset to NULL!'
                  , quote_literal(_user_id), NEW.id;  -- optional
   END IF;

   RETURN NULL;  -- OK for AFTER trigger
END
$func$  LANGUAGE plpgsql;

db<>在这里摆弄

笔记：

  • 避免与列名匹配的变量名。非常容易出错。一种流行的命名约定是在变量名前加上下划线：_user_id.

  • 在声明时分配以保存一项分配。注意数据类型text。我们将在整理出无效输入后进行转换。

  • 尽可能避免引发/捕获异常。手册：

    包含EXCEPTION子句的块的进入和退出比没有子句的块要昂贵得多。因此，不要在 EXCEPTION没有必要的情况下使用。

  • 测试有效的整数字符串。这个简单的正则表达式只允许数字（没有前导符号，没有空格）_user_id ~ '^\d+$'：. 对于任何无效输入，我都会重置为 NULL。适应您的需求。

  • WARNING为了您的调试方便，我添加了一个可选选项。

  • 仅由于自定义选项是字符串文字（类型）而出现的情况3.，无法自动强制执行有效的数据类型。4.text

有关的：

除此之外，根据您的确切要求，可能会有更优雅的解决方案来解决您在没有定制选项的情况下尝试做的事情。也许是这样：

于 2018-08-28T08:23:53.843 回答
5

当值不存在时，您可以捕获异常 - 这是我为使其正常工作所做的更改：

CREATE OR REPLACE FUNCTION add_transition1() RETURNS TRIGGER AS $$
    DECLARE
        user_id integer;
    BEGIN
        BEGIN
            user_id := current_setting('myvars.user_id')::integer;
        EXCEPTION WHEN OTHERS THEN
            user_id := 0;
        END;

        INSERT INTO transitions1 (user_id, house_id) VALUES (user_id, NEW.id);
        RETURN NULL;
    END;
$$ LANGUAGE plpgsql;

 CREATE OR REPLACE FUNCTION insert_house() RETURNS void as $$
 DECLARE
    user_id integer;
 BEGIN 
   PERFORM set_config('myvars.user_id', '55', false);

   INSERT INTO houses (name) VALUES ('HOUSE PARTY');
 END; $$ LANGUAGE plpgsql;
于 2018-08-28T00:11:55.240 回答