You are viewing a plain text version of this content. The canonical link for it is here.
Posted to jira@arrow.apache.org by "Remi Dettai (Jira)" <ji...@apache.org> on 2020/11/13 08:20:00 UTC
[jira] [Created] (ARROW-10577) [Rust][DataFusion] Hash Aggregator
stream finishes unexpectedly after going to Pending state
Remi Dettai created ARROW-10577:
-----------------------------------
Summary: [Rust][DataFusion] Hash Aggregator stream finishes unexpectedly after going to Pending state
Key: ARROW-10577
URL: https://issues.apache.org/jira/browse/ARROW-10577
Project: Apache Arrow
Issue Type: Bug
Components: Rust, Rust - DataFusion
Affects Versions: 2.0.0
Reporter: Remi Dettai
This happens when executing a DataFusion query plan with hash aggregation where the data source is not ready on the first call by the Executor, and the async state machine is passed to a _pending_ state
In the {{Stream}} implem of {{GroupedHashAggregateStream}} and {{HashAggregateStream}}, the state is set to {{self.finished = true}} on the first call to {{poll_next()}}. If the inner stream is {{Poll::Pending}} on the first call, this means that the next call resolves to {{Poll::Ready(None)}}, thus finishing the stream instead of actually consuming the inner data.
I think that it does not happen with most current sources because they never trigger the {{Poll::Pending}} state. Parquet is implemented with a blocking call inside {{poll_next()}} (which is also problematic but an other issue), Memory yields directly, and CSV also always yields {{Poll::Ready}}
An analysis should be performed on all physical plans to check if the issue occurs in other places.
--
This message was sent by Atlassian Jira
(v8.3.4#803005)