--
-- MULTI_COLOCATED_SHARD_TRANSFER
--
-- These tables are created in multi_colocation_utils test
-- test repair
-- manually set shardstate as inactive
UPDATE pg_dist_shard_placement SET shardstate = 3 WHERE nodeport = :worker_2_port AND (shardid = 1300000 OR shardid = 1300004);
UPDATE pg_dist_shard_placement SET shardstate = 3 WHERE nodeport = :worker_2_port AND shardid = 1300016;
UPDATE pg_dist_shard_placement SET shardstate = 3 WHERE nodeport = :worker_2_port AND shardid = 1300020;
-- test repairing colocated shards
-- status before shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    colocationid = (SELECT colocationid FROM pg_dist_partition WHERE logicalrelid = 'table1_group1'::regclass)
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300000 | table1_group1 |    57638 |         1000 |          3
 1300000 | table1_group1 |    57637 |         1000 |          1
 1300001 | table1_group1 |    57637 |         1000 |          1
 1300001 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57638 |         1000 |          3
 1300004 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57638 |         1000 |          1
(16 rows)

-- repair colocated shards
SELECT master_copy_shard_placement(1300000, 'localhost', :worker_1_port, 'localhost', :worker_2_port);
 master_copy_shard_placement 
-----------------------------
 
(1 row)

-- status after shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    colocationid = (SELECT colocationid FROM pg_dist_partition WHERE logicalrelid = 'table1_group1'::regclass)
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300000 | table1_group1 |    57638 |         1000 |          1
 1300000 | table1_group1 |    57637 |         1000 |          1
 1300001 | table1_group1 |    57637 |         1000 |          1
 1300001 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57638 |         1000 |          1
(16 rows)

-- test repairing NOT colocated shard
-- status before shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    p.logicalrelid = 'table5_groupX'::regclass
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300016 | table5_groupx |    57637 |            0 |          1
 1300016 | table5_groupx |    57638 |            0 |          3
 1300017 | table5_groupx |    57637 |            0 |          1
 1300017 | table5_groupx |    57638 |            0 |          1
 1300018 | table5_groupx |    57637 |            0 |          1
 1300018 | table5_groupx |    57638 |            0 |          1
 1300019 | table5_groupx |    57638 |            0 |          1
 1300019 | table5_groupx |    57637 |            0 |          1
(8 rows)

-- repair NOT colocated shard
SELECT master_copy_shard_placement(1300016, 'localhost', :worker_1_port, 'localhost', :worker_2_port);
 master_copy_shard_placement 
-----------------------------
 
(1 row)

-- status after shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    p.logicalrelid = 'table5_groupX'::regclass
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300016 | table5_groupx |    57637 |            0 |          1
 1300016 | table5_groupx |    57638 |            0 |          1
 1300017 | table5_groupx |    57637 |            0 |          1
 1300017 | table5_groupx |    57638 |            0 |          1
 1300018 | table5_groupx |    57637 |            0 |          1
 1300018 | table5_groupx |    57638 |            0 |          1
 1300019 | table5_groupx |    57638 |            0 |          1
 1300019 | table5_groupx |    57637 |            0 |          1
(8 rows)

-- test repairing shard in append distributed table
-- status before shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    p.logicalrelid = 'table6_append'::regclass
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300020 | table6_append |    57637 |            0 |          1
 1300020 | table6_append |    57638 |            0 |          3
 1300021 | table6_append |    57637 |            0 |          1
 1300021 | table6_append |    57638 |            0 |          1
(4 rows)

-- repair  shard in append distributed table
SELECT master_copy_shard_placement(1300020, 'localhost', :worker_1_port, 'localhost', :worker_2_port);
 master_copy_shard_placement 
-----------------------------
 
(1 row)

-- status after shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    p.logicalrelid = 'table6_append'::regclass
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300020 | table6_append |    57637 |            0 |          1
 1300020 | table6_append |    57638 |            0 |          1
 1300021 | table6_append |    57637 |            0 |          1
 1300021 | table6_append |    57638 |            0 |          1
(4 rows)

-- test repair while all placements of one shard in colocation group is unhealthy
-- manually set shardstate as inactive
UPDATE pg_dist_shard_placement SET shardstate = 3 WHERE shardid = 1300000;
-- status before shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    colocationid = (SELECT colocationid FROM pg_dist_partition WHERE logicalrelid = 'table1_group1'::regclass)
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300000 | table1_group1 |    57638 |         1000 |          3
 1300000 | table1_group1 |    57637 |         1000 |          3
 1300001 | table1_group1 |    57637 |         1000 |          1
 1300001 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57638 |         1000 |          1
(16 rows)

-- repair while all placements of one shard in colocation group is unhealthy
SELECT master_copy_shard_placement(1300000, 'localhost', :worker_1_port, 'localhost', :worker_2_port);
ERROR:  source placement must be in finalized state
-- status after shard repair
SELECT s.shardid, s.logicalrelid::regclass, sp.nodeport, p.colocationid, sp.shardstate
FROM 
    pg_dist_partition p, pg_dist_shard s, pg_dist_shard_placement sp
WHERE
    p.logicalrelid = s.logicalrelid AND
    s.shardid = sp.shardid AND
    colocationid = (SELECT colocationid FROM pg_dist_partition WHERE logicalrelid = 'table1_group1'::regclass)
ORDER BY s.shardid;
 shardid | logicalrelid  | nodeport | colocationid | shardstate 
---------+---------------+----------+--------------+------------
 1300000 | table1_group1 |    57638 |         1000 |          3
 1300000 | table1_group1 |    57637 |         1000 |          3
 1300001 | table1_group1 |    57637 |         1000 |          1
 1300001 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57638 |         1000 |          1
 1300002 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57637 |         1000 |          1
 1300003 | table1_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57638 |         1000 |          1
 1300004 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57637 |         1000 |          1
 1300005 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57638 |         1000 |          1
 1300006 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57637 |         1000 |          1
 1300007 | table2_group1 |    57638 |         1000 |          1
(16 rows)