ARROW-12504: Buffer::from_slice_ref set correct capacity

apache · Apr 22, 2021 · b03f2ba · b03f2ba
1 parent 74d3567
commit b03f2ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arrow/src/buffer/immutable.rs b/arrow/src/buffer/immutable.rs
@@ -55,8 +55,8 @@ impl Buffer {
     /// Initializes a [Buffer] from a slice of items.
     pub fn from_slice_ref<U: ArrowNativeType, T: AsRef<[U]>>(items: &T) -> Self {
         let slice = items.as_ref();
-        let len = slice.len();
-        let mut buffer = MutableBuffer::with_capacity(len);
+        let capacity = slice.len() * std::mem::size_of::<T>();
+        let mut buffer = MutableBuffer::with_capacity(capacity);
         buffer.extend_from_slice(slice);
         buffer.into()
     }