Skip to content

Commit 1eb63ec

Browse files
committed
metadata_size_hint in incremental scan
1 parent a294d1a commit 1eb63ec

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

crates/iceberg/src/arrow/incremental.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ async fn process_incremental_append_task(
6262
task: AppendedFileScanTask,
6363
batch_size: Option<usize>,
6464
file_io: FileIO,
65+
metadata_size_hint: Option<usize>,
6566
) -> Result<ArrowRecordBatchStream> {
6667
let mut virtual_columns = Vec::new();
6768

@@ -83,7 +84,7 @@ async fn process_incremental_append_task(
8384
file_io,
8485
true,
8586
arrow_reader_options,
86-
None,
87+
metadata_size_hint,
8788
task.base.file_size_in_bytes,
8889
)
8990
.await?;
@@ -246,6 +247,7 @@ impl StreamsInto<ArrowReader, UnzippedIncrementalBatchRecordStream>
246247
channel::<Result<RecordBatch>>(reader.concurrency_limit_data_files);
247248

248249
let batch_size = reader.batch_size;
250+
let metadata_size_hint = reader.metadata_size_hint;
249251

250252
let (append_stream, delete_stream) = self;
251253

@@ -259,7 +261,7 @@ impl StreamsInto<ArrowReader, UnzippedIncrementalBatchRecordStream>
259261
async move {
260262
spawn(async move {
261263
let record_batch_stream =
262-
process_incremental_append_task(append_task, batch_size, file_io)
264+
process_incremental_append_task(append_task, batch_size, file_io, metadata_size_hint)
263265
.await;
264266

265267
process_record_batch_stream(

0 commit comments

Comments
 (0)