[enhancement] partitioned agg node return error when increase reserva… (

apache#11012) * [enhancement] partitioned agg node return error when increase reservation failed Co-authored-by: yixiutt <yixiu@selectdb.com> Co-authored-by: Yongqiang YANG <98214048+dataroaring@users.noreply.github.com>
Henry2SS · Jul 21, 2022 · 7992ae8 · 7992ae8
1 parent e501ea3
commit 7992ae8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/be/src/exec/partitioned_aggregation_node.cc b/be/src/exec/partitioned_aggregation_node.cc
@@ -745,9 +745,12 @@ Status PartitionedAggregationNode::Partition::InitStreams() {
     RETURN_IF_ERROR(aggregated_row_stream->Init(parent->id(), true));
     bool got_buffer;
     RETURN_IF_ERROR(aggregated_row_stream->PrepareForWrite(&got_buffer));
-    DCHECK(got_buffer) << "Buffer included in reservation " << parent->_id << "\n"
-                       << parent->_buffer_pool_client.DebugString() << "\n"
-                       << parent->DebugString(2);
+    if (!got_buffer) {
+        LOG(WARNING) << "Buffer included in reservation " << parent->_id << "\n"
+                     << parent->_buffer_pool_client.DebugString() << "\n"
+                     << parent->DebugString(2);
+        return Status::BufferAllocFailed("PartitionedAggregationNode buffer prepare failed");
+    }
 
     if (!parent->is_streaming_preagg_) {
         unaggregated_row_stream.reset(new BufferedTupleStream3(