1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
/*
* sqltriga.c - Smart SQL-logging trigger.
*
* Copyright (c) 2007 Marko Kreen, Skype Technologies OÜ
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <postgres.h>
#include <executor/spi.h>
#include <commands/trigger.h>
#include <lib/stringinfo.h>
#include "common.h"
#include "stringutil.h"
PG_FUNCTION_INFO_V1(pgq_sqltriga);
Datum pgq_sqltriga(PG_FUNCTION_ARGS);
/*
* PgQ log trigger, takes 2 arguments:
* 1. queue name to be inserted to.
*
* Queue events will be in format:
* ev_type - operation type, I/U/D/R
* ev_data - partial SQL
* ev_extra1 - table name
* ev_extra2 - optional urlencoded backup
*/
Datum pgq_sqltriga(PG_FUNCTION_ARGS)
{
TriggerData *tg;
PgqTriggerEvent ev;
bool skip = false;
/*
* Get the trigger call context
*/
if (!CALLED_AS_TRIGGER(fcinfo))
elog(ERROR, "pgq.sqltriga not called as trigger");
tg = (TriggerData *)(fcinfo->context);
if (pgq_is_logging_disabled())
goto skip_it;
/*
* Connect to the SPI manager
*/
if (SPI_connect() < 0)
elog(ERROR, "sqltriga: SPI_connect() failed");
pgq_prepare_event(&ev, tg, true, false);
skip = ev.tgargs->skip;
appendStringInfoChar(ev.field[EV_TYPE], ev.op_type);
appendStringInfoString(ev.field[EV_EXTRA1], ev.info->table_name);
/*
* create sql and insert if interesting
*/
if (pgqtriga_make_sql(&ev, ev.field[EV_DATA]))
pgq_insert_tg_event(&ev);
if (SPI_finish() < 0)
elog(ERROR, "SPI_finish failed");
/*
* After trigger ignores result,
* before trigger skips event if NULL.
*/
skip_it:
if (TRIGGER_FIRED_AFTER(tg->tg_event) || skip)
return PointerGetDatum(NULL);
else if (TRIGGER_FIRED_BY_UPDATE(tg->tg_event))
return PointerGetDatum(tg->tg_newtuple);
else
return PointerGetDatum(tg->tg_trigtuple);
}
|
