模式中所有表的 PL/PGSQL 动态触发器

Question

我希望通过对 updated_at 列的自动更新来自动化每个表更新。我能够使用触发器对特定表进行这项工作。但是我在任何地方都找不到的主要目标是创建一个函数，该函数动态获取模式中的所有表，并创建相同的触发器，并且只更改触发器引用的表名。对于我的生活，我无法弄清楚。

我相信这不应该像我所做的那样棘手，因为我们架构中的表将具有完全相同的列名“updated_at”。

我尝试过并认为可行的一种解决方案是将表模式转换为数组，并在每次迭代中迭代以调用/创建触发器。但是我没有大量的 psql 经验，所以我发现自己在谷歌上搜索了几个小时来解决这个小问题。

SELECT ARRAY (
            SELECT
                table_name::text
            FROM 
                information_schema.tables

            WHERE table_schema = 'public') as tables;

我也试过：

DO $$
DECLARE
    t text;
BEGIN
    FOR t IN 
        SELECT table_name FROM information_schema.columns
        WHERE column_name = 'updated_at'    
    LOOP 
        EXECUTE format('CREATE TRIGGER update_updatedAt
                        BEFORE UPDATE ON %I
                        FOR EACH ROW EXECUTE PROCEDURE updated_at()',
                        t);
    END loop;
    END;
    $$ language 'plpgsql';

程序：

CREATE OR REPLACE FUNCTION updated_at()
RETURNS TRIGGER AS $$
BEGIN
    NEW.updated_at = now();
    RETURN NEW;
END;
$$ language 'plpgsql';

Answer 1

你的DO块有效。唯一的问题是我们不能为多个触发器使用相同的触发器名称。因此，您可以为触发器名称添加 table_name 后缀/前缀。

DO $$
DECLARE
    t text;
BEGIN
    FOR t IN 
        SELECT  table_name FROM information_schema.columns
             WHERE column_name = 'updated_at'    
    LOOP 


        EXECUTE format('CREATE TRIGGER update_updatedAt_%I
                        BEFORE UPDATE ON %I
                        FOR EACH ROW EXECUTE PROCEDURE updated_at()',
                        t,t);
    END loop;
    END;
$$ language 'plpgsql';

此外，您可以添加一个检查以查看触发器是否已存在information_schema.triggers以确保安全。

IF NOT EXISTS ( SELECT 1 from information_schema.triggers 
                     where trigger_name = 'update_updatedat_'|| t)                                                                                      
  THEN