mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-08-18 17:33:24 +02:00
To ensure that "make installcheck" can be used safely against an existing installation, we need to be careful about what global object names (database, role, and tablespace names) we use; otherwise we might accidentally clobber important objects. There's been a weak consensus that test databases should have names including "regression", and that test role names should start with "regress_", but we didn't have any particular rule about tablespace names; and neither of the other rules was followed with any consistency either. This commit moves us a long way towards having a hard-and-fast rule that regression test databases must have names including "regression", and that test role and tablespace names must start with "regress_". It's not completely there because I did not touch some test cases in rolenames.sql that test creation of special role names like "session_user". That will require some rethinking of exactly what we want to test, whereas the intent of this patch is just to hit all the cases in which the needed renamings are cosmetic. There is no enforcement mechanism in this patch either, but if we don't add one we can expect that the tests will soon be violating the convention again. Again, that's not such a cosmetic change and it will require discussion. (But I did use a quick-hack enforcement patch to find these cases.) Discussion: <16638.1468620817@sss.pgh.pa.us>
1129 lines
29 KiB
Plaintext
1129 lines
29 KiB
Plaintext
CREATE EXTENSION dblink;
|
|
-- want context for notices
|
|
\set SHOW_CONTEXT always
|
|
CREATE TABLE foo(f1 int, f2 text, f3 text[], primary key (f1,f2));
|
|
INSERT INTO foo VALUES (0,'a','{"a0","b0","c0"}');
|
|
INSERT INTO foo VALUES (1,'b','{"a1","b1","c1"}');
|
|
INSERT INTO foo VALUES (2,'c','{"a2","b2","c2"}');
|
|
INSERT INTO foo VALUES (3,'d','{"a3","b3","c3"}');
|
|
INSERT INTO foo VALUES (4,'e','{"a4","b4","c4"}');
|
|
INSERT INTO foo VALUES (5,'f','{"a5","b5","c5"}');
|
|
INSERT INTO foo VALUES (6,'g','{"a6","b6","c6"}');
|
|
INSERT INTO foo VALUES (7,'h','{"a7","b7","c7"}');
|
|
INSERT INTO foo VALUES (8,'i','{"a8","b8","c8"}');
|
|
INSERT INTO foo VALUES (9,'j','{"a9","b9","c9"}');
|
|
-- misc utilities
|
|
-- list the primary key fields
|
|
SELECT *
|
|
FROM dblink_get_pkey('foo');
|
|
position | colname
|
|
----------+---------
|
|
1 | f1
|
|
2 | f2
|
|
(2 rows)
|
|
|
|
-- build an insert statement based on a local tuple,
|
|
-- replacing the primary key values with new ones
|
|
SELECT dblink_build_sql_insert('foo','1 2',2,'{"0", "a"}','{"99", "xyz"}');
|
|
dblink_build_sql_insert
|
|
-----------------------------------------------------------
|
|
INSERT INTO foo(f1,f2,f3) VALUES('99','xyz','{a0,b0,c0}')
|
|
(1 row)
|
|
|
|
-- too many pk fields, should fail
|
|
SELECT dblink_build_sql_insert('foo','1 2 3 4',4,'{"0", "a", "{a0,b0,c0}"}','{"99", "xyz", "{za0,zb0,zc0}"}');
|
|
ERROR: invalid attribute number 4
|
|
-- build an update statement based on a local tuple,
|
|
-- replacing the primary key values with new ones
|
|
SELECT dblink_build_sql_update('foo','1 2',2,'{"0", "a"}','{"99", "xyz"}');
|
|
dblink_build_sql_update
|
|
----------------------------------------------------------------------------------------
|
|
UPDATE foo SET f1 = '99', f2 = 'xyz', f3 = '{a0,b0,c0}' WHERE f1 = '99' AND f2 = 'xyz'
|
|
(1 row)
|
|
|
|
-- too many pk fields, should fail
|
|
SELECT dblink_build_sql_update('foo','1 2 3 4',4,'{"0", "a", "{a0,b0,c0}"}','{"99", "xyz", "{za0,zb0,zc0}"}');
|
|
ERROR: invalid attribute number 4
|
|
-- build a delete statement based on a local tuple,
|
|
SELECT dblink_build_sql_delete('foo','1 2',2,'{"0", "a"}');
|
|
dblink_build_sql_delete
|
|
---------------------------------------------
|
|
DELETE FROM foo WHERE f1 = '0' AND f2 = 'a'
|
|
(1 row)
|
|
|
|
-- too many pk fields, should fail
|
|
SELECT dblink_build_sql_delete('foo','1 2 3 4',4,'{"0", "a", "{a0,b0,c0}"}');
|
|
ERROR: invalid attribute number 4
|
|
-- retest using a quoted and schema qualified table
|
|
CREATE SCHEMA "MySchema";
|
|
CREATE TABLE "MySchema"."Foo"(f1 int, f2 text, f3 text[], primary key (f1,f2));
|
|
INSERT INTO "MySchema"."Foo" VALUES (0,'a','{"a0","b0","c0"}');
|
|
-- list the primary key fields
|
|
SELECT *
|
|
FROM dblink_get_pkey('"MySchema"."Foo"');
|
|
position | colname
|
|
----------+---------
|
|
1 | f1
|
|
2 | f2
|
|
(2 rows)
|
|
|
|
-- build an insert statement based on a local tuple,
|
|
-- replacing the primary key values with new ones
|
|
SELECT dblink_build_sql_insert('"MySchema"."Foo"','1 2',2,'{"0", "a"}','{"99", "xyz"}');
|
|
dblink_build_sql_insert
|
|
------------------------------------------------------------------------
|
|
INSERT INTO "MySchema"."Foo"(f1,f2,f3) VALUES('99','xyz','{a0,b0,c0}')
|
|
(1 row)
|
|
|
|
-- build an update statement based on a local tuple,
|
|
-- replacing the primary key values with new ones
|
|
SELECT dblink_build_sql_update('"MySchema"."Foo"','1 2',2,'{"0", "a"}','{"99", "xyz"}');
|
|
dblink_build_sql_update
|
|
-----------------------------------------------------------------------------------------------------
|
|
UPDATE "MySchema"."Foo" SET f1 = '99', f2 = 'xyz', f3 = '{a0,b0,c0}' WHERE f1 = '99' AND f2 = 'xyz'
|
|
(1 row)
|
|
|
|
-- build a delete statement based on a local tuple,
|
|
SELECT dblink_build_sql_delete('"MySchema"."Foo"','1 2',2,'{"0", "a"}');
|
|
dblink_build_sql_delete
|
|
----------------------------------------------------------
|
|
DELETE FROM "MySchema"."Foo" WHERE f1 = '0' AND f2 = 'a'
|
|
(1 row)
|
|
|
|
CREATE FUNCTION connection_parameters() RETURNS text LANGUAGE SQL AS $f$
|
|
SELECT $$dbname='$$||current_database()||$$' port=$$||current_setting('port');
|
|
$f$;
|
|
-- regular old dblink
|
|
SELECT *
|
|
FROM dblink(connection_parameters(),'SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
a | b | c
|
|
---+---+------------
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
(2 rows)
|
|
|
|
-- should generate "connection not available" error
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
ERROR: connection not available
|
|
-- The first-level connection's backend will crash on exit given OpenLDAP
|
|
-- [2.4.24, 2.4.31]. We won't see evidence of any crash until the victim
|
|
-- process terminates and the postmaster responds. If process termination
|
|
-- entails writing a core dump, that can take awhile. Wait for the process to
|
|
-- vanish. At that point, the postmaster has called waitpid() on the crashed
|
|
-- process, and it will accept no new connections until it has reinitialized
|
|
-- the cluster. (We can't exploit pg_stat_activity, because the crash happens
|
|
-- after the backend updates shared memory to reflect its impending exit.)
|
|
DO $pl$
|
|
DECLARE
|
|
detail text;
|
|
BEGIN
|
|
PERFORM wait_pid(crash_pid)
|
|
FROM dblink(connection_parameters(), $$
|
|
SELECT pg_backend_pid() FROM dblink(
|
|
'service=test_ldap '||connection_parameters(),
|
|
-- This string concatenation is a hack to shoehorn a
|
|
-- set_pgservicefile call into the SQL statement.
|
|
'SELECT 1' || set_pgservicefile('pg_service.conf')
|
|
) t(c int)
|
|
$$) AS t(crash_pid int);
|
|
EXCEPTION WHEN OTHERS THEN
|
|
GET STACKED DIAGNOSTICS detail = PG_EXCEPTION_DETAIL;
|
|
-- Expected error in a non-LDAP build.
|
|
IF NOT detail LIKE 'syntax error in service file%' THEN RAISE; END IF;
|
|
END
|
|
$pl$;
|
|
-- create a persistent connection
|
|
SELECT dblink_connect(connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- use the persistent connection
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
a | b | c
|
|
---+---+------------
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
(2 rows)
|
|
|
|
-- open a cursor with bad SQL and fail_on_error set to false
|
|
SELECT dblink_open('rmt_foo_cursor','SELECT * FROM foobar',false);
|
|
NOTICE: relation "foobar" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not open cursor.
|
|
dblink_open
|
|
-------------
|
|
ERROR
|
|
(1 row)
|
|
|
|
-- reset remote transaction state
|
|
SELECT dblink_exec('ABORT');
|
|
dblink_exec
|
|
-------------
|
|
ROLLBACK
|
|
(1 row)
|
|
|
|
-- open a cursor
|
|
SELECT dblink_open('rmt_foo_cursor','SELECT * FROM foo');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- close the cursor
|
|
SELECT dblink_close('rmt_foo_cursor',false);
|
|
dblink_close
|
|
--------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- open the cursor again
|
|
SELECT dblink_open('rmt_foo_cursor','SELECT * FROM foo');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- fetch some data
|
|
SELECT *
|
|
FROM dblink_fetch('rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
---+---+------------
|
|
0 | a | {a0,b0,c0}
|
|
1 | b | {a1,b1,c1}
|
|
2 | c | {a2,b2,c2}
|
|
3 | d | {a3,b3,c3}
|
|
(4 rows)
|
|
|
|
SELECT *
|
|
FROM dblink_fetch('rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
---+---+------------
|
|
4 | e | {a4,b4,c4}
|
|
5 | f | {a5,b5,c5}
|
|
6 | g | {a6,b6,c6}
|
|
7 | h | {a7,b7,c7}
|
|
(4 rows)
|
|
|
|
-- this one only finds two rows left
|
|
SELECT *
|
|
FROM dblink_fetch('rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
---+---+------------
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
(2 rows)
|
|
|
|
-- intentionally botch a fetch
|
|
SELECT *
|
|
FROM dblink_fetch('rmt_foobar_cursor',4,false) AS t(a int, b text, c text[]);
|
|
NOTICE: cursor "rmt_foobar_cursor" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not fetch from cursor.
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- reset remote transaction state
|
|
SELECT dblink_exec('ABORT');
|
|
dblink_exec
|
|
-------------
|
|
ROLLBACK
|
|
(1 row)
|
|
|
|
-- close the wrong cursor
|
|
SELECT dblink_close('rmt_foobar_cursor',false);
|
|
NOTICE: cursor "rmt_foobar_cursor" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not close cursor.
|
|
dblink_close
|
|
--------------
|
|
ERROR
|
|
(1 row)
|
|
|
|
-- should generate 'cursor "rmt_foo_cursor" not found' error
|
|
SELECT *
|
|
FROM dblink_fetch('rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
ERROR: cursor "rmt_foo_cursor" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not fetch from cursor.
|
|
-- this time, 'cursor "rmt_foo_cursor" not found' as a notice
|
|
SELECT *
|
|
FROM dblink_fetch('rmt_foo_cursor',4,false) AS t(a int, b text, c text[]);
|
|
NOTICE: cursor "rmt_foo_cursor" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not fetch from cursor.
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- close the persistent connection
|
|
SELECT dblink_disconnect();
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- should generate "connection not available" error
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
ERROR: connection not available
|
|
-- put more data into our slave table, first using arbitrary connection syntax
|
|
-- but truncate the actual return value so we can use diff to check for success
|
|
SELECT substr(dblink_exec(connection_parameters(),'INSERT INTO foo VALUES(10,''k'',''{"a10","b10","c10"}'')'),1,6);
|
|
substr
|
|
--------
|
|
INSERT
|
|
(1 row)
|
|
|
|
-- create a persistent connection
|
|
SELECT dblink_connect(connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- put more data into our slave table, using persistent connection syntax
|
|
-- but truncate the actual return value so we can use diff to check for success
|
|
SELECT substr(dblink_exec('INSERT INTO foo VALUES(11,''l'',''{"a11","b11","c11"}'')'),1,6);
|
|
substr
|
|
--------
|
|
INSERT
|
|
(1 row)
|
|
|
|
-- let's see it
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foo') AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
----+---+---------------
|
|
0 | a | {a0,b0,c0}
|
|
1 | b | {a1,b1,c1}
|
|
2 | c | {a2,b2,c2}
|
|
3 | d | {a3,b3,c3}
|
|
4 | e | {a4,b4,c4}
|
|
5 | f | {a5,b5,c5}
|
|
6 | g | {a6,b6,c6}
|
|
7 | h | {a7,b7,c7}
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
11 | l | {a11,b11,c11}
|
|
(12 rows)
|
|
|
|
-- bad remote select
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foobar',false) AS t(a int, b text, c text[]);
|
|
NOTICE: relation "foobar" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not execute query.
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- change some data
|
|
SELECT dblink_exec('UPDATE foo SET f3[2] = ''b99'' WHERE f1 = 11');
|
|
dblink_exec
|
|
-------------
|
|
UPDATE 1
|
|
(1 row)
|
|
|
|
-- let's see it
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE a = 11;
|
|
a | b | c
|
|
----+---+---------------
|
|
11 | l | {a11,b99,c11}
|
|
(1 row)
|
|
|
|
-- botch a change to some other data
|
|
SELECT dblink_exec('UPDATE foobar SET f3[2] = ''b99'' WHERE f1 = 11',false);
|
|
NOTICE: relation "foobar" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "unnamed": could not execute command.
|
|
dblink_exec
|
|
-------------
|
|
ERROR
|
|
(1 row)
|
|
|
|
-- delete some data
|
|
SELECT dblink_exec('DELETE FROM foo WHERE f1 = 11');
|
|
dblink_exec
|
|
-------------
|
|
DELETE 1
|
|
(1 row)
|
|
|
|
-- let's see it
|
|
SELECT *
|
|
FROM dblink('SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE a = 11;
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- close the persistent connection
|
|
SELECT dblink_disconnect();
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
--
|
|
-- tests for the new named persistent connection syntax
|
|
--
|
|
-- should generate "missing "=" after "myconn" in connection info string" error
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
ERROR: could not establish connection
|
|
DETAIL: missing "=" after "myconn" in connection info string
|
|
|
|
-- create a named persistent connection
|
|
SELECT dblink_connect('myconn',connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- use the named persistent connection
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
a | b | c
|
|
----+---+---------------
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
(3 rows)
|
|
|
|
-- use the named persistent connection, but get it wrong
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foobar',false) AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
NOTICE: relation "foobar" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "myconn": could not execute query.
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- create a second named persistent connection
|
|
-- should error with "duplicate connection name"
|
|
SELECT dblink_connect('myconn',connection_parameters());
|
|
ERROR: duplicate connection name
|
|
-- create a second named persistent connection with a new name
|
|
SELECT dblink_connect('myconn2',connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- use the second named persistent connection
|
|
SELECT *
|
|
FROM dblink('myconn2','SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
a | b | c
|
|
----+---+---------------
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
(3 rows)
|
|
|
|
-- close the second named persistent connection
|
|
SELECT dblink_disconnect('myconn2');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- open a cursor incorrectly
|
|
SELECT dblink_open('myconn','rmt_foo_cursor','SELECT * FROM foobar',false);
|
|
NOTICE: relation "foobar" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "myconn": could not open cursor.
|
|
dblink_open
|
|
-------------
|
|
ERROR
|
|
(1 row)
|
|
|
|
-- reset remote transaction state
|
|
SELECT dblink_exec('myconn','ABORT');
|
|
dblink_exec
|
|
-------------
|
|
ROLLBACK
|
|
(1 row)
|
|
|
|
-- test opening cursor in a transaction
|
|
SELECT dblink_exec('myconn','BEGIN');
|
|
dblink_exec
|
|
-------------
|
|
BEGIN
|
|
(1 row)
|
|
|
|
-- an open transaction will prevent dblink_open() from opening its own
|
|
SELECT dblink_open('myconn','rmt_foo_cursor','SELECT * FROM foo');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- this should not commit the transaction because the client opened it
|
|
SELECT dblink_close('myconn','rmt_foo_cursor');
|
|
dblink_close
|
|
--------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- this should succeed because we have an open transaction
|
|
SELECT dblink_exec('myconn','DECLARE xact_test CURSOR FOR SELECT * FROM foo');
|
|
dblink_exec
|
|
----------------
|
|
DECLARE CURSOR
|
|
(1 row)
|
|
|
|
-- commit remote transaction
|
|
SELECT dblink_exec('myconn','COMMIT');
|
|
dblink_exec
|
|
-------------
|
|
COMMIT
|
|
(1 row)
|
|
|
|
-- test automatic transactions for multiple cursor opens
|
|
SELECT dblink_open('myconn','rmt_foo_cursor','SELECT * FROM foo');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- the second cursor
|
|
SELECT dblink_open('myconn','rmt_foo_cursor2','SELECT * FROM foo');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- this should not commit the transaction
|
|
SELECT dblink_close('myconn','rmt_foo_cursor2');
|
|
dblink_close
|
|
--------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- this should succeed because we have an open transaction
|
|
SELECT dblink_exec('myconn','DECLARE xact_test CURSOR FOR SELECT * FROM foo');
|
|
dblink_exec
|
|
----------------
|
|
DECLARE CURSOR
|
|
(1 row)
|
|
|
|
-- this should commit the transaction
|
|
SELECT dblink_close('myconn','rmt_foo_cursor');
|
|
dblink_close
|
|
--------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- this should fail because there is no open transaction
|
|
SELECT dblink_exec('myconn','DECLARE xact_test CURSOR FOR SELECT * FROM foo');
|
|
ERROR: DECLARE CURSOR can only be used in transaction blocks
|
|
CONTEXT: Error occurred on dblink connection named "myconn": could not execute command.
|
|
-- reset remote transaction state
|
|
SELECT dblink_exec('myconn','ABORT');
|
|
dblink_exec
|
|
-------------
|
|
ROLLBACK
|
|
(1 row)
|
|
|
|
-- open a cursor
|
|
SELECT dblink_open('myconn','rmt_foo_cursor','SELECT * FROM foo');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- fetch some data
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
---+---+------------
|
|
0 | a | {a0,b0,c0}
|
|
1 | b | {a1,b1,c1}
|
|
2 | c | {a2,b2,c2}
|
|
3 | d | {a3,b3,c3}
|
|
(4 rows)
|
|
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
---+---+------------
|
|
4 | e | {a4,b4,c4}
|
|
5 | f | {a5,b5,c5}
|
|
6 | g | {a6,b6,c6}
|
|
7 | h | {a7,b7,c7}
|
|
(4 rows)
|
|
|
|
-- this one only finds three rows left
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
----+---+---------------
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
(3 rows)
|
|
|
|
-- fetch some data incorrectly
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','rmt_foobar_cursor',4,false) AS t(a int, b text, c text[]);
|
|
NOTICE: cursor "rmt_foobar_cursor" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "myconn": could not fetch from cursor.
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- reset remote transaction state
|
|
SELECT dblink_exec('myconn','ABORT');
|
|
dblink_exec
|
|
-------------
|
|
ROLLBACK
|
|
(1 row)
|
|
|
|
-- should generate 'cursor "rmt_foo_cursor" not found' error
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','rmt_foo_cursor',4) AS t(a int, b text, c text[]);
|
|
ERROR: cursor "rmt_foo_cursor" does not exist
|
|
CONTEXT: Error occurred on dblink connection named "myconn": could not fetch from cursor.
|
|
-- close the named persistent connection
|
|
SELECT dblink_disconnect('myconn');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- should generate "missing "=" after "myconn" in connection info string" error
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE t.a > 7;
|
|
ERROR: could not establish connection
|
|
DETAIL: missing "=" after "myconn" in connection info string
|
|
|
|
-- create a named persistent connection
|
|
SELECT dblink_connect('myconn',connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- put more data into our slave table, using named persistent connection syntax
|
|
-- but truncate the actual return value so we can use diff to check for success
|
|
SELECT substr(dblink_exec('myconn','INSERT INTO foo VALUES(11,''l'',''{"a11","b11","c11"}'')'),1,6);
|
|
substr
|
|
--------
|
|
INSERT
|
|
(1 row)
|
|
|
|
-- let's see it
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
----+---+---------------
|
|
0 | a | {a0,b0,c0}
|
|
1 | b | {a1,b1,c1}
|
|
2 | c | {a2,b2,c2}
|
|
3 | d | {a3,b3,c3}
|
|
4 | e | {a4,b4,c4}
|
|
5 | f | {a5,b5,c5}
|
|
6 | g | {a6,b6,c6}
|
|
7 | h | {a7,b7,c7}
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
11 | l | {a11,b11,c11}
|
|
(12 rows)
|
|
|
|
-- change some data
|
|
SELECT dblink_exec('myconn','UPDATE foo SET f3[2] = ''b99'' WHERE f1 = 11');
|
|
dblink_exec
|
|
-------------
|
|
UPDATE 1
|
|
(1 row)
|
|
|
|
-- let's see it
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE a = 11;
|
|
a | b | c
|
|
----+---+---------------
|
|
11 | l | {a11,b99,c11}
|
|
(1 row)
|
|
|
|
-- delete some data
|
|
SELECT dblink_exec('myconn','DELETE FROM foo WHERE f1 = 11');
|
|
dblink_exec
|
|
-------------
|
|
DELETE 1
|
|
(1 row)
|
|
|
|
-- let's see it
|
|
SELECT *
|
|
FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[])
|
|
WHERE a = 11;
|
|
a | b | c
|
|
---+---+---
|
|
(0 rows)
|
|
|
|
-- close the named persistent connection
|
|
SELECT dblink_disconnect('myconn');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- close the named persistent connection again
|
|
-- should get 'connection "myconn" not available' error
|
|
SELECT dblink_disconnect('myconn');
|
|
ERROR: connection "myconn" not available
|
|
-- test asynchronous queries
|
|
SELECT dblink_connect('dtest1', connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT * from
|
|
dblink_send_query('dtest1', 'select * from foo where f1 < 3') as t1;
|
|
t1
|
|
----
|
|
1
|
|
(1 row)
|
|
|
|
SELECT dblink_connect('dtest2', connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT * from
|
|
dblink_send_query('dtest2', 'select * from foo where f1 > 2 and f1 < 7') as t1;
|
|
t1
|
|
----
|
|
1
|
|
(1 row)
|
|
|
|
SELECT dblink_connect('dtest3', connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT * from
|
|
dblink_send_query('dtest3', 'select * from foo where f1 > 6') as t1;
|
|
t1
|
|
----
|
|
1
|
|
(1 row)
|
|
|
|
CREATE TEMPORARY TABLE result AS
|
|
(SELECT * from dblink_get_result('dtest1') as t1(f1 int, f2 text, f3 text[]))
|
|
UNION
|
|
(SELECT * from dblink_get_result('dtest2') as t2(f1 int, f2 text, f3 text[]))
|
|
UNION
|
|
(SELECT * from dblink_get_result('dtest3') as t3(f1 int, f2 text, f3 text[]))
|
|
ORDER by f1;
|
|
-- dblink_get_connections returns an array with elements in a machine-dependent
|
|
-- ordering, so we must resort to unnesting and sorting for a stable result
|
|
create function unnest(anyarray) returns setof anyelement
|
|
language sql strict immutable as $$
|
|
select $1[i] from generate_series(array_lower($1,1), array_upper($1,1)) as i
|
|
$$;
|
|
SELECT * FROM unnest(dblink_get_connections()) ORDER BY 1;
|
|
unnest
|
|
--------
|
|
dtest1
|
|
dtest2
|
|
dtest3
|
|
(3 rows)
|
|
|
|
SELECT dblink_is_busy('dtest1');
|
|
dblink_is_busy
|
|
----------------
|
|
0
|
|
(1 row)
|
|
|
|
SELECT dblink_disconnect('dtest1');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT dblink_disconnect('dtest2');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT dblink_disconnect('dtest3');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT * from result;
|
|
f1 | f2 | f3
|
|
----+----+---------------
|
|
0 | a | {a0,b0,c0}
|
|
1 | b | {a1,b1,c1}
|
|
2 | c | {a2,b2,c2}
|
|
3 | d | {a3,b3,c3}
|
|
4 | e | {a4,b4,c4}
|
|
5 | f | {a5,b5,c5}
|
|
6 | g | {a6,b6,c6}
|
|
7 | h | {a7,b7,c7}
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
(11 rows)
|
|
|
|
SELECT dblink_connect('dtest1', connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT * from
|
|
dblink_send_query('dtest1', 'select * from foo where f1 < 3') as t1;
|
|
t1
|
|
----
|
|
1
|
|
(1 row)
|
|
|
|
SELECT dblink_cancel_query('dtest1');
|
|
dblink_cancel_query
|
|
---------------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT dblink_error_message('dtest1');
|
|
dblink_error_message
|
|
----------------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT dblink_disconnect('dtest1');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- test foreign data wrapper functionality
|
|
CREATE ROLE regress_dblink_user;
|
|
DO $d$
|
|
BEGIN
|
|
EXECUTE $$CREATE SERVER fdtest FOREIGN DATA WRAPPER dblink_fdw
|
|
OPTIONS (dbname '$$||current_database()||$$',
|
|
port '$$||current_setting('port')||$$'
|
|
)$$;
|
|
END;
|
|
$d$;
|
|
CREATE USER MAPPING FOR public SERVER fdtest
|
|
OPTIONS (server 'localhost'); -- fail, can't specify server here
|
|
ERROR: invalid option "server"
|
|
HINT: Valid options in this context are: user, password
|
|
CREATE USER MAPPING FOR public SERVER fdtest OPTIONS (user :'USER');
|
|
GRANT USAGE ON FOREIGN SERVER fdtest TO regress_dblink_user;
|
|
GRANT EXECUTE ON FUNCTION dblink_connect_u(text, text) TO regress_dblink_user;
|
|
SET SESSION AUTHORIZATION regress_dblink_user;
|
|
-- should fail
|
|
SELECT dblink_connect('myconn', 'fdtest');
|
|
ERROR: password is required
|
|
DETAIL: Non-superusers must provide a password in the connection string.
|
|
-- should succeed
|
|
SELECT dblink_connect_u('myconn', 'fdtest');
|
|
dblink_connect_u
|
|
------------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT * FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]);
|
|
a | b | c
|
|
----+---+---------------
|
|
0 | a | {a0,b0,c0}
|
|
1 | b | {a1,b1,c1}
|
|
2 | c | {a2,b2,c2}
|
|
3 | d | {a3,b3,c3}
|
|
4 | e | {a4,b4,c4}
|
|
5 | f | {a5,b5,c5}
|
|
6 | g | {a6,b6,c6}
|
|
7 | h | {a7,b7,c7}
|
|
8 | i | {a8,b8,c8}
|
|
9 | j | {a9,b9,c9}
|
|
10 | k | {a10,b10,c10}
|
|
(11 rows)
|
|
|
|
\c - -
|
|
REVOKE USAGE ON FOREIGN SERVER fdtest FROM regress_dblink_user;
|
|
REVOKE EXECUTE ON FUNCTION dblink_connect_u(text, text) FROM regress_dblink_user;
|
|
DROP USER regress_dblink_user;
|
|
DROP USER MAPPING FOR public SERVER fdtest;
|
|
DROP SERVER fdtest;
|
|
-- test asynchronous notifications
|
|
SELECT dblink_connect(connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
--should return listen
|
|
SELECT dblink_exec('LISTEN regression');
|
|
dblink_exec
|
|
-------------
|
|
LISTEN
|
|
(1 row)
|
|
|
|
--should return listen
|
|
SELECT dblink_exec('LISTEN foobar');
|
|
dblink_exec
|
|
-------------
|
|
LISTEN
|
|
(1 row)
|
|
|
|
SELECT dblink_exec('NOTIFY regression');
|
|
dblink_exec
|
|
-------------
|
|
NOTIFY
|
|
(1 row)
|
|
|
|
SELECT dblink_exec('NOTIFY foobar');
|
|
dblink_exec
|
|
-------------
|
|
NOTIFY
|
|
(1 row)
|
|
|
|
SELECT notify_name, be_pid = (select t.be_pid from dblink('select pg_backend_pid()') as t(be_pid int)) AS is_self_notify, extra from dblink_get_notify();
|
|
notify_name | is_self_notify | extra
|
|
-------------+----------------+-------
|
|
regression | t |
|
|
foobar | t |
|
|
(2 rows)
|
|
|
|
SELECT * from dblink_get_notify();
|
|
notify_name | be_pid | extra
|
|
-------------+--------+-------
|
|
(0 rows)
|
|
|
|
SELECT dblink_disconnect();
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
-- test dropped columns in dblink_build_sql_insert, dblink_build_sql_update
|
|
CREATE TEMP TABLE test_dropped
|
|
(
|
|
col1 INT NOT NULL DEFAULT 111,
|
|
id SERIAL PRIMARY KEY,
|
|
col2 INT NOT NULL DEFAULT 112,
|
|
col2b INT NOT NULL DEFAULT 113
|
|
);
|
|
INSERT INTO test_dropped VALUES(default);
|
|
ALTER TABLE test_dropped
|
|
DROP COLUMN col1,
|
|
DROP COLUMN col2,
|
|
ADD COLUMN col3 VARCHAR(10) NOT NULL DEFAULT 'foo',
|
|
ADD COLUMN col4 INT NOT NULL DEFAULT 42;
|
|
SELECT dblink_build_sql_insert('test_dropped', '1', 1,
|
|
ARRAY['1'::TEXT], ARRAY['2'::TEXT]);
|
|
dblink_build_sql_insert
|
|
---------------------------------------------------------------------------
|
|
INSERT INTO test_dropped(id,col2b,col3,col4) VALUES('2','113','foo','42')
|
|
(1 row)
|
|
|
|
SELECT dblink_build_sql_update('test_dropped', '1', 1,
|
|
ARRAY['1'::TEXT], ARRAY['2'::TEXT]);
|
|
dblink_build_sql_update
|
|
-------------------------------------------------------------------------------------------
|
|
UPDATE test_dropped SET id = '2', col2b = '113', col3 = 'foo', col4 = '42' WHERE id = '2'
|
|
(1 row)
|
|
|
|
SELECT dblink_build_sql_delete('test_dropped', '1', 1,
|
|
ARRAY['2'::TEXT]);
|
|
dblink_build_sql_delete
|
|
-----------------------------------------
|
|
DELETE FROM test_dropped WHERE id = '2'
|
|
(1 row)
|
|
|
|
-- test local mimicry of remote GUC values that affect datatype I/O
|
|
SET datestyle = ISO, MDY;
|
|
SET intervalstyle = postgres;
|
|
SET timezone = UTC;
|
|
SELECT dblink_connect('myconn',connection_parameters());
|
|
dblink_connect
|
|
----------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT dblink_exec('myconn', 'SET datestyle = GERMAN, DMY;');
|
|
dblink_exec
|
|
-------------
|
|
SET
|
|
(1 row)
|
|
|
|
-- single row synchronous case
|
|
SELECT *
|
|
FROM dblink('myconn',
|
|
'SELECT * FROM (VALUES (''12.03.2013 00:00:00+00'')) t')
|
|
AS t(a timestamptz);
|
|
a
|
|
------------------------
|
|
2013-03-12 00:00:00+00
|
|
(1 row)
|
|
|
|
-- multi-row synchronous case
|
|
SELECT *
|
|
FROM dblink('myconn',
|
|
'SELECT * FROM
|
|
(VALUES (''12.03.2013 00:00:00+00''),
|
|
(''12.03.2013 00:00:00+00'')) t')
|
|
AS t(a timestamptz);
|
|
a
|
|
------------------------
|
|
2013-03-12 00:00:00+00
|
|
2013-03-12 00:00:00+00
|
|
(2 rows)
|
|
|
|
-- single-row asynchronous case
|
|
SELECT *
|
|
FROM dblink_send_query('myconn',
|
|
'SELECT * FROM
|
|
(VALUES (''12.03.2013 00:00:00+00'')) t');
|
|
dblink_send_query
|
|
-------------------
|
|
1
|
|
(1 row)
|
|
|
|
CREATE TEMPORARY TABLE result AS
|
|
(SELECT * from dblink_get_result('myconn') as t(t timestamptz))
|
|
UNION ALL
|
|
(SELECT * from dblink_get_result('myconn') as t(t timestamptz));
|
|
SELECT * FROM result;
|
|
t
|
|
------------------------
|
|
2013-03-12 00:00:00+00
|
|
(1 row)
|
|
|
|
DROP TABLE result;
|
|
-- multi-row asynchronous case
|
|
SELECT *
|
|
FROM dblink_send_query('myconn',
|
|
'SELECT * FROM
|
|
(VALUES (''12.03.2013 00:00:00+00''),
|
|
(''12.03.2013 00:00:00+00'')) t');
|
|
dblink_send_query
|
|
-------------------
|
|
1
|
|
(1 row)
|
|
|
|
CREATE TEMPORARY TABLE result AS
|
|
(SELECT * from dblink_get_result('myconn') as t(t timestamptz))
|
|
UNION ALL
|
|
(SELECT * from dblink_get_result('myconn') as t(t timestamptz))
|
|
UNION ALL
|
|
(SELECT * from dblink_get_result('myconn') as t(t timestamptz));
|
|
SELECT * FROM result;
|
|
t
|
|
------------------------
|
|
2013-03-12 00:00:00+00
|
|
2013-03-12 00:00:00+00
|
|
(2 rows)
|
|
|
|
DROP TABLE result;
|
|
-- Try an ambiguous interval
|
|
SELECT dblink_exec('myconn', 'SET intervalstyle = sql_standard;');
|
|
dblink_exec
|
|
-------------
|
|
SET
|
|
(1 row)
|
|
|
|
SELECT *
|
|
FROM dblink('myconn',
|
|
'SELECT * FROM (VALUES (''-1 2:03:04'')) i')
|
|
AS i(i interval);
|
|
i
|
|
-------------------
|
|
-1 days -02:03:04
|
|
(1 row)
|
|
|
|
-- Try swapping to another format to ensure the GUCs are tracked
|
|
-- properly through a change.
|
|
CREATE TEMPORARY TABLE result (t timestamptz);
|
|
SELECT dblink_exec('myconn', 'SET datestyle = ISO, MDY;');
|
|
dblink_exec
|
|
-------------
|
|
SET
|
|
(1 row)
|
|
|
|
INSERT INTO result
|
|
SELECT *
|
|
FROM dblink('myconn',
|
|
'SELECT * FROM (VALUES (''03.12.2013 00:00:00+00'')) t')
|
|
AS t(a timestamptz);
|
|
SELECT dblink_exec('myconn', 'SET datestyle = GERMAN, DMY;');
|
|
dblink_exec
|
|
-------------
|
|
SET
|
|
(1 row)
|
|
|
|
INSERT INTO result
|
|
SELECT *
|
|
FROM dblink('myconn',
|
|
'SELECT * FROM (VALUES (''12.03.2013 00:00:00+00'')) t')
|
|
AS t(a timestamptz);
|
|
SELECT * FROM result;
|
|
t
|
|
------------------------
|
|
2013-03-12 00:00:00+00
|
|
2013-03-12 00:00:00+00
|
|
(2 rows)
|
|
|
|
DROP TABLE result;
|
|
-- Check error throwing in dblink_fetch
|
|
SELECT dblink_open('myconn','error_cursor',
|
|
'SELECT * FROM (VALUES (''1''), (''not an int'')) AS t(text);');
|
|
dblink_open
|
|
-------------
|
|
OK
|
|
(1 row)
|
|
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','error_cursor', 1) AS t(i int);
|
|
i
|
|
---
|
|
1
|
|
(1 row)
|
|
|
|
SELECT *
|
|
FROM dblink_fetch('myconn','error_cursor', 1) AS t(i int);
|
|
ERROR: invalid input syntax for integer: "not an int"
|
|
-- Make sure that the local settings have retained their values in spite
|
|
-- of shenanigans on the connection.
|
|
SHOW datestyle;
|
|
DateStyle
|
|
-----------
|
|
ISO, MDY
|
|
(1 row)
|
|
|
|
SHOW intervalstyle;
|
|
IntervalStyle
|
|
---------------
|
|
postgres
|
|
(1 row)
|
|
|
|
-- Clean up GUC-setting tests
|
|
SELECT dblink_disconnect('myconn');
|
|
dblink_disconnect
|
|
-------------------
|
|
OK
|
|
(1 row)
|
|
|
|
RESET datestyle;
|
|
RESET intervalstyle;
|
|
RESET timezone;
|