mirror of https://github.com/citusdata/citus.git
314 lines
7.0 KiB
SQL
314 lines
7.0 KiB
SQL
-- ===================================================================
|
|
-- test recursive planning functionality
|
|
-- ===================================================================
|
|
|
|
SET client_min_messages TO DEBUG1;
|
|
|
|
-- subqueries in FROM clause with LIMIT should be recursively planned
|
|
SELECT
|
|
user_id
|
|
FROM
|
|
(SELECT
|
|
DISTINCT users_table.user_id
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (1,2,3,4)
|
|
ORDER BY 1 DESC LIMIT 5
|
|
) as foo
|
|
ORDER BY 1 DESC;
|
|
|
|
|
|
-- subqueries in FROM clause with DISTINCT on non-partition key
|
|
-- should be recursively planned
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
DISTINCT users_table.value_1
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (1,2,3,4)
|
|
ORDER BY 1
|
|
) as foo
|
|
ORDER BY 1 DESC;
|
|
|
|
-- subqueries in FROM clause with GROUP BY on non-partition key
|
|
-- should be recursively planned
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
users_table.value_2, avg(value_1)
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (1,2,3,4)
|
|
GROUP BY users_table.value_2
|
|
ORDER BY 1 DESC
|
|
) as foo
|
|
ORDER BY 2 DESC, 1;
|
|
|
|
-- subqueries with only generate_series
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
events_table.value_2
|
|
FROM
|
|
events_table
|
|
WHERE
|
|
event_type IN (1,2,3,4)
|
|
ORDER BY 1 DESC
|
|
LIMIT 5
|
|
) as foo,
|
|
(
|
|
SELECT i FROM generate_series(0, 100) i
|
|
) as bar
|
|
WHERE foo.value_2 = bar.i
|
|
ORDER BY 2 DESC, 1;
|
|
|
|
-- subquery with aggregates without GROUP BY
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
count(*)
|
|
FROM
|
|
events_table
|
|
WHERE
|
|
event_type IN (1,2,3,4)
|
|
) as foo;
|
|
|
|
-- subquery having without GROUP BY
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
SUM(events_table.user_id)
|
|
FROM
|
|
events_table
|
|
WHERE
|
|
event_type IN (1,2,3,4)
|
|
HAVING
|
|
MIN(value_2) > 2
|
|
) as foo;
|
|
|
|
-- multiple subqueries in FROM clause should be replaced
|
|
-- and the final query is router query
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
users_table.value_2
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (1,2,3,4)
|
|
GROUP BY users_table.value_2
|
|
ORDER BY 1 DESC
|
|
) as foo,
|
|
(SELECT
|
|
users_table.value_3
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (5,6,7,8)
|
|
GROUP BY users_table.value_3
|
|
ORDER BY 1 DESC
|
|
) as bar
|
|
WHERE foo.value_2 = bar.value_3
|
|
ORDER BY 2 DESC, 1;
|
|
|
|
-- same query with alias in the subquery
|
|
SELECT
|
|
DISTINCT ON (citus) citus, postgres, citus + 1 as c1, postgres-1 as p1
|
|
FROM
|
|
(SELECT
|
|
users_table.value_2
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (1,2,3,4)
|
|
GROUP BY users_table.value_2
|
|
ORDER BY 1 DESC
|
|
) as foo(postgres),
|
|
(SELECT
|
|
users_table.user_id
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (5,6,7,8)
|
|
ORDER BY 1 DESC
|
|
) as bar (citus)
|
|
WHERE foo.postgres = bar.citus
|
|
ORDER BY 1 DESC, 2 DESC
|
|
LIMIT 3;
|
|
|
|
-- foo is replaced
|
|
-- and the final query is real-time
|
|
SELECT
|
|
*
|
|
FROM
|
|
(SELECT
|
|
users_table.value_2
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (1,2,3,4)
|
|
GROUP BY users_table.value_2
|
|
ORDER BY 1 DESC
|
|
) as foo,
|
|
(SELECT
|
|
users_table.user_id
|
|
FROM
|
|
users_table, events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
event_type IN (5,6,7,8)
|
|
ORDER BY 1 DESC
|
|
) as bar
|
|
WHERE foo.value_2 = bar.user_id
|
|
ORDER BY 2 DESC, 1 DESC
|
|
LIMIT 3;
|
|
|
|
-- subqueries in WHERE should be replaced
|
|
SELECT DISTINCT user_id
|
|
FROM users_table
|
|
WHERE
|
|
user_id IN (SELECT DISTINCT value_2 FROM users_table WHERE value_1 >= 1 AND value_1 <= 20 ORDER BY 1 LIMIT 5)
|
|
ORDER BY 1 DESC;
|
|
|
|
-- subquery in FROM -> FROM -> FROM should be replaced due to OFFSET
|
|
SELECT
|
|
DISTINCT user_id
|
|
FROM
|
|
(
|
|
SELECT users_table.user_id FROM users_table,
|
|
(
|
|
SELECT
|
|
event_type, user_id
|
|
FROM
|
|
(SELECT event_type, users_table.user_id FROM users_table,
|
|
(SELECT user_id, event_type FROM events_table WHERE value_2 < 3 OFFSET 3) as foo
|
|
WHERE foo.user_id = users_table.user_id
|
|
) bar
|
|
|
|
) as baz
|
|
WHERE baz.user_id = users_table.user_id
|
|
|
|
) as sub1
|
|
ORDER BY 1 DESC
|
|
LIMIT 3;
|
|
|
|
|
|
-- subquery in FROM -> FROM -> WHERE should be replaced due to LIMIT
|
|
SELECT user_id, array_length(events_table, 1)
|
|
FROM (
|
|
SELECT user_id, array_agg(event ORDER BY time) AS events_table
|
|
FROM (
|
|
SELECT
|
|
u.user_id, e.event_type::text AS event, e.time
|
|
FROM
|
|
users_table AS u,
|
|
events_table AS e
|
|
WHERE u.user_id = e.user_id AND
|
|
u.user_id IN
|
|
(
|
|
SELECT
|
|
user_id
|
|
FROM
|
|
users_table
|
|
WHERE value_2 >= 5
|
|
AND EXISTS (SELECT user_id FROM events_table WHERE event_type > 1 AND event_type <= 3 AND value_3 > 1 AND user_id = users_table.user_id)
|
|
AND NOT EXISTS (SELECT user_id FROM events_table WHERE event_type > 3 AND event_type <= 4 AND value_3 > 1 AND user_id = users_table.user_id)
|
|
LIMIT 5
|
|
)
|
|
) t
|
|
GROUP BY user_id
|
|
) q
|
|
ORDER BY 2 DESC, 1;
|
|
|
|
|
|
-- subquery (i.e., subquery_2) in WHERE->FROM should be replaced due to LIMIT
|
|
SELECT
|
|
user_id
|
|
FROM
|
|
users_table
|
|
WHERE
|
|
user_id IN
|
|
(
|
|
SELECT
|
|
user_id
|
|
FROM (
|
|
SELECT
|
|
subquery_1.user_id, count_pay
|
|
FROM
|
|
(
|
|
(SELECT
|
|
users_table.user_id,
|
|
'action=>1' AS event,
|
|
events_table.time
|
|
FROM
|
|
users_table,
|
|
events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
users_table.user_id >= 1 AND
|
|
users_table.user_id <= 3 AND
|
|
events_table.event_type > 1 AND events_table.event_type < 3
|
|
)
|
|
UNION
|
|
(SELECT
|
|
users_table.user_id,
|
|
'action=>2' AS event,
|
|
events_table.time
|
|
FROM
|
|
users_table,
|
|
events_table
|
|
WHERE
|
|
users_table.user_id = events_table.user_id AND
|
|
users_table.user_id >= 1 AND
|
|
users_table.user_id <= 3 AND
|
|
events_table.event_type > 2 AND events_table.event_type < 4
|
|
)
|
|
) AS subquery_1
|
|
LEFT JOIN
|
|
(SELECT
|
|
user_id,
|
|
COUNT(*) AS count_pay
|
|
FROM
|
|
users_table
|
|
WHERE
|
|
user_id >= 1 AND
|
|
user_id <= 3 AND
|
|
users_table.value_1 > 3 AND users_table.value_1 < 5
|
|
GROUP BY
|
|
user_id
|
|
HAVING
|
|
COUNT(*) > 1
|
|
LIMIT 10
|
|
|
|
) AS subquery_2
|
|
ON
|
|
subquery_1.user_id = subquery_2.user_id
|
|
GROUP BY
|
|
subquery_1.user_id,
|
|
count_pay) AS subquery_top
|
|
GROUP BY
|
|
count_pay, user_id
|
|
)
|
|
GROUP BY user_id
|
|
HAVING count(*) > 1 AND sum(value_2) > 29
|
|
ORDER BY 1;
|
|
|