--
-- MULTI_CROSS_SHARD
--
-- Tests to log cross shard queries according to error log level
--
-- Create a distributed table and add data to it
CREATE TABLE multi_task_table
(
	id int,
	name varchar(20)
);
SELECT create_distributed_table('multi_task_table', 'id');
 create_distributed_table
---------------------------------------------------------------------

(1 row)

INSERT INTO multi_task_table VALUES(1, 'elem_1');
INSERT INTO multi_task_table VALUES(2, 'elem_2');
INSERT INTO multi_task_table VALUES(3, 'elem_3');
-- Shouldn't log anything when the log level is 'off'
SHOW citus.multi_task_query_log_level;
 citus.multi_task_query_log_level
---------------------------------------------------------------------
 off
(1 row)

SELECT * FROM multi_task_table ORDER BY 1;
 id |  name
---------------------------------------------------------------------
  1 | elem_1
  2 | elem_2
  3 | elem_3
(3 rows)

-- Get messages with the log level 'notice'
SET citus.multi_task_query_log_level TO notice;
SELECT * FROM multi_task_table ORDER BY 1;
NOTICE:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
 id |  name
---------------------------------------------------------------------
  1 | elem_1
  2 | elem_2
  3 | elem_3
(3 rows)

SELECT AVG(id) AS avg_id FROM multi_task_table;
NOTICE:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
       avg_id
---------------------------------------------------------------------
 2.0000000000000000
(1 row)

-- Get messages with the log level 'error'
SET citus.multi_task_query_log_level TO error;
SELECT * FROM multi_task_table;
ERROR:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
-- Check the log message with INSERT INTO ... SELECT
CREATE TABLE raw_table
(
    id int,
    order_count int
);
CREATE TABLE summary_table
(
    id int,
    order_sum BIGINT
);
SELECT create_distributed_table('raw_table', 'id');
 create_distributed_table
---------------------------------------------------------------------

(1 row)

SELECT create_distributed_table('summary_table', 'id');
 create_distributed_table
---------------------------------------------------------------------

(1 row)

INSERT INTO raw_table VALUES(1, '15');
INSERT INTO raw_table VALUES(2, '15');
INSERT INTO raw_table VALUES(3, '15');
INSERT INTO raw_table VALUES(1, '20');
INSERT INTO raw_table VALUES(2, '25');
INSERT INTO raw_table VALUES(3, '35');
--  Should notice user that the query is multi-task one
SET citus.multi_task_query_log_level TO notice;
INSERT INTO summary_table SELECT id, SUM(order_count) FROM raw_table GROUP BY id;
NOTICE:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
-- Should error out since the query is multi-task one
SET citus.multi_task_query_log_level TO error;
INSERT INTO summary_table SELECT id, SUM(order_count) FROM raw_table GROUP BY id;
ERROR:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
-- Shouldn't error out since it is a single task insert-into select query
INSERT INTO summary_table SELECT id, SUM(order_count) FROM raw_table WHERE id = 1 GROUP BY id;
-- Should have four rows (three rows from the query without where and the one from with where)
SET citus.multi_task_query_log_level to DEFAULT;
SELECT * FROM summary_table ORDER BY 1,2;
 id | order_sum
---------------------------------------------------------------------
  1 |        35
  1 |        35
  2 |        40
  3 |        50
(4 rows)

-- Set log-level to different levels inside the transaction
BEGIN;
--  Should notice user that the query is multi-task one
SET citus.multi_task_query_log_level TO notice;
INSERT INTO summary_table SELECT id, SUM(order_count) FROM raw_table GROUP BY id;
NOTICE:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
-- Should error out since the query is multi-task one
SET citus.multi_task_query_log_level TO error;
INSERT INTO summary_table SELECT id, SUM(order_count) FROM raw_table GROUP BY id;
ERROR:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
ROLLBACK;
-- Should have only four rows since the transaction is rollbacked.
SET citus.multi_task_query_log_level to DEFAULT;
SELECT * FROM summary_table ORDER BY 1,2;
 id | order_sum
---------------------------------------------------------------------
  1 |        35
  1 |        35
  2 |        40
  3 |        50
(4 rows)

-- Test router-select query
SET citus.multi_task_query_log_level TO notice;
-- Shouldn't log since it is a router select query
SELECT * FROM raw_table WHERE ID = 1;
 id | order_count
---------------------------------------------------------------------
  1 |          15
  1 |          20
(2 rows)

-- Task tracker query test
CREATE TABLE tt1
(
	id int,
	name varchar(20)
);
CREATE TABLE tt2
(
	id int,
	name varchar(20),
	count bigint
);
SELECT create_distributed_table('tt1', 'id');
 create_distributed_table
---------------------------------------------------------------------

(1 row)

SELECT create_distributed_table('tt2', 'name');
 create_distributed_table
---------------------------------------------------------------------

(1 row)

INSERT INTO tt1 VALUES(1, 'Ahmet');
INSERT INTO tt1 VALUES(2, 'Mehmet');
INSERT INTO tt2 VALUES(1, 'Ahmet', 5);
INSERT INTO tt2 VALUES(2, 'Mehmet', 15);
-- Should notice since it is a task-tracker query
SET citus.task_executor_type to "task-tracker";
SELECT tt1.id, tt2.count from tt1,tt2 where tt1.id = tt2.id;
NOTICE:  multi-task query about to be executed
HINT:  Queries are split to multiple tasks if they have to be split into several queries on the workers.
 id | count
---------------------------------------------------------------------
  1 |     5
  2 |    15
(2 rows)

SET citus.task_executor_type to DEFAULT;
DROP TABLE tt2;
DROP TABLE tt1;
DROP TABLE multi_task_table;
DROP TABLE raw_table;
DROP TABLE summary_table;