From 021c21698f9b76ca90e8e40a846c9d6db188e46a Mon Sep 17 00:00:00 2001 From: Adrian Garcia Badaracco <1755071+adriangb@users.noreply.github.com> Date: Fri, 11 Oct 2024 05:48:28 -0500 Subject: [PATCH] Add append_nulls to dictionary builders (#6542) * Add append_nulls to dictionary builders * Update arrow-array/src/builder/generic_bytes_dictionary_builder.rs Co-authored-by: Kyle Barron --------- Co-authored-by: Kyle Barron --- arrow-array/src/builder/generic_bytes_dictionary_builder.rs | 6 ++++++ arrow-array/src/builder/primitive_dictionary_builder.rs | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/arrow-array/src/builder/generic_bytes_dictionary_builder.rs b/arrow-array/src/builder/generic_bytes_dictionary_builder.rs index a327c622a753..bb0fb5e91be2 100644 --- a/arrow-array/src/builder/generic_bytes_dictionary_builder.rs +++ b/arrow-array/src/builder/generic_bytes_dictionary_builder.rs @@ -273,6 +273,12 @@ where self.keys_builder.append_null() } + /// Infallibly append `n` null slots into the builder + #[inline] + pub fn append_nulls(&mut self, n: usize) { + self.keys_builder.append_nulls(n) + } + /// Append an `Option` value into the builder /// /// # Panics diff --git a/arrow-array/src/builder/primitive_dictionary_builder.rs b/arrow-array/src/builder/primitive_dictionary_builder.rs index 35abe5ba5fb1..ac40f8a469d3 100644 --- a/arrow-array/src/builder/primitive_dictionary_builder.rs +++ b/arrow-array/src/builder/primitive_dictionary_builder.rs @@ -271,6 +271,12 @@ where self.keys_builder.append_null() } + /// Append `n` null slots into the builder + #[inline] + pub fn append_nulls(&mut self, n: usize) { + self.keys_builder.append_nulls(n) + } + /// Append an `Option` value into the builder /// /// # Panics