Skip to content

Commit c2747a3

Browse files
author
Rickard Lundin
committed
fix
1 parent 7a495ec commit c2747a3

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

src/chunked/recordbatch_output.rs

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ impl DeltaOut {
154154

155155
let mut writer = RecordBatchWriter::for_table(&table).expect("Failed to make RecordBatchWriter")
156156
.with_writer_properties(writer_properties);
157-
157+
158158
let metadata = table
159159
.metadata()
160160
.expect("Failed to get metadata for the table");
@@ -175,7 +175,12 @@ impl DeltaOut {
175175
// let nrb=rb;
176176
info!("nrb rows={} nrb col len {} batchcount={} usize={}",nrb.num_rows(),nrb.columns().len(),writer.buffered_record_batch_count(),writer.buffer_len());
177177
writer.write(nrb).await.expect("writing");
178-
178+
if (writer.buffered_record_batch_count()>200) {
179+
writer
180+
.flush()
181+
.await
182+
.expect("Failed to flush write");
183+
}
179184
// writer.write_with_mode(nrb,WriteMode::MergeSchema);
180185

181186
}

0 commit comments

Comments
 (0)