PostgreSQL中的事件触发器

上山打老虎

　　作者：瀚高PG实验室（Highgo PG Lab） 丹心明月

注：本文章主要翻译自《PostgreSQL 13.0 Documentation》第三十九章

PostgreSQL还提供了事件触发器实现第38章探讨的触发器机制。与普通的触发器不同，事件触发器针对整个数据库，且可捕获DDL事件。
事件触发器可使用具有事件触发器支持的过程语言或C编写，但不可使用纯SQL编写。39.1 事件触发器概览事件触发器在相关事件发生时触发。当前，支持的事件有ddl_command_start，ddl_command_end，table_rewrite和sql_drop。
ddl_command_start事件在执行CREATE，ALTER，DROP，SECURITY LABEL，COMMENT，GRANT或REVOKE命令前发生。在事件触发器触发之前，不会检查所影响的对象是否存在。不过，针对共享对象（如数据库，角色和表空间）或对事件触发器本身的DDL命令不会触发该事件。事件触发器机制不支持此类对象类型。因为SELECT INTO命令等价于CREATE TABLE AS，故也会触发ddl_command_start事件。
ddl_comand_end事件在相同DDL命令之后触发。触发器在操作发生后（事务提交前）触发。
sql_drop在删除数据库对象时触发（在ddl_comand_end之前）。可使用pg_event_trigger_dropped_objects()列出删除的数据库对象。该触发器是在将对象从系统视图删除后执行的。
table_rewrite事件在表被ALTER TABLE和ALTER TYPE重写前触发。
事件触发器支持的命令列表，请参见第39.2节。
事件触发器使用命令CREATE EVENT TRIGGER命令创建。创建事件触发器之前，需要先创建返回event_trigger类型的触发器函数。
触发器定义中，也可使用WHEN指定触发的特定命令。39.2 事件触发器触发矩阵表39.1列出了事件触发器支持的所有命令。 39.3 使用C编写事件触发器函数本节介绍使用C编写事件触发器函数相关信息。
事件触发器函数必须使用“version  1”函数管理接口。
事件触发器管理者调用函数时，不会传递普通参数，而是一个指向EventTriggerData架构的上下文指针。C函数可使用如下宏检查是否由时间触发器管理者调用：

CALLED_AS_EVENT_TRIGGER(fcinfo)

扩展为：

((fcinfo)->context != NULL && IsA((fcinfo)->context, EventTriggerData))

如果返回true，则可将fcinfo->context转换为EventTriggerData 并使用EventTriggerData结构。函数不可对EventTriggerData结构或其指向的任何数据进行修改。
struct EventTriggerData在commands/event_trigger.h中定义：

typedef struct EventTriggerData{
       NodeTag type;       const char event; /* event name /
       Node parsetree; /* parse tree /
       CommandTag tag; / command tag */} EventTriggerData;

参数解释：type    T_EventTriggerData。event    触发事件。parsetree    指向命令解析树的指针。tag    命令标签。
事件触发器仅可返回NULL指针。39.4 事件触发器示例此处为C编写的事件触发器函数示例。
函数noddl每次调用时均返回异常。触发事件为ddl_command_start。其作用为禁止运行所有的DDL命令。
以下为触发器函数的源码：

#include "postgres.h"#include "commands/event_trigger.h"PG_MODULE_MAGIC;
PG_FUNCTION_INFO_V1(noddl);Datumnoddl(PG_FUNCTION_ARGS){
       EventTriggerData trigdata;       if (!CALLED_AS_EVENT_TRIGGER(fcinfo)) / internal error */
              elog(ERROR, "not fired by event trigger manager");
       trigdata = (EventTriggerData*)fcinfo->context;
       ereport(ERROR,
              (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                      errmsg("command "%s" denied", trigdata->tag)));
       PG_RETURN_NULL();
}

编译源码后，声明函数和触发器：

CREATE FUNCTION noddl() RETURNS event_triggerAS 'noddl' LANGUAGE C;CREATE EVENT TRIGGER noddl ON ddl_command_startEXECUTE FUNCTION noddl();

测试：

=# \dy
List of event triggers
Name | Event | Owner | Enabled | Function | Tags
-------+-------------------+-------+---------+----------+
noddl | ddl_command_start | dim | enabled | noddl |
(1 row)
=# CREATE TABLE foo(id serial);
ERROR: command "CREATE TABLE" denied

这种情形下，如果需要执行DDL，要么删掉该触发器，要么暂时禁用该触发器。可仅在事务期间禁用该触发器：

BEGIN;ALTER EVENT TRIGGER noddl DISABLE;CREATE TABLE foo (id serial);ALTER EVENT TRIGGER noddl ENABLE;COMMIT;

39.5 表重写事件触发器示例以下示例，使用table_event事件，实现仅在维护期间可进行表重写：

CREATE OR REPLACE FUNCTION no_rewrite()
RETURNS event_trigger
LANGUAGE plpgsql AS
$$
--- Implement local Table Rewriting policy:
--- public.foo is not allowed rewriting, ever
--- other tables are only allowed rewriting between 1am and 6am
--- unless they have more than 100 blocks
DECLARE
table_oid oid := pg_event_trigger_table_rewrite_oid();
current_hour integer := extract('hour' from current_time);
pages integer;
max_pages integer := 100;
BEGIN
IF pg_event_trigger_table_rewrite_oid() = 'public.foo'::regclass
THEN
RAISE EXCEPTION 'you''re not allowed to rewrite the table
%',
table_oid::regclass;
END IF;
SELECT INTO pages relpages FROM pg_class WHERE oid = table_oid;
IF pages > max_pages
THEN
RAISE EXCEPTION 'rewrites only allowed for table with less
than % pages',
max_pages;
END IF;
IF current_hour NOT BETWEEN 1 AND 6
THEN
RAISE EXCEPTION 'rewrites only allowed between 1am and
6am';
END IF;
END;
$$;
CREATE EVENT TRIGGER no_rewrite_allowed
ON table_rewrite
EXECUTE FUNCTION no_rewrite();