rok commented on code in PR #8029:
URL: https://github.com/apache/arrow-rs/pull/8029#discussion_r2247753605
##########
parquet/src/arrow/arrow_writer/mod.rs:
##########
@@ -402,6 +402,25 @@ impl<W: Write + Send> ArrowWriter<W> {
pub fn close(mut self) -> Result<crate::format::FileMetaData> {
self.finish()
}
+
+ /// Create a new row group writer and return its column writers.
+ pub fn get_column_writers(&mut self) -> Result<Vec<ArrowColumnWriter>> {
+ let _ = self.flush();
+ let in_progress = self
+ .row_group_writer_factory
+ .create_row_group_writer(self.writer.flushed_row_groups().len())?;
+ Ok(in_progress.writers)
+ }
+
+ /// Append the given column chunks to the current row group.
+ pub fn append_to_row_groups(&mut self, chunks: Vec<ArrowColumnChunk>) ->
Result<()> {
Review Comment:
Suggested seems good!
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscr...@arrow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
