42.6. Триггерные функции на PL/Tcl
На PL/Tcl можно написать триггерные функции. Postgres Pro требует, чтобы функция, которая будет вызываться как триггерная, была объявлена как функция без аргументов и возвращала тип trigger
.
Информация от менеджера триггеров передаётся в тело функции в следующих переменных:
$TG_name
Имя триггера из оператора
CREATE TRIGGER
.$TG_relid
Идентификатор объекта таблицы, для которой будет вызываться триггерная функция.
$TG_table_name
Имя таблицы, для которой будет вызываться триггерная функция.
$TG_table_schema
Схема таблицы, для которой будет вызываться триггерная функция.
$TG_relatts
Список языка Tcl, содержащий имена столбцов таблицы. В начало списка добавлен пустой элемент, поэтому при поиске в этом списке имени столбца с помощью стандартной в Tcl команды
lsearch
будет возвращён номер элемента, начиная с 1, так же, как нумеруются столбцы в Postgres Pro. (В позициях удалённых столбцов также содержатся пустые элементы, так что нумерация следующих за ними атрибутов не нарушается.)$TG_when
Строка
BEFORE
,AFTER
илиINSTEAD OF
, в зависимости от типа события триггера.$TG_level
Строка
ROW
илиSTATEMENT
, в зависимости от уровня события триггера.$TG_op
Строка
INSERT
,UPDATE
,DELETE
илиTRUNCATE
, в зависимости от действия события триггера.$NEW
Ассоциативный массив, содержащий значения новой строки таблицы для действий
INSERT
илиUPDATE
, либо пустой массив дляDELETE
. Индексами в массиве являются имена столбцов. Столбцы со значениями NULL в нём отсутствуют. Для триггеров уровня оператора этот массив не определяется.$OLD
Ассоциативный массив, содержащий значения старой строки таблицы для действий
UPDATE
илиDELETE
, либо пустой массив дляINSERT
. Индексами в массиве являются имена столбцов. Столбцы со значениями NULL в нём отсутствуют. Для триггеров уровня оператора этот массив не определяется.$args
Список на языке Tcl аргументов функции, заданных в операторе
CREATE TRIGGER
. Эти аргументы также доступны под обозначениями$1
...$
в теле функции.n
Возвращаемым значением триггерной функции может быть строка OK
или SKIP
либо список пар имя столбца/значение. Если возвращается значение OK
, операция (INSERT
/UPDATE
/DELETE
), которая привела к срабатыванию триггера, выполняется нормально. Значение SKIP
указывает менеджеру триггеров просто пропустить эту операцию с текущей строкой данных. Если возвращается список, через него PL/Tcl передаёт менеджеру триггеров изменённую строку; содержимое изменённой строки задаётся именами и значениями столбцов в списке. Все столбцы, не перечисленные в этом списке, получают значения NULL. Возвращать изменённую строку имеет смысл только для триггеров уровня строки с порядком BEFORE
команд INSERT
и UPDATE
, в которых вместо заданной в $NEW
будет записываться изменённая строка; либо с порядком INSTEAD OF
команд INSERT
и UPDATE
, в которых возвращаемая строка служит исходными данными для предложений INSERT RETURNING
или UPDATE RETURNING
. В триггерах уровня строки с порядком BEFORE
или INSTEAD OF
команды DELETE
возврат изменённой строки воспринимается так же, как и возврат значения OK
, то есть операция выполняется. Для всех остальных типов триггеров возвращаемое значение игнорируется.
Подсказка
Список результатов можно создать из изменённого кортежа, представленного в виде массива, с помощью команды array get
языка Tcl.
Следующий небольшой пример показывает триггерную функцию, которая ведёт в таблице целочисленный счётчик числа изменений, выполненных в строке. Для новых строк счётчик инициализируется нулевым значением, а затем увеличивается на единицу при каждом изменении.
CREATE FUNCTION trigfunc_modcount() RETURNS trigger AS $$ switch $TG_op { INSERT { set NEW($1) 0 } UPDATE { set NEW($1) $OLD($1) incr NEW($1) } default { return OK } } return [array get NEW] $$ LANGUAGE pltcl; CREATE TABLE mytab (num integer, description text, modcnt integer); CREATE TRIGGER trig_mytab_modcount BEFORE INSERT OR UPDATE ON mytab FOR EACH ROW EXECUTE FUNCTION trigfunc_modcount('modcnt');
Заметьте, что сама триггерная функция не знает имени столбца; оно передаётся в аргументах триггера. Это позволяет применять эту функцию для различных таблиц.
43.3. Built-in Functions
43.3.1. Database Access from PL/Perl
Access to the database itself from your Perl function can be done via the following functions:
-
spi_exec_query
(query
[,limit
]) spi_exec_query
executes an SQL command and returns the entire row set as a reference to an array of hash references. Iflimit
is specified and is greater than zero, thenspi_exec_query
retrieves at mostlimit
rows, much as if the query included aLIMIT
clause. Omittinglimit
or specifying it as zero results in no row limit.You should only use this command when you know that the result set will be relatively small. Here is an example of a query (
SELECT
command) with the optional maximum number of rows:$rv = spi_exec_query('SELECT * FROM my_table', 5);
This returns up to 5 rows from the table
my_table
. Ifmy_table
has a columnmy_column
, you can get that value from row$i
of the result like this:$foo = $rv->{rows}[$i]->{my_column};
The total number of rows returned from a
SELECT
query can be accessed like this:$nrows = $rv->{processed}
Here is an example using a different command type:
$query = "INSERT INTO my_table VALUES (1, 'test')"; $rv = spi_exec_query($query);
You can then access the command status (e.g.,
SPI_OK_INSERT
) like this:$res = $rv->{status};
To get the number of rows affected, do:
$nrows = $rv->{processed};
Here is a complete example:
CREATE TABLE test ( i int, v varchar ); INSERT INTO test (i, v) VALUES (1, 'first line'); INSERT INTO test (i, v) VALUES (2, 'second line'); INSERT INTO test (i, v) VALUES (3, 'third line'); INSERT INTO test (i, v) VALUES (4, 'immortal'); CREATE OR REPLACE FUNCTION test_munge() RETURNS SETOF test AS $$ my $rv = spi_exec_query('select i, v from test;'); my $status = $rv->{status}; my $nrows = $rv->{processed}; foreach my $rn (0 .. $nrows - 1) { my $row = $rv->{rows}[$rn]; $row->{i} += 200 if defined($row->{i}); $row->{v} =~ tr/A-Za-z/a-zA-Z/ if (defined($row->{v})); return_next($row); } return undef; $$ LANGUAGE plperl; SELECT * FROM test_munge();
-
spi_query(
command
)
spi_fetchrow(
cursor
)
spi_cursor_close(
cursor
) spi_query
andspi_fetchrow
work together as a pair for row sets which might be large, or for cases where you wish to return rows as they arrive.spi_fetchrow
works only withspi_query
. The following example illustrates how you use them together:CREATE TYPE foo_type AS (the_num INTEGER, the_text TEXT); CREATE OR REPLACE FUNCTION lotsa_md5 (INTEGER) RETURNS SETOF foo_type AS $$ use Digest::MD5 qw(md5_hex); my $file = '/usr/share/dict/words'; my $t = localtime; elog(NOTICE, "opening file $file at $t" ); open my $fh, '<', $file # ooh, it's a file access! or elog(ERROR, "cannot open $file for reading: $!"); my @words = <$fh>; close $fh; $t = localtime; elog(NOTICE, "closed file $file at $t"); chomp(@words); my $row; my $sth = spi_query("SELECT * FROM generate_series(1,$_[0]) AS b(a)"); while (defined ($row = spi_fetchrow($sth))) { return_next({ the_num => $row->{a}, the_text => md5_hex($words[rand @words]) }); } return; $$ LANGUAGE plperlu; SELECT * from lotsa_md5(500);
Normally,
spi_fetchrow
should be repeated until it returnsundef
, indicating that there are no more rows to read. The cursor returned byspi_query
is automatically freed whenspi_fetchrow
returnsundef
. If you do not wish to read all the rows, instead callspi_cursor_close
to free the cursor. Failure to do so will result in memory leaks.-
spi_prepare(
command
,argument types
)
spi_query_prepared(
plan
,arguments
)
spi_exec_prepared(
plan
[,attributes
],arguments
)
spi_freeplan(
plan
) spi_prepare
,spi_query_prepared
,spi_exec_prepared
, andspi_freeplan
implement the same functionality but for prepared queries.spi_prepare
accepts a query string with numbered argument placeholders ($1, $2, etc) and a string list of argument types:$plan = spi_prepare('SELECT * FROM test WHERE id > $1 AND name = $2', 'INTEGER', 'TEXT');
Once a query plan is prepared by a call to
spi_prepare
, the plan can be used instead of the string query, either inspi_exec_prepared
, where the result is the same as returned byspi_exec_query
, or inspi_query_prepared
which returns a cursor exactly asspi_query
does, which can be later passed tospi_fetchrow
. The optional second parameter tospi_exec_prepared
is a hash reference of attributes; the only attribute currently supported islimit
, which sets the maximum number of rows returned from the query. Omittinglimit
or specifying it as zero results in no row limit.The advantage of prepared queries is that is it possible to use one prepared plan for more than one query execution. After the plan is not needed anymore, it can be freed with
spi_freeplan
:CREATE OR REPLACE FUNCTION init() RETURNS VOID AS $$ $_SHARED{my_plan} = spi_prepare('SELECT (now() + $1)::date AS now', 'INTERVAL'); $$ LANGUAGE plperl; CREATE OR REPLACE FUNCTION add_time( INTERVAL ) RETURNS TEXT AS $$ return spi_exec_prepared( $_SHARED{my_plan}, $_[0] )->{rows}->[0]->{now}; $$ LANGUAGE plperl; CREATE OR REPLACE FUNCTION done() RETURNS VOID AS $$ spi_freeplan( $_SHARED{my_plan}); undef $_SHARED{my_plan}; $$ LANGUAGE plperl; SELECT init(); SELECT add_time('1 day'), add_time('2 days'), add_time('3 days'); SELECT done(); add_time | add_time | add_time ------------+------------+------------ 2005-12-10 | 2005-12-11 | 2005-12-12
Note that the parameter subscript in
spi_prepare
is defined via $1, $2, $3, etc, so avoid declaring query strings in double quotes that might easily lead to hard-to-catch bugs.Another example illustrates usage of an optional parameter in
spi_exec_prepared
:CREATE TABLE hosts AS SELECT id, ('192.168.1.'||id)::inet AS address FROM generate_series(1,3) AS id; CREATE OR REPLACE FUNCTION init_hosts_query() RETURNS VOID AS $$ $_SHARED{plan} = spi_prepare('SELECT * FROM hosts WHERE address << $1', 'inet'); $$ LANGUAGE plperl; CREATE OR REPLACE FUNCTION query_hosts(inet) RETURNS SETOF hosts AS $$ return spi_exec_prepared( $_SHARED{plan}, {limit => 2}, $_[0] )->{rows}; $$ LANGUAGE plperl; CREATE OR REPLACE FUNCTION release_hosts_query() RETURNS VOID AS $$ spi_freeplan($_SHARED{plan}); undef $_SHARED{plan}; $$ LANGUAGE plperl; SELECT init_hosts_query(); SELECT query_hosts('192.168.1.0/30'); SELECT release_hosts_query(); query_hosts ----------------- (1,192.168.1.1) (2,192.168.1.2) (2 rows)
-
spi_commit()
spi_rollback()
Commit or roll back the current transaction. This can only be called in a procedure or anonymous code block (
DO
command) called from the top level. (Note that it is not possible to run the SQL commandsCOMMIT
orROLLBACK
viaspi_exec_query
or similar. It has to be done using these functions.) After a transaction is ended, a new transaction is automatically started, so there is no separate function for that.Here is an example:
CREATE PROCEDURE transaction_test1() LANGUAGE plperl AS $$ foreach my $i (0..9) { spi_exec_query("INSERT INTO test1 (a) VALUES ($i)"); if ($i % 2 == 0) { spi_commit(); } else { spi_rollback(); } } $$; CALL transaction_test1();
43.3.2. Utility Functions in PL/Perl
-
elog(
level
,msg
) Emit a log or error message. Possible levels are
DEBUG
,LOG
,INFO
,NOTICE
,WARNING
, andERROR
.ERROR
raises an error condition; if this is not trapped by the surrounding Perl code, the error propagates out to the calling query, causing the current transaction or subtransaction to be aborted. This is effectively the same as the Perldie
command. The other levels only generate messages of different priority levels. Whether messages of a particular priority are reported to the client, written to the server log, or both is controlled by the log_min_messages and client_min_messages configuration variables. See Chapter 18 for more information.-
quote_literal(
string
) Return the given string suitably quoted to be used as a string literal in an SQL statement string. Embedded single-quotes and backslashes are properly doubled. Note that
quote_literal
returns undef on undef input; if the argument might be undef,quote_nullable
is often more suitable.-
quote_nullable(
string
) Return the given string suitably quoted to be used as a string literal in an SQL statement string; or, if the argument is undef, return the unquoted string "NULL". Embedded single-quotes and backslashes are properly doubled.
-
quote_ident(
string
) Return the given string suitably quoted to be used as an identifier in an SQL statement string. Quotes are added only if necessary (i.e., if the string contains non-identifier characters or would be case-folded). Embedded quotes are properly doubled.
-
decode_bytea(
string
) Return the unescaped binary data represented by the contents of the given string, which should be
bytea
encoded.-
encode_bytea(
string
) Return the
bytea
encoded form of the binary data contents of the given string.-
encode_array_literal(
array
)
encode_array_literal(
array
,delimiter
) Returns the contents of the referenced array as a string in array literal format (see Section 8.15.2). Returns the argument value unaltered if it's not a reference to an array. The delimiter used between elements of the array literal defaults to "
,
" if a delimiter is not specified or is undef.-
encode_typed_literal(
value
,typename
) Converts a Perl variable to the value of the data type passed as a second argument and returns a string representation of this value. Correctly handles nested arrays and values of composite types.
-
encode_array_constructor(
array
) Returns the contents of the referenced array as a string in array constructor format (see Section 4.2.12). Individual values are quoted using
quote_nullable
. Returns the argument value, quoted usingquote_nullable
, if it's not a reference to an array.-
looks_like_number(
string
) Returns a true value if the content of the given string looks like a number, according to Perl, returns false otherwise. Returns undef if the argument is undef. Leading and trailing space is ignored.
Inf
andInfinity
are regarded as numbers.-
is_array_ref(
argument
) Returns a true value if the given argument may be treated as an array reference, that is, if ref of the argument is
ARRAY
orPostgreSQL::InServer::ARRAY
. Returns false otherwise.