[ARROw-6073] Reset Decimal128Builder in Finish().

kmjung · kmjung · commit 32fdb8f9d2ed · 2019-07-30T13:10:50.000-07:00
This change updates the FinishInternal implementation in
Decimal128Builder to reset the internal state at finish time.
diff --git a/cpp/src/arrow/array-test.cc b/cpp/src/arrow/array-test.cc
@@ -1755,6 +1755,7 @@ class DecimalTest : public ::testing::TestWithParam<int> {
 
     std::shared_ptr<Array> out;
     FinishAndCheckPadding(builder.get(), &out);
+    ASSERT_EQ(builder->length(), 0);
 
     std::vector<uint8_t> raw_bytes;
 
diff --git a/cpp/src/arrow/array/builder_decimal.cc b/cpp/src/arrow/array/builder_decimal.cc
@@ -69,7 +69,7 @@ Status Decimal128Builder::FinishInternal(std::shared_ptr<ArrayData>* out) {
   RETURN_NOT_OK(null_bitmap_builder_.Finish(&null_bitmap));
 
   *out = ArrayData::Make(type_, length_, {null_bitmap, data}, null_count_);
-
+  capacity_ = length_ = null_count_ = 0;
   return Status::OK();
 }
 

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ Status Decimal128Builder::FinishInternal(std::shared_ptr<ArrayData>* out) {`
`69`	`69`	`RETURN_NOT_OK(null_bitmap_builder_.Finish(&null_bitmap));`
`70`	`70`
`71`	`71`	`*out = ArrayData::Make(type_, length_, {null_bitmap, data}, null_count_);`
`72`		`-`
	`72`	`+ capacity_ = length_ = null_count_ = 0;`
`73`	`73`	`return Status::OK();`
`74`	`74`	`}`
`75`	`75`