Implement ReserveData(int64_t) method for BinaryBuilder

apache · Jan 16, 2018 · de318f4 · de318f4
1 parent e0434e6
commit de318f4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cpp/src/arrow/builder.cc b/cpp/src/arrow/builder.cc
@@ -1222,10 +1222,14 @@ Status BinaryBuilder::Init(int64_t elements) {
 Status BinaryBuilder::Resize(int64_t capacity) {
   DCHECK_LT(capacity, std::numeric_limits<int32_t>::max());
   // one more then requested for offsets
-  RETURN_NOT_OK(offsets_builder_.Resize((capacity + 1) * sizeof(int64_t)));
-  RETURN_NOT_OK(value_data_builder_.Resize(capacity * sizeof(int64_t)));
+  RETURN_NOT_OK(offsets_builder_.Resize((capacity + 1) * sizeof(int32_t)));
   return ArrayBuilder::Resize(capacity);
 }
+
+Status BinaryBuilder::ReserveData(int64_t capacity) {
+  DCHECK_LT(capacity, std::numeric_limits<int32_t>::max());
+  return value_data_builder_.Resize(capacity * sizeof(int64_t));
+}
 
 Status BinaryBuilder::AppendNextOffset() {
   const int64_t num_bytes = value_data_builder_.length();