matthewgapp commented on code in PR #8840: URL: https://github.com/apache/arrow-datafusion/pull/8840#discussion_r1468201796
########## datafusion/sqllogictest/test_files/cte.slt: ########## @@ -38,3 +38,616 @@ Projection: NUMBERS.a, NUMBERS.b, NUMBERS.c physical_plan ProjectionExec: expr=[1 as a, 2 as b, 3 as c] --PlaceholderRowExec + + + +# enable recursive CTEs +statement ok +set datafusion.execution.enable_recursive_ctes = true; + +# trivial recursive CTE works +query I rowsort +WITH RECURSIVE nodes AS ( + SELECT 1 as id + UNION ALL + SELECT id + 1 as id + FROM nodes + WHERE id < 10 +) +SELECT * FROM nodes +---- +1 +10 +2 +3 +4 +5 +6 +7 +8 +9 + +# explain trivial recursive CTE +query TT +EXPLAIN WITH RECURSIVE nodes AS ( + SELECT 1 as id + UNION ALL + SELECT id + 1 as id + FROM nodes + WHERE id < 10 +) +SELECT * FROM nodes +---- +logical_plan +Projection: nodes.id +--SubqueryAlias: nodes +----RecursiveQuery: is_distinct=false +------Projection: Int64(1) AS id +--------EmptyRelation +------Projection: nodes.id + Int64(1) AS id +--------Filter: nodes.id < Int64(10) +----------TableScan: nodes +physical_plan +RecursiveQueryExec: is_distinct=false +--ProjectionExec: expr=[1 as id] +----PlaceholderRowExec +--CoalescePartitionsExec +----ProjectionExec: expr=[id@0 + 1 as id] +------CoalesceBatchesExec: target_batch_size=8192 +--------FilterExec: id@0 < 10 +----------RepartitionExec: partitioning=RoundRobinBatch(4), input_partitions=1 +------------WorkTableExec: name=nodes + +# setup +statement ok +CREATE EXTERNAL TABLE balance STORED as CSV WITH HEADER ROW LOCATION '../../testing/data/csv/r_cte_balance.csv' Review Comment: Thanks for the tip. I'll leave for files for now - since some of the files have a non-trivial number of values. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
