external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

some more

pull/7626/head
paragjain 2024-06-14 17:02:13 +00:00 committed by Teja Mupparti
parent 76f68f47c4
commit e62ae64d00
2 changed files with 30 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/test/regress/expected/merge.out
View File

@ -27,7 +27,6 @@ SELECT 1 FROM master_add_node('localhost', :master_port, groupid => 0);
1 1
(1 row) (1 row)
ALTER SEQUENCE pg_catalog.pg_dist_colocationid_seq RESTART 13000;
RESET client_min_messages; RESET client_min_messages;
CREATE TABLE source CREATE TABLE source
( (
@ -1844,6 +1843,7 @@ SELECT compare_tables();
ROLLBACK; ROLLBACK;
-- let's create source and target table -- let's create source and target table
ALTER SEQUENCE pg_catalog.pg_dist_colocationid_seq RESTART 13000;
CREATE TABLE source_pushdowntest (id integer); CREATE TABLE source_pushdowntest (id integer);
CREATE TABLE target_pushdowntest (id integer ); CREATE TABLE target_pushdowntest (id integer );
-- let's distribute both table on id field -- let's distribute both table on id field
@ -1861,12 +1861,22 @@ SELECT create_distributed_table('target_pushdowntest', 'id');
-- we are doing this operation on single node setup let's figure out colocation id of both tables -- we are doing this operation on single node setup let's figure out colocation id of both tables
-- both has same colocation id so both are colocated. -- both has same colocation id so both are colocated.
select colocationid,logicalrelid from pg_dist_partition where logicalrelid = 'source_pushdowntest'::regclass OR logicalrelid = 'target_pushdowntest'::regclass; WITH colocations AS (
colocationid | logicalrelid SELECT colocationid
FROM pg_dist_partition
WHERE logicalrelid = 'source_pushdowntest'::regclass
OR logicalrelid = 'target_pushdowntest'::regclass
)
SELECT
CASE
WHEN COUNT(DISTINCT colocationid) = 1 THEN 'Same'
ELSE 'Different'
END AS colocation_status
FROM colocations;
colocation_status
--------------------------------------------------------------------- ---------------------------------------------------------------------
13000 | source_pushdowntest Same
13000 | target_pushdowntest (1 row)
(2 rows)
SET client_min_messages TO DEBUG1; SET client_min_messages TO DEBUG1;
-- Test 1 : tables are colocated AND query is multisharded AND Join On distributed column : should push down to workers. -- Test 1 : tables are colocated AND query is multisharded AND Join On distributed column : should push down to workers.

16
src/test/regress/sql/merge.sql
View File

@ -23,7 +23,6 @@ SET citus.shard_replication_factor TO 1;
SET citus.max_adaptive_executor_pool_size TO 1; SET citus.max_adaptive_executor_pool_size TO 1;
SET client_min_messages = warning; SET client_min_messages = warning;
SELECT 1 FROM master_add_node('localhost', :master_port, groupid => 0); SELECT 1 FROM master_add_node('localhost', :master_port, groupid => 0);
ALTER SEQUENCE pg_catalog.pg_dist_colocationid_seq RESTART 13000;
RESET client_min_messages; RESET client_min_messages;
CREATE TABLE source CREATE TABLE source
@ -1209,6 +1208,7 @@ ROLLBACK;
-- let's create source and target table -- let's create source and target table
ALTER SEQUENCE pg_catalog.pg_dist_colocationid_seq RESTART 13000;
CREATE TABLE source_pushdowntest (id integer); CREATE TABLE source_pushdowntest (id integer);
CREATE TABLE target_pushdowntest (id integer ); CREATE TABLE target_pushdowntest (id integer );
@ -1218,7 +1218,19 @@ SELECT create_distributed_table('target_pushdowntest', 'id');
-- we are doing this operation on single node setup let's figure out colocation id of both tables -- we are doing this operation on single node setup let's figure out colocation id of both tables
-- both has same colocation id so both are colocated. -- both has same colocation id so both are colocated.
select colocationid,logicalrelid from pg_dist_partition where logicalrelid = 'source_pushdowntest'::regclass OR logicalrelid = 'target_pushdowntest'::regclass; WITH colocations AS (
SELECT colocationid
FROM pg_dist_partition
WHERE logicalrelid = 'source_pushdowntest'::regclass
OR logicalrelid = 'target_pushdowntest'::regclass
)
SELECT
CASE
WHEN COUNT(DISTINCT colocationid) = 1 THEN 'Same'
ELSE 'Different'
END AS colocation_status
FROM colocations;
SET client_min_messages TO DEBUG1; SET client_min_messages TO DEBUG1;
-- Test 1 : tables are colocated AND query is multisharded AND Join On distributed column : should push down to workers. -- Test 1 : tables are colocated AND query is multisharded AND Join On distributed column : should push down to workers.