Address a remaining comment in apache#41352

zanmato1984 · May 14, 2024 · 4821398 · 4821398
1 parent dcc3b7a
commit 4821398
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/cpp/src/arrow/compute/row/grouper.cc b/cpp/src/arrow/compute/row/grouper.cc
@@ -600,6 +600,7 @@ struct GrouperFastImpl : public Grouper {
   }
 
   Status Reset() override {
+    ARROW_DCHECK_EQ(temp_stack_.AllocatedSize(), 0);
     rows_.Clean();
     rows_minibatch_.Clean();
     map_.cleanup();

diff --git a/cpp/src/arrow/compute/util_internal.h b/cpp/src/arrow/compute/util_internal.h
@@ -40,6 +40,8 @@ class ARROW_EXPORT TempVectorStack {
  public:
   Status Init(MemoryPool* pool, int64_t size);
 
+  int64_t AllocatedSize() const { return top_; }
+
  private:
   static int64_t EstimatedAllocationSize(int64_t size) {
     return PaddedAllocationSize(size) + 2 * sizeof(uint64_t);