在触发函数中，如何获取正在更新的字段

Question

这可能吗？我有兴趣找出在 UPDATE 请求中指定了哪些列，而不管正在发送的新值可能是也可能不是已经存储在数据库中的值。

我想这样做的原因是因为我们有一个可以接收来自多个来源的更新的表。以前，我们没有记录更新的来源。现在该表存储了哪个源执行了最近的更新。我们可以更改一些来源以发送标识符，但这并不是所有事情的选择。所以我希望能够识别UPDATE 请求何时没有标识符，以便我可以替换为默认值。

Answer 1

另一种方法是利用 PostgreSQL 最新版本中的 JSON/JSONB 函数。它的优点是可以处理任何可以转换为 JSON 对象（行或任何其他结构化数据）的东西，而且您甚至不需要知道记录类型。

要找出任何两行/记录之间的差异，您可以使用这个小技巧：

SELECT pre.key AS columname, pre.value AS prevalue, post.value AS postvalue
FROM jsonb_each(to_jsonb(OLD)) AS pre
CROSS JOIN jsonb_each(to_jsonb(NEW)) AS post
WHERE pre.key = post.key AND pre.value IS DISTINCT FROM post.value

其中OLD 和NEW 是在触发器函数中发现的内置记录，分别表示更改记录的前后状态。请注意，我使用了表别名 pre 和 post 而不是 old 和 new 以避免与 OLD 和 NEW 内置对象发生冲突。还要注意使用IS DISTINCT FROM 而不是简单的!= 或<> 来适当地处理NULL 值。

当然，这也适用于任何 ROW 构造函数，例如 ROW(1,2,3,...) 或其简写 (1,2,3,...)。它也适用于任何两个具有相同键的 JSONB 对象。

例如，考虑一个有两行的示例（为了示例的目的，已经转换为 JSONB）：

SELECT pre.key AS columname, pre.value AS prevalue, post.value AS postvalue
FROM jsonb_each('{"col1": "same", "col2": "prediff", "col3": 1, "col4": false}') AS pre
CROSS JOIN jsonb_each('{"col1": "same", "col2": "postdiff", "col3": 1, "col4": true}') AS post
WHERE pre.key = post.key AND pre.value IS DISTINCT FROM post.value

查询将显示已更改值的列：

 columname | prevalue  | postvalue
-----------+-----------+------------
 col2      | "prediff" | "postdiff"
 col4      | false     | true

这种方法很酷的一点是，按列过滤很简单。例如，假设您只想检测列col1 和col2 中的变化：

SELECT pre.key AS columname, pre.value AS prevalue, post.value AS postvalue
FROM jsonb_each('{"col1": "same", "col2": "prediff", "col3": 1, "col4": false}') AS pre
CROSS JOIN jsonb_each('{"col1": "same", "col2": "postdiff", "col3": 1, "col4": true}') AS post
WHERE pre.key = post.key AND pre.value IS DISTINCT FROM post.value
AND pre.key IN ('col1', 'col2')

即使 col3 的值已更改，新结果也会从结果中排除：

 columname | prevalue  | postvalue
-----------+-----------+------------
 col2      | "prediff" | "postdiff"

很容易看出这种方法可以如何以多种方式进行扩展。例如，假设您想在某些列被更新时抛出异常。您可以通过通用触发功能来实现这一点，即可以应用于任何/所有表格的触发功能，而无需知道表格类型：

CREATE OR REPLACE FUNCTION yourschema.yourtriggerfunction()
RETURNS TRIGGER AS
$$
DECLARE
    immutable_cols TEXT[] := ARRAY['createdon', 'createdby'];
BEGIN

    IF TG_OP = 'UPDATE' AND EXISTS(
        SELECT 1
        FROM jsonb_each(to_jsonb(OLD)) AS pre, jsonb_each(to_jsonb(NEW)) AS post
        WHERE pre.key = post.key AND pre.value IS DISTINCT FROM post.value
        AND pre.key = ANY(immutable_cols)
    ) THEN
        RAISE EXCEPTION 'Error 12345 updating table %.%. Cannot alter these immutable cols: %.',
            TG_TABLE_SCHEMA, TG_TABLE_NAME, immutable_cols;
    END IF;

END
$$
LANGUAGE plpgsql VOLATILE

然后，您可以将上述触发器函数注册到您想要通过以下方式控制的任何和所有表中：

