mirror of https://github.com/citusdata/citus.git
70 lines
3.0 KiB
SQL
70 lines
3.0 KiB
SQL
--
|
|
-- failure_mx_metadata_sync.sql
|
|
--
|
|
CREATE SCHEMA IF NOT EXISTS mx_metadata_sync;
|
|
SET SEARCH_PATH = mx_metadata_sync;
|
|
SET citus.shard_count TO 2;
|
|
SET citus.next_shard_id TO 16000000;
|
|
SET citus.shard_replication_factor TO 1;
|
|
SET citus.replication_model TO 'streaming';
|
|
|
|
SELECT pg_backend_pid() as pid \gset
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
|
|
CREATE TABLE t1 (id int PRIMARY KEY);
|
|
SELECT create_distributed_table('t1', 'id');
|
|
INSERT INTO t1 SELECT x FROM generate_series(1,100) AS f(x);
|
|
|
|
-- Initial metadata status
|
|
SELECT hasmetadata FROM pg_dist_node WHERE nodeport=:worker_2_proxy_port;
|
|
|
|
-- Failure to set groupid in the worker
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^UPDATE pg_dist_local_group SET groupid").cancel(' || :pid || ')');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^UPDATE pg_dist_local_group SET groupid").kill()');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
|
|
-- Failure to drop all tables in pg_dist_partition
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT worker_drop_distributed_table").cancel(' || :pid || ')');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^SELECT worker_drop_distributed_table").kill()');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
|
|
-- Failure to truncate pg_dist_node in the worker
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^TRUNCATE pg_dist_node CASCADE").cancel(' || :pid || ')');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^TRUNCATE pg_dist_node CASCADE").kill()');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
|
|
-- Failure to populate pg_dist_node in the worker
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^INSERT INTO pg_dist_node").cancel(' || :pid || ')');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
SELECT citus.mitmproxy('conn.onQuery(query="^INSERT INTO pg_dist_node").kill()');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
|
|
-- Verify that coordinator knows worker does not have valid metadata
|
|
SELECT hasmetadata FROM pg_dist_node WHERE nodeport=:worker_2_proxy_port;
|
|
|
|
-- Verify we can sync metadata after unsuccessful attempts
|
|
SELECT citus.mitmproxy('conn.allow()');
|
|
SELECT start_metadata_sync_to_node('localhost', :worker_2_proxy_port);
|
|
SELECT hasmetadata FROM pg_dist_node WHERE nodeport=:worker_2_proxy_port;
|
|
|
|
-- Check failures on DDL command propagation
|
|
CREATE TABLE t2 (id int PRIMARY KEY);
|
|
|
|
SELECT citus.mitmproxy('conn.onParse(query="^INSERT INTO pg_dist_placement").kill()');
|
|
SELECT create_distributed_table('t2', 'id');
|
|
|
|
SELECT citus.mitmproxy('conn.onParse(query="^INSERT INTO pg_dist_shard").cancel(' || :pid || ')');
|
|
SELECT create_distributed_table('t2', 'id');
|
|
|
|
-- Verify that the table was not distributed
|
|
SELECT count(*) > 0 AS is_table_distributed
|
|
FROM pg_dist_partition
|
|
WHERE logicalrelid='t2'::regclass;
|
|
|
|
DROP TABLE t1;
|
|
DROP TABLE t2;
|
|
DROP SCHEMA mx_metadata_sync CASCADE;
|