Skip to content

Commit 0ac042b

Browse files
authored
fix: bug with nulls in Decimal96 (#49)
1 parent 9e3dd6c commit 0ac042b

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

  • parquet/src/arrow/arrow_writer

parquet/src/arrow/arrow_writer/mod.rs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1051,7 +1051,15 @@ fn write_leaf(writer: &mut ColumnWriter<'_>, levels: &ArrayLevels) -> Result<usi
10511051
let array = column.as_primitive::<Decimal128Type>();
10521052
let int96_values: Vec<Int96> =
10531053
array.values().iter().map(|v| i128_to_int96(*v)).collect();
1054-
typed.write_batch(&int96_values, levels.def_levels(), levels.rep_levels())
1054+
typed.write_batch_internal(
1055+
&int96_values,
1056+
Some(levels.non_null_indices()),
1057+
levels.def_levels(),
1058+
levels.rep_levels(),
1059+
None,
1060+
None,
1061+
None,
1062+
)
10551063
}
10561064
_ => unreachable!("INT96 column writer only supports Decimal128 for Decimal96"),
10571065
},

0 commit comments

Comments
 (0)