16.4. 例

src/test/regress/regress.cとcontrib/spiにはもっと複雑な例があります。

トリガの使い方に関するとても簡単な例をここに示します。 trigf 関数は、トリガを発行した ttest リレーション内にあるタプル数を報告し、問い合わせが x に NULL 値を挿入しようとしていた場合は、その操作をとばします (つまり、この関数は、トランザクションを中断させない NOT NULL 制約のような動作をします)。

#include "executor/spi.h" /* SPI  */
#include "commands/trigger.h"   /* -"- およびトリガを使用する場合に必要です */

extern Datum trigf(PG_FUNCTION_ARGS);

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;

    /* trigdata が期待どおりのものを指していることを確認します */
    if (!CALLED_AS_TRIGGER(fcinfo))
        elog(ERROR, "trigf:not fired by trigger manager");

    /* エクゼキュータに返すタプル */
    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(INFO, "trigf (fired %s):SPI_connect returned %d", when, ret);

    /* リレーション中のタプル数を取得します */
    ret = SPI_exec("SELECT count(*) FROM ttest", 0);

    if (ret < 0)
        elog (NOTICE, "trigf (fired %s):SPI_exec returned %d", when, ret);

    /* PG7.2 では count(*) は int8 を返します。 変換に注意して下さい*/
    i = (int) DatumGetInt64(SPI_getbinval(SPI_tuptable->vals[0],
                                          SPI_tuptable->tupdesc,
                                          1,
                                          &isnull));

    elog (NOTICE, "trigf (fired %s):there are %d tuples in ttest", when, i);

    SPI_finish();

    if (checknull)
    {
        (void) SPI_getbinval(rettuple, tupdesc, 1, &isnull);
        if (isnull)
            rettuple = NULL;
    }

    return PointerGetDatum(rettuple);
}

ここでコンパイルを行い、トリガ関数を作ります。

CREATE FUNCTION trigf () RETURNS TRIGGER AS
'...path_to_so' LANGUAGE C;

CREATE TABLE ttest (x int4);

vac=> CREATE TRIGGER tbefore BEFORE INSERT OR UPDATE OR DELETE ON ttest
FOR EACH ROW EXECUTE PROCEDURE trigf();
CREATE
vac=> CREATE TRIGGER tafter AFTER INSERT OR UPDATE OR DELETE ON ttest
FOR EACH ROW EXECUTE PROCEDURE trigf();
CREATE
vac=> INSERT INTO ttest VALUES (NULL);
WARNING:  trigf (fired before):there are 0 tuples in ttest
INSERT 0 0

-- 挿入操作はとばされ、また、AFTERトリガも発行されません。

vac=> SELECT * FROM ttest;
x
---
(0 rows)

vac=> INSERT INTO ttest VALUES (1);
INFO:  trigf (fired before):there are 0 tuples in ttest
INFO:  trigf (fired after ):there are 1 tuples in ttest
                                       ^^^^^^^^
                            可視性の説明を思い出して下さい。
INSERT 167793 1
vac=> SELECT * FROM ttest;
x
---
 1
(1 row)

vac=> INSERT INTO ttest SELECT x * 2 FROM ttest;
INFO:  trigf (fired before):there are 1 tuples in ttest
INFO:  trigf (fired after ):there are 2 tuples in ttest
                                       ^^^^^^^^
                            可視性の説明を思い出して下さい。
INSERT 167794 1
vac=> SELECT * FROM ttest;
x
---
 1
 2
(2 rows)

vac=> UPDATE ttest SET x = NULL WHERE x = 2;
INFO:  trigf (fired before):there are 2 tuples in ttest
UPDATE 0
vac=> UPDATE ttest SET x = 4 WHERE x = 2;
INFO:  trigf (fired before):there are 2 tuples in ttest
INFO:  trigf (fired after ):there are 2 tuples in ttest
UPDATE 1
vac=> SELECT * FROM ttest;
x
---
 1
 4
(2 rows)

vac=> DELETE FROM ttest;
INFO:  trigf (fired before):there are 2 tuples in ttest
INFO:  trigf (fired after ):there are 1 tuples in ttest
INFO:  trigf (fired before):there are 1 tuples in ttest
INFO:  trigf (fired after ):there are 0 tuples in ttest
                                       ^^^^^^^^
                            可視性の説明を思い出して下さい。
DELETE 2
vac=> SELECT * FROM ttest;
x
---
(0 rows)