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 <kylebarron2@gmail.com> --------- Co-authored-by: Kyle Barron <kylebarron2@gmail.com>
apache · Oct 11, 2024 · 021c216 · 021c216
1 parent 4adbeee
commit 021c216
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git 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
@@ -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