阅读(1589) (9)

PostgreSQL 用 C 编写触发器函数

2021-08-21 11:05:27 更新

这一节描述了一个触发器函数的接口的低层细节。只有用 C 编写触发器函数时才需要这些信息。如果你使用一种更高层的语言,那么这些细节就不需要你来处理。在大部分情况下,你应该优先考虑使用一种过程语言。每一种过程语言的文档阐述了如何使用那种语言编写一个触发器。

触发器函数必须使用版本 1函数管理器接口。

当一个函数被触发器管理器调用时,不会给它传递任何常规的参数,但是会有一个context指针传递给它,该指针指向一个TriggerData结构。C 函数可以通过执行一个宏来检查它们是否是从触发器管理器被调用:

CALLED_AS_TRIGGER(fcinfo)

它会展开成为:

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

如果这返回真,那么将fcinfo->context造型成类型TriggerData *并且利用所指向的TriggerData结构就是安全的。该函数不能修改该TriggerData结构或者它指向的任何数据。

struct TriggerData被定义在commands/trigger.h中:

typedef struct TriggerData
{
    NodeTag          type;
    TriggerEvent     tg_event;
    Relation         tg_relation;
    HeapTuple        tg_trigtuple;
    HeapTuple        tg_newtuple;
    Trigger         *tg_trigger;
    TupleTableSlot  *tg_trigslot;
    TupleTableSlot  *tg_newslot;
    Tuplestorestate *tg_oldtable;
    Tuplestorestate *tg_newtable;
    const Bitmapset *tg_updatedcols;
} TriggerData;

其中的成员被定义如下:

type

总是T_TriggerData.

tg_event

描述该函数是为什么事件被调用的。你可以使用下列宏来检查tg_event

TRIGGER_FIRED_BEFORE(tg_event)

如果该触发器在操作前被引发则返回真。

TRIGGER_FIRED_AFTER(tg_event)

如果该触发器在操作后被引发则返回真。

TRIGGER_FIRED_INSTEAD(tg_event)

如果该触发器被引发替代操作则返回真。

TRIGGER_FIRED_FOR_ROW(tg_event)

如果该触发器为一个行级事件而引发则返回真。

TRIGGER_FIRED_FOR_STATEMENT(tg_event)

如果该触发器为一个语句级事件而引发则返回真。

TRIGGER_FIRED_BY_INSERT(tg_event)

如果该触发器由一个INSERT命令引发则返回真。

TRIGGER_FIRED_BY_UPDATE(tg_event)

如果该触发器由一个UPDATE命令引发则返回真。

TRIGGER_FIRED_BY_DELETE(tg_event)

如果该触发器由一个DELETE命令引发则返回真。

TRIGGER_FIRED_BY_TRUNCATE(tg_event)

如果该触发器由一个TRUNCATE命令引发则返回真。

tg_relation

一个结构指针,该结构描述该触发器为其引发的关系。关于这个结构的细节请参考utils/rel.h。最有趣的东西是tg_relation->rd_att(该关系元组的描述符)和tg_relation->rd_rel->relname(关系名称,该类型不是char*而是 NameData。如果你需要该名称的一个拷贝,可使用SPI_getrelname(tg_relation)来得到一个char*)。

tg_trigtuple

一个该触发器为其引发的行的指针。这是被插入、更新或删除的行。如果这个触发器是为一个INSERTDELETE而引发,在你不想把该行替换成另一行(在INSERT的情况中)或不想跳过该操作时你应该从该函数中返回它。 对于外部表上的触发器,此中的系统列值未被指定。

tg_newtuple

如果该触发器为一个UPDATE而引发,则是一个指向该行新版本的指针。如果是为一个INSERTDELETE而引发,则是NULL。如果事件是一个UPDATE并且你并不想用一个不同的行替换这个行或者不想跳过该操作时,你必须从函数中返回它。对于外部表上的触发器,此中的系统列值未被指定。

tg_trigger

一个指向类型为Trigger的结构的指针,定义在utils/reltrigger.h中:

typedef struct Trigger
{
    Oid         tgoid;
    char       *tgname;
    Oid         tgfoid;
    int16       tgtype;
    char        tgenabled;
    bool        tgisinternal;
    Oid         tgconstrrelid;
    Oid         tgconstrindid;
    Oid         tgconstraint;
    bool        tgdeferrable;
    bool        tginitdeferred;
    int16       tgnargs;
    int16       tgnattr;
    int16      *tgattr;
    char      **tgargs;
    char       *tgqual;
    char       *tgoldtable;
    char       *tgnewtable;
} Trigger;

其中tgname是该触发器的名称,tgnargstgargs中参数的数量,而tgargs是一个指向CREATE TRIGGER语句中指定的参数的指针数组。其他成员只用于内部用途。

tg_trigtuplebuf

包含tg_trigtuple的插槽。或者一个NULL指针,如果没有这样的元组的话。

tg_newtuplebuf

包含tg_trigtuple的插槽。或者一个NULL指针,如果没有这样的元组的话。

tg_oldtable

一个指向Tuplestorestate类型的结构的指针,该结构包含格式由tg_relation指定的零行或者多行。如果没有OLD TABLE传递关系,则为NULL指针。

tg_newtable

一个指向Tuplestorestate类型的结构的指针,该结构包含格式由tg_relation指定的零行或者多行。如果没有NEW TABLE传递关系,则为NULL指针。

tg_updatedcols

对于 UPDATE 触发器,位图集指示由触发命令更新的列。 通用触发器函数可以使用它来优化操作,而不必处理未更改的列。

例如,要确定属性编号为 attnum(基于 1)的列是否是此位图集的成员,请调用 bms_is_member(attnum - FirstLowInvalidHeapAttributeNumber, trigdata->tg_updatedcols))

对于UPDATE触发器以外的触发器,这将是NULL

为了允许通过SPI发出的查询引用传递表,请参考SPI_register_trigger_data

一个触发器函数必须返回一个HeapTuple指针或一个NULL指针(不是一个 SQL 空值,也就是不会设置isNull为真)。如果你不希望修改正在被操作的行,要小心地根据情况返回tg_trigtupletg_newtuple