mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-08-29 23:57:33 +02:00
This patch providies for support for password protected SSL client keys in libpq, and for DER format keys, both encrypted and unencrypted. There is a new connection parameter sslpassword, which is supplied to the OpenSSL libraries via a callback function. The callback function can also be set by an application by calling PQgetSSLKeyPassHook(). There is also a function to retreive the connection setting, PQsslpassword(). Craig Ringer and Andrew Dunstan Reviewed by: Greg Nancarrow Discussion: https://postgr.es/m/f7ee88ed-95c4-95c1-d4bf-7b415363ab62@2ndQuadrant.com
1182 lines
32 KiB
Plaintext
1182 lines
32 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
|
|
-- repeat the test for table with primary key index with included columns
|
|
CREATE TABLE foo_1(f1 int, f2 text, f3 text[], primary key (f1,f2) include (f3));
|
|
INSERT INTO foo_1 VALUES (0,'a','{"a0","b0","c0"}');
|
|
INSERT INTO foo_1 VALUES (1,'b','{"a1","b1","c1"}');
|
|
INSERT INTO foo_1 VALUES (2,'c','{"a2","b2","c2"}');
|
|
INSERT INTO foo_1 VALUES (3,'d','{"a3","b3","c3"}');
|
|
INSERT INTO foo_1 VALUES (4,'e','{"a4","b4","c4"}');
|
|
INSERT INTO foo_1 VALUES (5,'f','{"a5","b5","c5"}');
|
|
INSERT INTO foo_1 VALUES (6,'g','{"a6","b6","c6"}');
|
|
INSERT INTO foo_1 VALUES (7,'h','{"a7","b7","c7"}');
|
|
INSERT INTO foo_1 VALUES (8,'i','{"a8","b8","c8"}');
|
|
INSERT INTO foo_1 VALUES (9,'j','{"a9","b9","c9"}');
|
|
-- misc utilities
|
|
-- list the primary key fields
|
|
SELECT *
|
|
FROM dblink_get_pkey('foo_1');
|
|
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','1 2',2,'{"0", "a"}','{"99", "xyz"}');
|
|
dblink_build_sql_insert
|
|
-------------------------------------------------------------
|
|
INSERT INTO foo_1(f1,f2,f3) VALUES('99','xyz','{a0,b0,c0}')
|
|
(1 row)
|
|
|
|
-- too many pk fields, should fail
|
|
SELECT dblink_build_sql_insert('foo_1','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','1 2',2,'{"0", "a"}','{"99", "xyz"}');
|
|
dblink_build_sql_update
|
|
------------------------------------------------------------------------------------------
|
|
UPDATE foo_1 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','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','1 2',2,'{"0", "a"}');
|
|
dblink_build_sql_delete
|
|
-----------------------------------------------
|
|
DELETE FROM foo_1 WHERE f1 = '0' AND f2 = 'a'
|
|
(1 row)
|
|
|
|
-- too many pk fields, should fail
|
|
SELECT dblink_build_sql_delete('foo_1','1 2 3 4',4,'{"0", "a", "{a0,b0,c0}"}');
|
|
ERROR: invalid attribute number 4
|
|
DROP TABLE foo_1;
|
|
-- 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: while opening cursor "rmt_foo_cursor" on unnamed dblink connection
|
|
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: while fetching from cursor "rmt_foobar_cursor" on unnamed dblink connection
|
|
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: while closing cursor "rmt_foobar_cursor" on unnamed dblink connection
|
|
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: while fetching from cursor "rmt_foo_cursor" on unnamed dblink connection
|
|
-- 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: while fetching from cursor "rmt_foo_cursor" on unnamed dblink connection
|
|
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 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 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: while executing query on unnamed dblink connection
|
|
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: while executing command on unnamed dblink connection
|
|
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: while executing query on dblink connection named "myconn"
|
|
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: while opening cursor "rmt_foo_cursor" on dblink connection named "myconn"
|
|
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: while executing command on dblink connection named "myconn"
|
|
-- 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: while fetching from cursor "rmt_foobar_cursor" on dblink connection named "myconn"
|
|
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: while fetching from cursor "rmt_foo_cursor" on dblink connection named "myconn"
|
|
-- 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 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, sslpassword
|
|
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 type 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;
|