Skip to content

Commit ea35831

Browse files
committed
fix: set correct number of rows
1 parent b06ebd2 commit ea35831

File tree

1 file changed

+2
-1
lines changed
  • src/query/storages/fuse/src/io/read/block/parquet

1 file changed

+2
-1
lines changed

src/query/storages/fuse/src/io/read/block/parquet/mod.rs

+2-1
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,7 @@ impl BlockReader {
9595
let mut blocks = Vec::with_capacity(record_batches.len());
9696

9797
for record_batch in record_batches {
98+
let num_rows_record_batch = record_batch.num_rows();
9899
let mut columns = Vec::with_capacity(self.projected_schema.fields.len());
99100
for ((i, field), column_node) in self
100101
.projected_schema
@@ -151,7 +152,7 @@ impl BlockReader {
151152
};
152153
columns.push(BlockEntry::new(data_type, value));
153154
}
154-
blocks.push(DataBlock::new(columns, num_rows));
155+
blocks.push(DataBlock::new(columns, num_rows_record_batch));
155156
}
156157

157158
Ok(blocks)

0 commit comments

Comments
 (0)