[BugFix] Fix array map over flow (#55966)

Signed-off-by: before-Sunrise <[email protected]>
StarRocks · Feb 18, 2025 · cd1f63c · cd1f63c
1 parent 64d0f20
commit cd1f63c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/be/src/exprs/array_map_expr.cpp b/be/src/exprs/array_map_expr.cpp
@@ -192,6 +192,8 @@ StatusOr<ColumnPtr> ArrayMapExpr::evaluate_lambda_expr(ExprContext* context, Chu
         tmp_col->check_or_die();
         ASSIGN_OR_RETURN(column, tmp_col->replicate(aligned_offsets->get_data()));
         column = ColumnHelper::align_return_type(column, type().children[0], column->size(), true);
+
+        RETURN_IF_ERROR(column->capacity_limit_reached());
     } else {
         // if all input arguments are constant and lambda expr doesn't rely on other capture columns,
         // we can evaluate it based on const column
@@ -225,6 +227,7 @@ StatusOr<ColumnPtr> ArrayMapExpr::evaluate_lambda_expr(ExprContext* context, Chu
             }
         }
     }
+
     DCHECK(column != nullptr);
     column = ColumnHelper::cast_to_nullable_column(column);