Skip to content

Commit

Permalink
Ends ParquetRecordBatchStream when polling on StreamState::Error (#3404)
Browse files Browse the repository at this point in the history
* Remove unnecessary StreamState

* Return None for StreamState::Error
  • Loading branch information
viirya authored Dec 28, 2022
1 parent 8f99692 commit 513d543
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion parquet/src/arrow/async_reader/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -568,7 +568,7 @@ where
return Poll::Ready(Some(Err(e)));
}
},
StreamState::Error => return Poll::Pending,
StreamState::Error => return Poll::Ready(None), // Ends the stream as error happens.
}
}
}
Expand Down

0 comments on commit 513d543

Please sign in to comment.