[DRILL-8137] Prevent reading union inputs after cancellation request

• Type: Bug
• Status: Closed
• Priority: Critical
• Resolution: Fixed
• Affects Version/s: 1.19.0
• Fix Version/s: 1.20.0
• Assignee: Vova Vysotskyi
• Reporter: Vova Vysotskyi
• Created: 15/Feb/22 20:48
• Updated: 27/May/22 08:36
• Resolved: 27/May/22 08:36

Description

When running a union all query that has right side operators like join or aggregate, and limit on top of the union, such query will fail for the case when right input shouldn't read because the left one had required number of records for limit.

Example of such failing query (thanks to James Turton for helping to minimize it):

WITH foo AS
  (SELECT 1 AS a
   FROM cp.`/tpch/nation.parquet`
   UNION ALL SELECT 1 AS a
   FROM cp.`/tpch/nation.parquet`
   WHERE n_nationkey >
       (SELECT 1) )
SELECT *
FROM foo
LIMIT 1