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();
