mirror of
https://git.postgresql.org/git/postgresql.git
synced 2026-02-08 09:37:30 +08:00
5ebeb579b9
Commit 59702716 added transition table support to PL/pgsql so that SQL queries in trigger functions could access those transient tables. In order to provide the same level of support for PL/perl, PL/python and PL/tcl, refactor the relevant code into a new function SPI_register_trigger_data. Call the new function in the trigger handler of all four PLs, and document it as a public SPI function so that authors of out-of-tree PLs can do the same. Also get rid of a second QueryEnvironment object that was maintained by PL/pgsql. That was previously used to deal with cursors, but the same approach wasn't appropriate for PLs that are less tangled up with core code. Instead, have SPI_cursor_open install the connection's current QueryEnvironment, as already happens for SPI_execute_plan. While in the docs, remove the note that transition tables were only supported in C and PL/pgSQL triggers, and correct some ommissions. Thomas Munro with some work by Kevin Grittner (mostly docs)
239 lines
7.3 KiB
PL/PgSQL
239 lines
7.3 KiB
PL/PgSQL
-- suppress CONTEXT so that function OIDs aren't in output
|
|
\set VERBOSITY terse
|
|
|
|
insert into T_pkey1 values (1, 'key1-1', 'test key');
|
|
insert into T_pkey1 values (1, 'key1-2', 'test key');
|
|
insert into T_pkey1 values (1, 'key1-3', 'test key');
|
|
insert into T_pkey1 values (2, 'key2-1', 'test key');
|
|
insert into T_pkey1 values (2, 'key2-2', 'test key');
|
|
insert into T_pkey1 values (2, 'key2-3', 'test key');
|
|
|
|
insert into T_pkey2 values (1, 'key1-1', 'test key');
|
|
insert into T_pkey2 values (1, 'key1-2', 'test key');
|
|
insert into T_pkey2 values (1, 'key1-3', 'test key');
|
|
insert into T_pkey2 values (2, 'key2-1', 'test key');
|
|
insert into T_pkey2 values (2, 'key2-2', 'test key');
|
|
insert into T_pkey2 values (2, 'key2-3', 'test key');
|
|
|
|
select * from T_pkey1;
|
|
|
|
-- key2 in T_pkey2 should have upper case only
|
|
select * from T_pkey2;
|
|
|
|
insert into T_pkey1 values (1, 'KEY1-3', 'should work');
|
|
|
|
-- Due to the upper case translation in trigger this must fail
|
|
insert into T_pkey2 values (1, 'KEY1-3', 'should fail');
|
|
|
|
insert into T_dta1 values ('trec 1', 1, 'key1-1');
|
|
insert into T_dta1 values ('trec 2', 1, 'key1-2');
|
|
insert into T_dta1 values ('trec 3', 1, 'key1-3');
|
|
|
|
-- Must fail due to unknown key in T_pkey1
|
|
insert into T_dta1 values ('trec 4', 1, 'key1-4');
|
|
|
|
insert into T_dta2 values ('trec 1', 1, 'KEY1-1');
|
|
insert into T_dta2 values ('trec 2', 1, 'KEY1-2');
|
|
insert into T_dta2 values ('trec 3', 1, 'KEY1-3');
|
|
|
|
-- Must fail due to unknown key in T_pkey2
|
|
insert into T_dta2 values ('trec 4', 1, 'KEY1-4');
|
|
|
|
select * from T_dta1;
|
|
|
|
select * from T_dta2;
|
|
|
|
update T_pkey1 set key2 = 'key2-9' where key1 = 2 and key2 = 'key2-1';
|
|
update T_pkey1 set key2 = 'key1-9' where key1 = 1 and key2 = 'key1-1';
|
|
delete from T_pkey1 where key1 = 2 and key2 = 'key2-2';
|
|
delete from T_pkey1 where key1 = 1 and key2 = 'key1-2';
|
|
|
|
update T_pkey2 set key2 = 'KEY2-9' where key1 = 2 and key2 = 'KEY2-1';
|
|
update T_pkey2 set key2 = 'KEY1-9' where key1 = 1 and key2 = 'KEY1-1';
|
|
delete from T_pkey2 where key1 = 2 and key2 = 'KEY2-2';
|
|
delete from T_pkey2 where key1 = 1 and key2 = 'KEY1-2';
|
|
|
|
select * from T_pkey1;
|
|
select * from T_pkey2;
|
|
select * from T_dta1;
|
|
select * from T_dta2;
|
|
|
|
select tcl_avg(key1) from T_pkey1;
|
|
select tcl_sum(key1) from T_pkey1;
|
|
select tcl_avg(key1) from T_pkey2;
|
|
select tcl_sum(key1) from T_pkey2;
|
|
|
|
-- The following should return NULL instead of 0
|
|
select tcl_avg(key1) from T_pkey1 where key1 = 99;
|
|
select tcl_sum(key1) from T_pkey1 where key1 = 99;
|
|
|
|
select 1 @< 2;
|
|
select 100 @< 4;
|
|
|
|
select * from T_pkey1 order by key1 using @<, key2 collate "C";
|
|
select * from T_pkey2 order by key1 using @<, key2 collate "C";
|
|
|
|
-- show dump of trigger data
|
|
insert into trigger_test values(1,'insert');
|
|
|
|
insert into trigger_test_view values(2,'insert');
|
|
update trigger_test_view set v = 'update' where i=1;
|
|
delete from trigger_test_view;
|
|
|
|
update trigger_test set v = 'update', test_skip=true where i = 1;
|
|
update trigger_test set v = 'update' where i = 1;
|
|
delete from trigger_test;
|
|
truncate trigger_test;
|
|
|
|
-- Test composite-type arguments
|
|
select tcl_composite_arg_ref1(row('tkey', 42, 'ref2'));
|
|
select tcl_composite_arg_ref2(row('tkey', 42, 'ref2'));
|
|
|
|
-- Test argisnull primitive
|
|
select tcl_argisnull('foo');
|
|
select tcl_argisnull('');
|
|
select tcl_argisnull(null);
|
|
-- should error
|
|
insert into trigger_test(test_argisnull) values(true);
|
|
select trigger_data();
|
|
|
|
-- Test spi_lastoid primitive
|
|
create temp table t1 (f1 int);
|
|
select tcl_lastoid('t1');
|
|
create temp table t2 (f1 int) with oids;
|
|
select tcl_lastoid('t2') > 0;
|
|
|
|
-- test some error cases
|
|
create function tcl_error(out a int, out b int) as $$return {$$ language pltcl;
|
|
select tcl_error();
|
|
|
|
create function bad_record(out a text, out b text) as $$return [list a]$$ language pltcl;
|
|
select bad_record();
|
|
|
|
create function bad_field(out a text, out b text) as $$return [list a 1 b 2 cow 3]$$ language pltcl;
|
|
select bad_field();
|
|
|
|
-- test compound return
|
|
select * from tcl_test_cube_squared(5);
|
|
|
|
-- test SRF
|
|
select * from tcl_test_squared_rows(0,5);
|
|
|
|
select * from tcl_test_sequence(0,5) as a;
|
|
|
|
select 1, tcl_test_sequence(0,5);
|
|
|
|
create function non_srf() returns int as $$return_next 1$$ language pltcl;
|
|
select non_srf();
|
|
|
|
create function bad_record_srf(out a text, out b text) returns setof record as $$
|
|
return_next [list a]
|
|
$$ language pltcl;
|
|
select bad_record_srf();
|
|
|
|
create function bad_field_srf(out a text, out b text) returns setof record as $$
|
|
return_next [list a 1 b 2 cow 3]
|
|
$$ language pltcl;
|
|
select bad_field_srf();
|
|
|
|
-- test quote
|
|
select tcl_eval('quote foo bar');
|
|
select tcl_eval('quote [format %c 39]');
|
|
select tcl_eval('quote [format %c 92]');
|
|
|
|
-- Test argisnull
|
|
select tcl_eval('argisnull');
|
|
select tcl_eval('argisnull 14');
|
|
select tcl_eval('argisnull abc');
|
|
|
|
-- Test return_null
|
|
select tcl_eval('return_null 14');
|
|
-- should error
|
|
insert into trigger_test(test_return_null) values(true);
|
|
|
|
-- Test spi_exec
|
|
select tcl_eval('spi_exec');
|
|
select tcl_eval('spi_exec -count');
|
|
select tcl_eval('spi_exec -array');
|
|
select tcl_eval('spi_exec -count abc');
|
|
select tcl_eval('spi_exec query loop body toomuch');
|
|
select tcl_eval('spi_exec "begin; rollback;"');
|
|
|
|
-- Test spi_execp
|
|
select tcl_eval('spi_execp');
|
|
select tcl_eval('spi_execp -count');
|
|
select tcl_eval('spi_execp -array');
|
|
select tcl_eval('spi_execp -count abc');
|
|
select tcl_eval('spi_execp -nulls');
|
|
select tcl_eval('spi_execp ""');
|
|
|
|
-- test spi_prepare
|
|
select tcl_eval('spi_prepare');
|
|
select tcl_eval('spi_prepare a b');
|
|
select tcl_eval('spi_prepare a "b {"');
|
|
select tcl_error_handling_test($tcl$spi_prepare "select moo" []$tcl$);
|
|
|
|
-- test full error text
|
|
select tcl_error_handling_test($tcl$
|
|
spi_exec "DO $$
|
|
BEGIN
|
|
RAISE 'my message'
|
|
USING HINT = 'my hint'
|
|
, DETAIL = 'my detail'
|
|
, SCHEMA = 'my schema'
|
|
, TABLE = 'my table'
|
|
, COLUMN = 'my column'
|
|
, CONSTRAINT = 'my constraint'
|
|
, DATATYPE = 'my datatype'
|
|
;
|
|
END$$;"
|
|
$tcl$);
|
|
|
|
-- verify tcl_error_handling_test() properly reports non-postgres errors
|
|
select tcl_error_handling_test('moo');
|
|
|
|
-- test elog
|
|
select tcl_eval('elog');
|
|
select tcl_eval('elog foo bar');
|
|
|
|
-- test forced error
|
|
select tcl_eval('error "forced error"');
|
|
|
|
-- test loop control in spi_exec[p]
|
|
select tcl_spi_exec(true, 'break');
|
|
select tcl_spi_exec(true, 'continue');
|
|
select tcl_spi_exec(true, 'error');
|
|
select tcl_spi_exec(true, 'return');
|
|
select tcl_spi_exec(false, 'break');
|
|
select tcl_spi_exec(false, 'continue');
|
|
select tcl_spi_exec(false, 'error');
|
|
select tcl_spi_exec(false, 'return');
|
|
|
|
-- forcibly run the Tcl event loop for awhile, to check that we have not
|
|
-- messed things up too badly by disabling the Tcl notifier subsystem
|
|
select tcl_eval($$
|
|
unset -nocomplain ::tcl_vwait
|
|
after 100 {set ::tcl_vwait 1}
|
|
vwait ::tcl_vwait
|
|
unset -nocomplain ::tcl_vwait$$);
|
|
|
|
-- test transition table visibility
|
|
create table transition_table_test (id int, name text);
|
|
insert into transition_table_test values (1, 'a');
|
|
create function transition_table_test_f() returns trigger language pltcl as
|
|
$$
|
|
spi_exec -array C "SELECT id, name FROM old_table" {
|
|
elog INFO "old: $C(id) -> $C(name)"
|
|
}
|
|
spi_exec -array C "SELECT id, name FROM new_table" {
|
|
elog INFO "new: $C(id) -> $C(name)"
|
|
}
|
|
return OK
|
|
$$;
|
|
CREATE TRIGGER a_t AFTER UPDATE ON transition_table_test
|
|
REFERENCING OLD TABLE AS old_table NEW TABLE AS new_table
|
|
FOR EACH STATEMENT EXECUTE PROCEDURE transition_table_test_f();
|
|
update transition_table_test set name = 'b';
|
|
drop table transition_table_test;
|
|
drop function transition_table_test_f();
|