[ARROW-6659] [Rust] [DataFusion] Refactor of HashAggregateExec to support custom merge

• Type: Sub-task
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Fix Version/s: 0.16.0
• Component/s: Rust, Rust - DataFusion
• Labels: pull-request-available
• Assignee: Kyle McCarthy
• Reporter: Andy Grove
• Created: 22/Sep/19 17:39
• Updated: 16/Oct/19 02:15
• Resolved: 16/Oct/19 02:15

Description

HashAggregateExec current creates one HashPartition per input partition for the initial aggregate per partition, and then explicitly calls MergeExec and then creates another HashPartition for the final reduce operation.

This is fine for in-memory queries in DataFusion but is not extensible. For example, it is not possible to provide a different MergeExec implementation that would distribute queries to a cluster.

A better design would be to move the logic into the query planner so that the physical plan contains explicit steps such as:

 

- HashAggregate // final aggregate
  - MergeExec
    - HashAggregate // aggregate per partition
 

This would then make it easier to customize the plan in other projects, to support distributed execution:

 - HashAggregate // final aggregate
   - MergeExec
      - DistributedExec
         - HashAggregate // aggregate per partition

Issue Links

• links to: GitHub Pull Request #5648