[ARROW-12255] [Rust] [Ballista] Integrate scheduler with DataFusion

• Type: New Feature
• Status: Closed
• Priority: Major
• Resolution: Won't Fix
• Fix Version/s: 5.0.0
• Component/s: Rust - Ballista, Rust - DataFusion
• External issue URL: https://github.com/apache/arrow/issues/28066
• Assignee: Andy Grove
• Reporter: Andy Grove
• Created: 07/Apr/21 13:24
• Updated: 11/Jan/23 08:25
• Resolved: 25/Apr/21 13:56

Description

The Ballista scheduler breaks a query down into stages based on changes in partitioning int he plan, where each stage is broken down into tasks that can be executed concurrently.

Rather than trying to run all the partitions at once, Ballista executors process n concurrent tasks at a time and then request new tasks from the scheduler.

This approach would help DataFusion scale better and it would be ideal to use the same scheduler to scale across cores in DataFusion and across nodes in Ballista.