[BugFix] fix handling of const column in array_to_bitmap (#52423)

Signed-off-by: silverbullet233 <[email protected]>
(cherry picked from commit e0e71c2)

be/src/exprs/bitmap_functions.cpp

@@ -398,7 +398,7 @@ StatusOr<ColumnPtr> BitmapFunctions::bitmap_to_array(FunctionContext* context, c
 StatusOr<ColumnPtr> BitmapFunctions::array_to_bitmap(FunctionContext* context, const starrocks::Columns& columns) {
     RETURN_IF_COLUMNS_ONLY_NULL(columns);
     const constexpr LogicalType TYPE = TYPE_BIGINT;
-    size_t size = columns[0]->size();
+    size_t size = columns[0]->is_constant() ? 1 : columns[0]->size();
     ColumnBuilder<TYPE_OBJECT> builder(size);
 
     Column* data_column = ColumnHelper::get_data_column(columns[0].get());
@@ -440,7 +440,8 @@ StatusOr<ColumnPtr> BitmapFunctions::array_to_bitmap(FunctionContext* context, c
         // append bitmap
         builder.append(std::move(bitmap));
     }
-    return builder.build(ColumnHelper::is_all_const(columns));
+    auto result = builder.build(false);
+    return columns[0]->is_constant() ? ConstColumn::create(std::move(result), columns[0]->size()) : result;
 }
 
 StatusOr<ColumnPtr> BitmapFunctions::bitmap_max(FunctionContext* context, const starrocks::Columns& columns) {

test/sql/test_array_fn/R/test_array_fn

@@ -3562,6 +3562,19 @@ select REVERSE( cast ('[40360,40361]' as array<int>));
 -- result:
 [40361,40360]
 -- !result
+select bitmap_to_string(array_to_bitmap( cast ('[40360,40361]' as array<int>))) from table(generate_series(1,10,1));
+-- result:
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+40360,40361
+-- !result
 select array_contains( cast ('null' as array<int>), 40360);
 -- result:
 None

test/sql/test_array_fn/T/test_array_fn

@@ -685,6 +685,7 @@ select array_sum( cast ('[40360,40361]' as array<int>));
 select array_to_bitmap( cast ('[40360,40361]' as array<int>));
 select bitmap_to_array(array_to_bitmap(cast ('[40360,40361]' as array<int>)));
 select REVERSE( cast ('[40360,40361]' as array<int>));
+select bitmap_to_string(array_to_bitmap( cast ('[40360,40361]' as array<int>))) from table(generate_series(1,10,1));
 
 
 select array_contains( cast ('null' as array<int>), 40360);