34.4. C 언어로 구현된 트리거 예제

여기서 C 언어로 구현한 아주 간단한 트리거 함수 하나를 소개한다. (각 프로시져별 트리거 함수 예제는 각 언어별 설명서를 참조하라.)

trigf 함수는 ttest 테이블에서 어떤 작업이 일어나면, 그 테이블의 로우 개수를 보여주는 함수다. 이때, x 칼럼의 값이 null 이 될 경우는 어떠한 작업도 하지 않도록 지정했다.

먼저, 테이블 정의는 다음과 같다:

CREATE TABLE ttest (
    x integer
);

트리거 함수 코드는 다음과 같다.

#include "postgres.h"
#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;

    /* 트리거 작업으로 호출 되었는지를 검사한다 */
    if (!CALLED_AS_TRIGGER(fcinfo))
        elog(ERROR, "trigf: not called 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);

    /* count(*) 결과는 8바이트 정수형이다. 형 변환이 필요하다 */
    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);
}

소스를 컴파일 하고, 이 트리거 함수를 등록하고, 해당 테이블에 트리거 기능을 추가한다:

CREATE FUNCTION trigf() RETURNS trigger
    AS 'filename'
    LANGUAGE C;

CREATE TRIGGER tbefore BEFORE INSERT OR UPDATE OR DELETE ON ttest 
    FOR EACH ROW EXECUTE PROCEDURE trigf();

CREATE TRIGGER tafter AFTER INSERT OR UPDATE OR DELETE ON ttest 
    FOR EACH ROW EXECUTE PROCEDURE trigf();

이제 이 트리거가 정상 작동하는지 확인해 본다:

=> INSERT INTO ttest VALUES (NULL);
INFO:  trigf (fired before): there are 0 rows in ttest
INSERT 0 0

-- 값이 null이기 때문에, BEFORE 트리거는 작동하고, 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 row)

=> 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 rows)

=> 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 rows)

=> 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 rows)

보다 완전한 트리거 함수 예제는 src/test/regress/regress.c 소스와, contrib/spi 디렉토리 안에 있다.