You are viewing a plain text version of this content. The canonical link for it is here.

Posted to jira@arrow.apache.org by "Mike Seddon (Jira)" <ji...@apache.org> on 2021/02/13 02:44:00 UTC

[jira] [Created] (ARROW-11616) [Rust][DataFusion] Expose collect_partitioned for DataFrame

Mike Seddon created ARROW-11616:
-----------------------------------

             Summary: [Rust][DataFusion] Expose collect_partitioned for DataFrame
                 Key: ARROW-11616
                 URL: https://issues.apache.org/jira/browse/ARROW-11616
             Project: Apache Arrow
          Issue Type: Improvement
          Components: Rust - DataFusion
            Reporter: Mike Seddon
            Assignee: Mike Seddon


The DataFrame API has a `collect` method which invokes the `collect(plan: Arc<dyn ExecutionPlan>) -> Result<Vec<RecordBatch>>` function which will collect records into a single vector of RecordBatches removing the partitioning via `MergeExec`.

The DataFrame should also expose the `collect_partitioned` method so that partitions can be maintained.

```
collect_partitioned(
    plan: Arc<dyn ExecutionPlan>,
) -> Result<Vec<Vec<RecordBatch>>> 
```



--
This message was sent by Atlassian Jira
(v8.3.4#803005)