Skip to content

Commit cd0e1dd

Browse files
author
Devdutt Shenoi
committed
fix: don't keep the mapping after finish
1 parent 0905dfb commit cd0e1dd

1 file changed

Lines changed: 5 additions & 12 deletions

File tree

src/parseable/streams.rs

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -358,18 +358,11 @@ impl Stream {
358358
}
359359

360360
pub fn flush(&self) {
361-
let mut disk_writers = {
362-
let mut writer = self.writer.lock().unwrap();
363-
// Flush memory
364-
writer.mem.clear();
365-
// Take schema -> disk writer mapping
366-
std::mem::take(&mut writer.disk)
367-
};
368-
369-
// Flush disk
370-
for writer in disk_writers.values_mut() {
371-
_ = writer.finish();
372-
}
361+
let mut writer = self.writer.lock().unwrap();
362+
// Flush memory
363+
writer.mem.clear();
364+
// Drop schema -> disk writer mapping, triggers flush to disk
365+
writer.disk.drain();
373366
}
374367

375368
fn parquet_writer_props(

0 commit comments

Comments
 (0)