Ниже представлен простой пример триггерной функции, написанной на языке C. (Примеры триггеров на процедурных языках можно найти в документации к этим языкам.)
Функция trigf выводит количество строк в таблице ttest и пропускает операцию, если команда пытается вставить значение NULL в столбец
x. (Таким образом, триггер действует как ограничение NOT NULL, но не прерывает транзакцию.)
Сначала приведено определение таблицы:
CREATE TABLE ttest (
x integer
);
Исходный код триггерной функции:
#include "postgres.h"
#include "fmgr.h"
#include "executor/spi.h" /* это необходимо для работы с SPI */
#include "commands/trigger.h" /* ... триггеры ... */
#include "utils/rel.h" /* ... и отношения */
PG_MODULE_MAGIC;
PG_FUNCTION_INFO_V1(trigf);
Datum
trigf(PG_FUNCTION_ARGS)
{
TriggerData *trigdata = (TriggerData *) fcinfo->context;
TupleDesc tupdesc;
HeapTuple rettuple;
char *when;
bool checknull = false;
bool isnull;
int ret, i;
/* проверка того, что функция вызвана именно триггером */
if (!CALLED_AS_TRIGGER(fcinfo))
elog(ERROR, "trigf: not called by trigger manager");
/* кортеж, возвращаемый исполнителю (executor) */
if (TRIGGER_FIRED_BY_UPDATE(trigdata->tg_event))
rettuple = trigdata->tg_newtuple;
else
rettuple = trigdata->tg_trigtuple;
/* проверка на наличие значений NULL */
if (!TRIGGER_FIRED_BY_DELETE(trigdata->tg_event)
&& TRIGGER_FIRED_BEFORE(trigdata->tg_event))
checknull = true;
if (TRIGGER_FIRED_BEFORE(trigdata->tg_event))
when = "before";
else
when = "after ";
tupdesc = trigdata->tg_relation->rd_att;
/* подключение к менеджеру SPI */
if ((ret = SPI_connect()) < 0)
elog(ERROR, "trigf (fired %s): SPI_connect вернул %d", when, ret);
/* получение количества строк в таблице */
ret = SPI_exec("SELECT count(*) FROM ttest", 0);
if (ret < 0)
elog(ERROR, "trigf (fired %s): SPI_exec вернул %d", when, ret);
/* count(*) возвращает int8, поэтому необходимо выполнить преобразование */
i = DatumGetInt64(SPI_getbinval(SPI_tuptable->vals[0],
SPI_tuptable->tupdesc,
1,
&isnull));
elog (INFO, "trigf (fired %s): there are %d rows in ttest", when, i);
SPI_finish();
if (checknull)
{
SPI_getbinval(rettuple, tupdesc, 1, &isnull);
if (isnull)
rettuple = NULL;
}
return PointerGetDatum(rettuple);
}
После компиляции исходного кода (см. Раздел 5.1.10.5), объявите функцию и триггеры:
CREATE FUNCTION trigf() RETURNS trigger
AS 'имя_файла'
LANGUAGE C;
CREATE TRIGGER tbefore BEFORE INSERT OR UPDATE OR DELETE ON ttest
FOR EACH ROW EXECUTE FUNCTION trigf();
CREATE TRIGGER tafter AFTER INSERT OR UPDATE OR DELETE ON ttest
FOR EACH ROW EXECUTE FUNCTION trigf();
Теперь можно проверить работу триггера:
=> INSERT INTO ttest VALUES (NULL);
INFO: trigf (fired before): there are 0 rows in ttest
INSERT 0 0
-- Вставка пропущена, и триггер AFTER не срабатывает
=> SELECT * FROM ttest;
x
---
(0 rows)
=> INSERT INTO ttest VALUES (1);
INFO: trigf (fired before): there are 0 rows in ttest
INFO: trigf (fired after ): there are 1 rows in ttest
^^^^^^^^
вспомните правила видимости данных.
INSERT 167793 1
vac=> SELECT * FROM ttest;
x
---
1
(1 строка)
=> INSERT INTO ttest SELECT x * 2 FROM ttest;
INFO: trigf (fired before): there are 1 rows in ttest
INFO: trigf (fired after ): there are 2 rows in ttest
^^^^^^
вспомните правила видимости данных.
INSERT 167794 1
=> SELECT * FROM ttest;
x
---
1
2
(2 строки)
=> UPDATE ttest SET x = NULL WHERE x = 2;
INFO: trigf (fired before): there are 2 rows in ttest
UPDATE 0
=> UPDATE ttest SET x = 4 WHERE x = 2;
INFO: trigf (fired before): there are 2 rows in ttest
INFO: trigf (fired after ): there are 2 rows in ttest
UPDATE 1
vac=> SELECT * FROM ttest;
x
---
1
4
(2 строки)
=> DELETE FROM ttest;
INFO: trigf (fired before): there are 2 rows in ttest
INFO: trigf (fired before): there are 1 rows in ttest
INFO: trigf (fired after ): there are 0 rows in ttest
INFO: trigf (fired after ): there are 0 rows in ttest
^^^^^^
необходимо учитывать правила видимости данных.
DELETE 2
=> SELECT * FROM ttest;
x
---
(0 строк)
Более сложные примеры приведены в
src/test/regress/regress.c и
в spi.