mirror of https://github.com/citusdata/citus.git
309 lines
11 KiB
Plaintext
309 lines
11 KiB
Plaintext
--
|
|
-- failure_online_move_shard_placement
|
|
--
|
|
-- The tests cover moving shard placements using logical replication.
|
|
CREATE SCHEMA IF NOT EXISTS move_shard;
|
|
SET SEARCH_PATH = move_shard;
|
|
SET citus.shard_count TO 4;
|
|
SET citus.next_shard_id TO 100;
|
|
SET citus.shard_replication_factor TO 1;
|
|
SET citus.max_adaptive_executor_pool_size TO 1;
|
|
SELECT pg_backend_pid() as pid \gset
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
CREATE TABLE t(id int PRIMARY KEY, int_data int, data text);
|
|
CREATE INDEX index_failure ON t(id);
|
|
SELECT create_distributed_table('t', 'id');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
CREATE VIEW shards_in_workers AS
|
|
SELECT shardid,
|
|
(CASE WHEN nodeport = :worker_1_port THEN 'worker1' ELSE 'worker2' END) AS worker
|
|
FROM pg_dist_placement NATURAL JOIN pg_dist_node
|
|
WHERE shardstate != 4
|
|
ORDER BY 1,2 ASC;
|
|
-- Insert some data
|
|
INSERT INTO t SELECT x, x+1, MD5(random()::text) FROM generate_series(1,100000) AS f(x);
|
|
-- Initial shard placements
|
|
SELECT * FROM shards_in_workers;
|
|
shardid | worker
|
|
---------------------------------------------------------------------
|
|
100 | worker2
|
|
101 | worker1
|
|
102 | worker2
|
|
103 | worker1
|
|
(4 rows)
|
|
|
|
-- failure on sanity checks
|
|
SELECT citus.mitmproxy('conn.onQuery(query="DROP TABLE IF EXISTS move_shard.t CASCADE").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
-- cancellation on sanity checks
|
|
SELECT citus.mitmproxy('conn.onQuery(query="DROP TABLE IF EXISTS move_shard.t CASCADE").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on move_shard table creation
|
|
SELECT citus.mitmproxy('conn.onQuery(query="CREATE TABLE move_shard.t").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
-- cancellation on move_shard table creation
|
|
SELECT citus.mitmproxy('conn.onQuery(query="CREATE TABLE move_shard.t").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on polling subscription state
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT count\(\*\) FROM pg_subscription_rel").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
-- cancellation on polling subscription state
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT count\(\*\) FROM pg_subscription_rel").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on getting subscriber state
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT sum").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
-- cancellation on getting subscriber state
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT sum").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on polling last write-ahead log location reported to origin WAL sender
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT min\(latest_end_lsn").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
-- cancellation on polling last write-ahead log location reported to origin WAL sender
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT min\(latest_end_lsn").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on dropping subscription
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^DROP SUBSCRIPTION").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
ERROR: connection to the remote node localhost:xxxxx failed with the following error: connection not open
|
|
-- cancellation on dropping subscription
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^DROP SUBSCRIPTION").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on creating the primary key
|
|
SELECT citus.mitmproxy('conn.onQuery(query="t_pkey").kill()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
-- cancellation on creating the primary key
|
|
SELECT citus.mitmproxy('conn.onQuery(query="t_pkey").cancel(' || :pid || ')');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
ERROR: canceling statement due to user request
|
|
-- failure on create index
|
|
SELECT citus.mitmproxy('conn.matches(b"CREATE INDEX").killall()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
ERROR: connection to the remote node localhost:xxxxx failed with the following error: connection not open
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- lets create few more indexes and fail with both
|
|
-- parallel mode and sequential mode
|
|
CREATE INDEX index_failure_2 ON t(id);
|
|
CREATE INDEX index_failure_3 ON t(id);
|
|
CREATE INDEX index_failure_4 ON t(id);
|
|
CREATE INDEX index_failure_5 ON t(id);
|
|
-- failure on the third create index
|
|
ALTER SYSTEM SET citus.max_adaptive_executor_pool_size TO 1;
|
|
SELECT pg_reload_conf();
|
|
pg_reload_conf
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
SELECT citus.mitmproxy('conn.matches(b"CREATE INDEX").killall()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
ERROR: connection to the remote node localhost:xxxxx failed with the following error: connection not open
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- failure on parallel create index
|
|
ALTER SYSTEM RESET citus.max_adaptive_executor_pool_size;
|
|
SELECT pg_reload_conf();
|
|
pg_reload_conf
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
SELECT citus.mitmproxy('conn.matches(b"CREATE INDEX").killall()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
WARNING: connection not open
|
|
CONTEXT: while executing command on localhost:xxxxx
|
|
ERROR: connection to the remote node localhost:xxxxx failed with the following error: connection not open
|
|
-- Verify that the shard is not moved and the number of rows are still 100k
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT * FROM shards_in_workers;
|
|
shardid | worker
|
|
---------------------------------------------------------------------
|
|
100 | worker2
|
|
101 | worker1
|
|
102 | worker2
|
|
103 | worker1
|
|
(4 rows)
|
|
|
|
SELECT count(*) FROM t;
|
|
count
|
|
---------------------------------------------------------------------
|
|
100000
|
|
(1 row)
|
|
|
|
-- Verify that shard can be moved after a temporary failure
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
mitmproxy
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT master_move_shard_placement(101, 'localhost', :worker_1_port, 'localhost', :worker_2_proxy_port);
|
|
master_move_shard_placement
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT * FROM shards_in_workers;
|
|
shardid | worker
|
|
---------------------------------------------------------------------
|
|
100 | worker2
|
|
101 | worker2
|
|
102 | worker2
|
|
103 | worker1
|
|
(4 rows)
|
|
|
|
SELECT count(*) FROM t;
|
|
count
|
|
---------------------------------------------------------------------
|
|
100000
|
|
(1 row)
|
|
|
|
DROP SCHEMA move_shard CASCADE ;
|
|
NOTICE: drop cascades to 2 other objects
|
|
DETAIL: drop cascades to table t
|
|
drop cascades to view shards_in_workers
|