[BugFix] Fix array_map crash when capture const array columns (backport #61309) (#61311)

Signed-off-by: stdpain <[email protected]>
Co-authored-by: stdpain <[email protected]>

Author: mergify[bot]

be/src/exprs/array_map_expr.cpp

@@ -181,7 +181,7 @@ StatusOr<ColumnPtr> ArrayMapExpr::evaluate_lambda_expr(ExprContext* context, Chu
         if constexpr (independent_lambda_expr) {
             cur_chunk->append_column(captured_column, slot_id);
         } else {
-            if (captured_column->is_array()) {
+            if (!captured_column->is_constant() && captured_column->is_array()) {
                 auto view_column = ArrayViewColumn::from_array_column(captured_column);
                 ASSIGN_OR_RETURN(auto replicated_view_column, view_column->replicate(aligned_offsets->get_data()));
                 cur_chunk->append_column(replicated_view_column, slot_id);

test/sql/test_array_fn/R/test_array_map_2

@@ -219,4 +219,23 @@ set @arr=array_generate(1,10000);
 select /*+ SET_VAR(query_mem_limit=104857600)*/array_sum(array_map(x->array_contains(@arr,x), array_generate(1,100000)));
 -- result:
 10000
+-- !result
+CREATE TABLE `array_map_x` (
+  `id` tinyint(4) NOT NULL COMMENT "",
+  `arr_str` array<varchar(65533)> NULL COMMENT "",
+  `arr_largeint` array<largeint(40)> NULL COMMENT ""
+) ENGINE=OLAP 
+DUPLICATE KEY(`id`)
+DISTRIBUTED BY RANDOM
+PROPERTIES (
+"replication_num" = "1"
+);
+-- result:
+-- !result
+insert into array_map_x values (1, array_repeat("abcdefghasdasdasirnqwrq", 2), array_repeat(100, 2));
+-- result:
+-- !result
+select cast(if (1 > rand(), "[]", "") as array<string>) l , array_map((x)-> (concat(x,l)), arr_str) from array_map_x;
+-- result:
+[]	[["abcdefghasdasdasirnqwrq"],["abcdefghasdasdasirnqwrq"]]
 -- !result

test/sql/test_array_fn/T/test_array_map_2

@@ -74,3 +74,16 @@ select array_map(arg0 -> array_map(arg1 -> array_map(arg2 -> array_map(arg3 -> a
 
 set @arr=array_generate(1,10000);
 select /*+ SET_VAR(query_mem_limit=104857600)*/array_sum(array_map(x->array_contains(@arr,x), array_generate(1,100000)));
+
+CREATE TABLE `array_map_x` (
+  `id` tinyint(4) NOT NULL COMMENT "",
+  `arr_str` array<varchar(65533)> NULL COMMENT "",
+  `arr_largeint` array<largeint(40)> NULL COMMENT ""
+) ENGINE=OLAP 
+DUPLICATE KEY(`id`)
+DISTRIBUTED BY RANDOM
+PROPERTIES (
+"replication_num" = "1"
+);
+insert into array_map_x values (1, array_repeat("abcdefghasdasdasirnqwrq", 2), array_repeat(100, 2));
+select cast(if (1 > rand(), "[]", "") as array<string>) l , array_map((x)-> (concat(x,l)), arr_str) from array_map_x;