mirror of
https://github.com/postgres/postgres.git
synced 2025-06-11 20:28:21 +03:00
Allow TRUNCATE command to truncate foreign tables.
This commit introduces new foreign data wrapper API for TRUNCATE. It extends TRUNCATE command so that it accepts foreign tables as the targets to truncate and invokes that API. Also it extends postgres_fdw so that it can issue TRUNCATE command to foreign servers, by adding new routine for that TRUNCATE API. The information about options specified in TRUNCATE command, e.g., ONLY, CACADE, etc is passed to FDW via API. The list of foreign tables to truncate is also passed to FDW. FDW truncates the foreign data sources that the passed foreign tables specify, based on those information. For example, postgres_fdw constructs TRUNCATE command using them and issues it to the foreign server. For performance, TRUNCATE command invokes the FDW routine for TRUNCATE once per foreign server that foreign tables to truncate belong to. Author: Kazutaka Onishi, Kohei KaiGai, slightly modified by Fujii Masao Reviewed-by: Bharath Rupireddy, Michael Paquier, Zhihong Yu, Alvaro Herrera, Stephen Frost, Ashutosh Bapat, Amit Langote, Daniel Gustafsson, Ibrar Ahmed, Fujii Masao Discussion: https://postgr.es/m/CAOP8fzb_gkReLput7OvOK+8NHgw-RKqNv59vem7=524krQTcWA@mail.gmail.com Discussion: https://postgr.es/m/CAJuF6cMWDDqU-vn_knZgma+2GMaout68YUgn1uyDnexRhqqM5Q@mail.gmail.com
This commit is contained in:
Fujii Masao
@ -8215,6 +8215,205 @@ select * from rem3;
|
||||
drop foreign table rem3;
|
||||
drop table loc3;
|
||||
-- ===================================================================
|
||||
-- test for TRUNCATE
|
||||
-- ===================================================================
|
||||
CREATE TABLE tru_rtable0 (id int primary key);
|
||||
CREATE TABLE tru_rtable1 (id int primary key);
|
||||
CREATE FOREIGN TABLE tru_ftable (id int)
|
||||
SERVER loopback OPTIONS (table_name 'tru_rtable0');
|
||||
INSERT INTO tru_rtable0 (SELECT x FROM generate_series(1,10) x);
|
||||
CREATE TABLE tru_ptable (id int) PARTITION BY HASH(id);
|
||||
CREATE TABLE tru_ptable__p0 PARTITION OF tru_ptable
|
||||
FOR VALUES WITH (MODULUS 2, REMAINDER 0);
|
||||
CREATE FOREIGN TABLE tru_ftable__p1 PARTITION OF tru_ptable
|
||||
FOR VALUES WITH (MODULUS 2, REMAINDER 1)
|
||||
SERVER loopback OPTIONS (table_name 'tru_rtable1');
|
||||
INSERT INTO tru_ptable (SELECT x FROM generate_series(11,20) x);
|
||||
CREATE TABLE tru_pk_table(id int primary key);
|
||||
CREATE TABLE tru_fk_table(fkey int references tru_pk_table(id));
|
||||
INSERT INTO tru_pk_table (SELECT x FROM generate_series(1,10) x);
|
||||
INSERT INTO tru_fk_table (SELECT x % 10 + 1 FROM generate_series(5,25) x);
|
||||
CREATE FOREIGN TABLE tru_pk_ftable (id int)
|
||||
SERVER loopback OPTIONS (table_name 'tru_pk_table');
|
||||
CREATE TABLE tru_rtable_parent (id int);
|
||||
CREATE TABLE tru_rtable_child (id int);
|
||||
CREATE FOREIGN TABLE tru_ftable_parent (id int)
|
||||
SERVER loopback OPTIONS (table_name 'tru_rtable_parent');
|
||||
CREATE FOREIGN TABLE tru_ftable_child () INHERITS (tru_ftable_parent)
|
||||
SERVER loopback OPTIONS (table_name 'tru_rtable_child');
|
||||
INSERT INTO tru_rtable_parent (SELECT x FROM generate_series(1,8) x);
|
||||
INSERT INTO tru_rtable_child (SELECT x FROM generate_series(10, 18) x);
|
||||
-- normal truncate
|
||||
SELECT sum(id) FROM tru_ftable; -- 55
|
||||
sum
|
||||
-----
|
||||
55
|
||||
(1 row)
|
||||
|
||||
TRUNCATE tru_ftable;
|
||||
SELECT count(*) FROM tru_rtable0; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) FROM tru_ftable; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
-- 'truncatable' option
|
||||
ALTER SERVER loopback OPTIONS (ADD truncatable 'false');
|
||||
TRUNCATE tru_ftable; -- error
|
||||
ERROR: foreign table "tru_ftable" does not allow truncates
|
||||
ALTER FOREIGN TABLE tru_ftable OPTIONS (ADD truncatable 'true');
|
||||
TRUNCATE tru_ftable; -- accepted
|
||||
ALTER FOREIGN TABLE tru_ftable OPTIONS (SET truncatable 'false');
|
||||
TRUNCATE tru_ftable; -- error
|
||||
ERROR: foreign table "tru_ftable" does not allow truncates
|
||||
ALTER SERVER loopback OPTIONS (DROP truncatable);
|
||||
ALTER FOREIGN TABLE tru_ftable OPTIONS (SET truncatable 'false');
|
||||
TRUNCATE tru_ftable; -- error
|
||||
ERROR: foreign table "tru_ftable" does not allow truncates
|
||||
ALTER FOREIGN TABLE tru_ftable OPTIONS (SET truncatable 'true');
|
||||
TRUNCATE tru_ftable; -- accepted
|
||||
-- partitioned table with both local and foreign tables as partitions
|
||||
SELECT sum(id) FROM tru_ptable; -- 155
|
||||
sum
|
||||
-----
|
||||
155
|
||||
(1 row)
|
||||
|
||||
TRUNCATE tru_ptable;
|
||||
SELECT count(*) FROM tru_ptable; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) FROM tru_ptable__p0; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) FROM tru_ftable__p1; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) FROM tru_rtable1; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
-- 'CASCADE' option
|
||||
SELECT sum(id) FROM tru_pk_ftable; -- 55
|
||||
sum
|
||||
-----
|
||||
55
|
||||
(1 row)
|
||||
|
||||
TRUNCATE tru_pk_ftable; -- failed by FK reference
|
||||
ERROR: cannot truncate a table referenced in a foreign key constraint
|
||||
DETAIL: Table "tru_fk_table" references "tru_pk_table".
|
||||
HINT: Truncate table "tru_fk_table" at the same time, or use TRUNCATE ... CASCADE.
|
||||
CONTEXT: remote SQL command: TRUNCATE public.tru_pk_table CONTINUE IDENTITY RESTRICT
|
||||
TRUNCATE tru_pk_ftable CASCADE;
|
||||
SELECT count(*) FROM tru_pk_ftable; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) FROM tru_fk_table; -- also truncated,0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
-- truncate two tables at a command
|
||||
INSERT INTO tru_ftable (SELECT x FROM generate_series(1,8) x);
|
||||
INSERT INTO tru_pk_ftable (SELECT x FROM generate_series(3,10) x);
|
||||
SELECT count(*) from tru_ftable; -- 8
|
||||
count
|
||||
-------
|
||||
8
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) from tru_pk_ftable; -- 8
|
||||
count
|
||||
-------
|
||||
8
|
||||
(1 row)
|
||||
|
||||
TRUNCATE tru_ftable, tru_pk_ftable CASCADE;
|
||||
SELECT count(*) from tru_ftable; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
SELECT count(*) from tru_pk_ftable; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
-- truncate with ONLY clause
|
||||
TRUNCATE ONLY tru_ftable_parent;
|
||||
SELECT sum(id) FROM tru_ftable_parent; -- 126
|
||||
sum
|
||||
-----
|
||||
126
|
||||
(1 row)
|
||||
|
||||
TRUNCATE tru_ftable_parent;
|
||||
SELECT count(*) FROM tru_ftable_parent; -- 0
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
-- in case when remote table has inherited children
|
||||
CREATE TABLE tru_rtable0_child () INHERITS (tru_rtable0);
|
||||
INSERT INTO tru_rtable0 (SELECT x FROM generate_series(5,9) x);
|
||||
INSERT INTO tru_rtable0_child (SELECT x FROM generate_series(10,14) x);
|
||||
SELECT sum(id) FROM tru_ftable; -- 95
|
||||
sum
|
||||
-----
|
||||
95
|
||||
(1 row)
|
||||
|
||||
TRUNCATE ONLY tru_ftable; -- truncate only parent portion
|
||||
SELECT sum(id) FROM tru_ftable; -- 60
|
||||
sum
|
||||
-----
|
||||
60
|
||||
(1 row)
|
||||
|
||||
INSERT INTO tru_rtable0 (SELECT x FROM generate_series(21,25) x);
|
||||
SELECT sum(id) FROM tru_ftable; -- 175
|
||||
sum
|
||||
-----
|
||||
175
|
||||
(1 row)
|
||||
|
||||
TRUNCATE tru_ftable; -- truncate both of parent and child
|
||||
SELECT count(*) FROM tru_ftable; -- empty
|
||||
count
|
||||
-------
|
||||
0
|
||||
(1 row)
|
||||
|
||||
-- cleanup
|
||||
DROP FOREIGN TABLE tru_ftable_parent, tru_ftable_child, tru_pk_ftable,tru_ftable__p1,tru_ftable;
|
||||
DROP TABLE tru_rtable0, tru_rtable1, tru_ptable, tru_ptable__p0, tru_pk_table, tru_fk_table,
|
||||
tru_rtable_parent,tru_rtable_child, tru_rtable0_child;
|
||||
-- ===================================================================
|
||||
-- test IMPORT FOREIGN SCHEMA
|
||||
-- ===================================================================
|
||||
CREATE SCHEMA import_source;
|
||||
@ -8917,7 +9116,7 @@ DO $d$
|
||||
END;
|
||||
$d$;
|
||||
ERROR: invalid option "password"
|
||||
HINT: Valid options in this context are: service, passfile, channel_binding, connect_timeout, dbname, host, hostaddr, port, options, application_name, keepalives, keepalives_idle, keepalives_interval, keepalives_count, tcp_user_timeout, sslmode, sslcompression, sslcert, sslkey, sslrootcert, sslcrl, sslcrldir, sslsni, requirepeer, ssl_min_protocol_version, ssl_max_protocol_version, gssencmode, krbsrvname, gsslib, target_session_attrs, use_remote_estimate, fdw_startup_cost, fdw_tuple_cost, extensions, updatable, fetch_size, batch_size, async_capable, keep_connections
|
||||
HINT: Valid options in this context are: service, passfile, channel_binding, connect_timeout, dbname, host, hostaddr, port, options, application_name, keepalives, keepalives_idle, keepalives_interval, keepalives_count, tcp_user_timeout, sslmode, sslcompression, sslcert, sslkey, sslrootcert, sslcrl, sslcrldir, sslsni, requirepeer, ssl_min_protocol_version, ssl_max_protocol_version, gssencmode, krbsrvname, gsslib, target_session_attrs, use_remote_estimate, fdw_startup_cost, fdw_tuple_cost, extensions, updatable, truncatable, fetch_size, batch_size, async_capable, keep_connections
|
||||
CONTEXT: SQL statement "ALTER SERVER loopback_nopw OPTIONS (ADD password 'dummypw')"
|
||||
PL/pgSQL function inline_code_block line 3 at EXECUTE
|
||||
-- If we add a password for our user mapping instead, we should get a different
|
||||
|
||||
Reference in New Issue
Block a user