Skip to content

Commit 87def04

Browse files
committed
actually write out the nan counts
1 parent e3cddc8 commit 87def04

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

parquet/src/file/page_index/column_index.rs

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -275,11 +275,18 @@ impl<T: ParquetValueType> WriteThrift for PrimitiveColumnIndex<T> {
275275
.write_thrift_field(writer, 6, last_field_id)?;
276276
}
277277
if self.definition_level_histograms.is_some() {
278-
self.definition_level_histograms
278+
last_field_id = self
279+
.definition_level_histograms
279280
.as_ref()
280281
.unwrap()
281282
.write_thrift_field(writer, 7, last_field_id)?;
282283
}
284+
if self.nan_counts.is_some() {
285+
self.nan_counts
286+
.as_ref()
287+
.unwrap()
288+
.write_thrift_field(writer, 8, last_field_id)?;
289+
}
283290
writer.write_struct_end()
284291
}
285292
}
@@ -458,11 +465,18 @@ impl WriteThrift for ByteArrayColumnIndex {
458465
.write_thrift_field(writer, 6, last_field_id)?;
459466
}
460467
if self.definition_level_histograms.is_some() {
461-
self.definition_level_histograms
468+
last_field_id = self
469+
.definition_level_histograms
462470
.as_ref()
463471
.unwrap()
464472
.write_thrift_field(writer, 7, last_field_id)?;
465473
}
474+
if self.nan_counts.is_some() {
475+
self.nan_counts
476+
.as_ref()
477+
.unwrap()
478+
.write_thrift_field(writer, 8, last_field_id)?;
479+
}
466480
writer.write_struct_end()
467481
}
468482
}

0 commit comments

Comments
 (0)