作者:瀚高PG实验室(Highgo PG Lab) 丹心明月
注:本文章主要翻译自《PostgreSQL 13.0 Documentation》第三十九章
PostgreSQL还提供了事件触发器实现第38章探讨的触发器机制。与普通的触发器不同,事件触发器针对整个数据库,且可捕获DDL事件。
事件触发器可使用具有事件触发器支持的过程语言或C编写,但不可使用纯SQL编写。
事件触发器在相关事件发生时触发。当前,支持的事件有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.1列出了事件触发器支持的所有命令。
本节介绍使用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指针。
此处为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;
以下示例,使用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();