plpgsql 函数是原子的吗？

问题描述

我想知道 plpgsql 函数调用以及嵌套调用的原子性：

例如，我想将两条信息放入两个不同的表中：

CREATE OR REPLACE FUNCTION email_registration(
  i_email TEXT,i_nickname TEXT
) RETURNS JSONB AS $$
BEGIN
  DECLARE
    o_account "UserAccount";
    o_identity "UserIdentity";
  BEGIN
    INSERT INTO "UserAccount"  (nickname) VALUES (i_nickname) RETURNING * INTO o_account;
    INSERT INTO "UserIdentity" (email) VALUES (i_email) RETURNING * INTO o_identity;
    RETURN jsonb_build_object(
        'account',row_to_json(o_account),'identity',row_to_json(o_identity)
    );
  END;
END;
$$ LANGUAGE 'plpgsql'; 

调用 email_registration('hello','world') 会导致 hello 在 UserAccount 表中而 world 在 UserIdentity 表中丢失吗？

---

另外，如果两个插入被重构为函数，函数调用仍然是原子的吗？

即。

CREATE OR REPLACE FUNCTION email_registration(
  i_email TEXT,i_nickname TEXT
) RETURNS JSONB AS $$
BEGIN
  DECLARE
    o_account "UserAccount";
    o_identity "UserIdentity";
  BEGIN
    o_account := insert_email(i_email);
    o_identity := insert_nickname(i_nickname);
    RETURN jsonb_build_object(
        'account',row_to_json(o_identity)
    );
  END;
END;
$$ LANGUAGE 'plpgsql'; 

解决方法

一个函数总是在函数调用开始时处于活动状态的单个事务中运行，因此要么所有语句都成功，要么所有语句都失败。这是有保证的。

还有其他可能导致 INSERT 无效的因素，例如返回 NULL 的行级 BEFORE 触发器。