CREATE TRIGGER yourtiggername
BEFORE UPDATE ON yourschema.yourtable
FOR EACH ROW EXECUTE PROCEDURE yourschema.yourtriggerfunction();

Answer 2

如果“来源”没有“发送标识符”，则该列将保持不变。然后您就无法检测到当前UPDATE 是由与上一个相同的源完成的，还是由根本没有更改列的源完成的。换句话说：这不能正常工作。

如果任何session information function 都可以识别“来源”，则可以使用它。喜欢：

NEW.column = session_user;

每次更新都是无条件的。

一般解决方案

我找到了解决原始问题的方法。该列将在 any 更新中设置为默认值，其中列未更新（不在UPDATE 的SET 列表中）。

关键元素是 PostgreSQL 9.0 中引入的 per-column trigger - 使用 UPDATE OF column_name 子句的特定于列的触发器。

仅当列出的列中至少有一个列被触发时，触发器才会触发 提到作为UPDATE 命令的目标。

这是我发现的唯一简单的方法来区分列是否使用与旧值相同的新值更新，而不是根本没有更新。

一个可以也解析current_query()返回的文本。但这似乎很棘手且不可靠。

触发函数

我假设一列source 定义为NOT NULL。

第 1 步：将source 设置为NULL（如果未更改）：

CREATE OR REPLACE FUNCTION trg_tbl_upbef_step1()
  RETURNS trigger
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF NEW.source = OLD.source THEN
      NEW.source := NULL;      -- "impossible" value (source is NOT NULL)
   END IF;

   RETURN NEW;
END
$func$;

第 2 步：恢复为旧值。 只有在实际更新了值时才会触发触发器（见下文）：

CREATE OR REPLACE FUNCTION trg_tbl_upbef_step2()
  RETURNS trigger
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF NEW.source IS NULL THEN
      NEW.source := OLD.source;
   END IF;

   RETURN NEW;
END
$func$;

第 3 步：现在我们可以识别缺少的更新并改为设置默认值：

CREATE OR REPLACE FUNCTION trg_tbl_upbef_step3()
  RETURNS trigger
  LANGUAGE plpgsql AS
$func$
BEGIN
   IF NEW.source IS NULL THEN
      NEW.source := 'UPDATE default source';  -- optionally same as column default
   END IF;

   RETURN NEW;
END
$func$;

触发器

Step 2 的触发器按列触发！

CREATE TRIGGER upbef_step1
  BEFORE UPDATE ON tbl
  FOR EACH ROW
  EXECUTE PROCEDURE trg_tbl_upbef_step1();

CREATE TRIGGER upbef_step2
  BEFORE UPDATE OF source ON tbl             -- key element!
  FOR EACH ROW
  EXECUTE PROCEDURE trg_tbl_upbef_step2();
    
CREATE TRIGGER upbef_step3
  BEFORE UPDATE ON tbl
  FOR EACH ROW
  EXECUTE PROCEDURE trg_tbl_upbef_step3();

db小提琴here

触发器名称是相关的，因为它们按字母顺序触发（全部为BEFORE UPDATE）！

可以使用“每列触发器”之类的方法或任何其他方式来检查触发器中UPDATE 的目标列表，从而简化该过程。但我目前看不到这方面的处理方法（自 Postgres 14 起未更改）。

如果source 可以是NULL，则使用任何其他“不可能”的中间值并在触发函数1 中另外检查NULL：

IF OLD.source IS NOT DISTINCT FROM NEW.source THEN
    NEW.source := '#impossible_value#';
END IF;

相应地调整其余部分。

Answer 3

在 plpgsql 中，您可以在触发函数中执行以下操作：

IF NEW.column IS NULL THEN
  NEW.column = 'default value';
END IF;

Answer 4

我几乎很自然地获得了类似问题的另一种解决方案，因为我的表包含一个语义为“上次更新时间戳”的列（我们称之为 UPDT）。

因此，我决定一次在任何更新中包含 source 和 UPDT 的新值（或一个都不包含）。由于 UPDT 旨在在每次更新时进行更改，因此可以使用条件new.UPDT = old.UPDT 推断当前更新未指定源并替换默认源。

如果一个人的表中已经有“上次更新时间戳”列，则此解决方案将比创建三个触发器更简单。不确定在不需要时创建 UPDT 是否更好。如果更新如此频繁以至于存在时间戳相似的风险，则可以使用排序器代替时间戳。