--
-- MULTI_JOIN_ORDER_ADDITIONAL
--
-- Set configuration to print table join order and pruned shards
SET citus.log_multi_join_order TO TRUE;
SET client_min_messages TO DEBUG2;
-- The following query checks that we can correctly handle self-joins
EXPLAIN SELECT l1.l_quantity FROM lineitem l1, lineitem l2
	WHERE l1.l_orderkey = l2.l_orderkey AND l1.l_quantity > 5;
LOG:  join order: [ "lineitem" ][ local partition join "lineitem" ]
DEBUG:  join prunable for intervals [13921,14947] and [11554,13920]
DEBUG:  join prunable for intervals [13921,14947] and [8997,11554]
DEBUG:  join prunable for intervals [13921,14947] and [4965,5986]
DEBUG:  join prunable for intervals [13921,14947] and [2497,4964]
DEBUG:  join prunable for intervals [13921,14947] and [1,2496]
DEBUG:  join prunable for intervals [11554,13920] and [13921,14947]
DEBUG:  join prunable for intervals [11554,13920] and [4965,5986]
DEBUG:  join prunable for intervals [11554,13920] and [2497,4964]
DEBUG:  join prunable for intervals [11554,13920] and [1,2496]
DEBUG:  join prunable for intervals [8997,11554] and [13921,14947]
DEBUG:  join prunable for intervals [8997,11554] and [4965,5986]
DEBUG:  join prunable for intervals [8997,11554] and [2497,4964]
DEBUG:  join prunable for intervals [8997,11554] and [1,2496]
DEBUG:  join prunable for intervals [4965,5986] and [13921,14947]
DEBUG:  join prunable for intervals [4965,5986] and [11554,13920]
DEBUG:  join prunable for intervals [4965,5986] and [8997,11554]
DEBUG:  join prunable for intervals [4965,5986] and [2497,4964]
DEBUG:  join prunable for intervals [4965,5986] and [1,2496]
DEBUG:  join prunable for intervals [2497,4964] and [13921,14947]
DEBUG:  join prunable for intervals [2497,4964] and [11554,13920]
DEBUG:  join prunable for intervals [2497,4964] and [8997,11554]
DEBUG:  join prunable for intervals [2497,4964] and [4965,5986]
DEBUG:  join prunable for intervals [2497,4964] and [1,2496]
DEBUG:  join prunable for intervals [1,2496] and [13921,14947]
DEBUG:  join prunable for intervals [1,2496] and [11554,13920]
DEBUG:  join prunable for intervals [1,2496] and [8997,11554]
DEBUG:  join prunable for intervals [1,2496] and [4965,5986]
DEBUG:  join prunable for intervals [1,2496] and [2497,4964]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

-- Update configuration to treat lineitem and orders tables as large
SET citus.large_table_shard_count TO 2;
SET client_min_messages TO LOG;
-- The following queries check that we correctly handle joins and OR clauses. In
-- particular, these queries check that we factorize out OR clauses if possible,
-- and that we default to a cartesian product otherwise.
EXPLAIN SELECT count(*) FROM lineitem, orders
	WHERE (l_orderkey = o_orderkey AND l_quantity > 5)
	OR (l_orderkey = o_orderkey AND l_quantity < 10);
LOG:  join order: [ "lineitem" ][ local partition join "orders" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

EXPLAIN SELECT l_quantity FROM lineitem, orders
	WHERE (l_orderkey = o_orderkey OR l_quantity > 5);
LOG:  join order: [ "lineitem" ][ cartesian product "orders" ]
ERROR:  cannot perform distributed planning on this query
DETAIL:  Cartesian products are currently unsupported
-- The below queries modify the partition method in pg_dist_partition. We thus
-- begin a transaction here so the changes don't impact any other parallel
-- running tests.
BEGIN;
-- Validate that we take into account the partition method when building the
-- join-order plan.
UPDATE pg_dist_partition SET partmethod = 'h' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'lineitem');
EXPLAIN SELECT count(*) FROM orders, lineitem
	WHERE o_orderkey = l_orderkey;
LOG:  join order: [ "orders" ][ single partition join "lineitem" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

-- Verify we handle local joins between two hash-partitioned tables.
UPDATE pg_dist_partition SET partmethod = 'h' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'orders');
EXPLAIN SELECT count(*) FROM orders, lineitem
	WHERE o_orderkey = l_orderkey;
LOG:  join order: [ "orders" ][ local partition join "lineitem" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

UPDATE pg_dist_partition SET partmethod = 'a' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'lineitem');
UPDATE pg_dist_partition SET partmethod = 'a' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'orders');
-- Validate that we can handle broadcast joins with hash-partitioned tables.
UPDATE pg_dist_partition SET partmethod = 'h' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'customer');
EXPLAIN SELECT count(*) FROM customer, nation
	WHERE c_nationkey = n_nationkey;
LOG:  join order: [ "customer" ][ broadcast join "nation" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

UPDATE pg_dist_partition SET partmethod = 'a' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'customer');
-- Update the large table shard count for all the following tests.
SET citus.large_table_shard_count TO 1;
-- Validate that we don't use a single-partition join method for a hash
-- re-partitioned table, thus preventing a partition of just the customer table.
EXPLAIN SELECT count(*) FROM orders, lineitem, customer
	WHERE o_custkey = l_partkey AND o_custkey = c_nationkey;
LOG:  join order: [ "orders" ][ dual partition join "lineitem" ][ dual partition join "customer" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

-- Validate that we don't chose a single-partition join method with a
-- hash-partitioned base table
UPDATE pg_dist_partition SET partmethod = 'h' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'customer');
EXPLAIN SELECT count(*) FROM orders, customer
	WHERE c_custkey = o_custkey;
LOG:  join order: [ "orders" ][ dual partition join "customer" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

UPDATE pg_dist_partition SET partmethod = 'a' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'customer');
-- Validate that we can re-partition a hash partitioned table to join with a
-- range partitioned one.
UPDATE pg_dist_partition SET partmethod = 'h' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'orders');
EXPLAIN SELECT count(*) FROM orders, customer
	WHERE c_custkey = o_custkey;
LOG:  join order: [ "orders" ][ single partition join "customer" ]
                              QUERY PLAN                              
----------------------------------------------------------------------
 explain statements for distributed queries are currently unsupported
(1 row)

UPDATE pg_dist_partition SET partmethod = 'a' WHERE
       logicalrelid = (SELECT relfilenode FROM pg_class WHERE relname = 'orders');
COMMIT;
-- Reset client logging level to its previous value
SET client_min_messages TO NOTICE;