diff --git a/fe/fe-core/src/main/java/org/apache/doris/analysis/AggregateInfo.java b/fe/fe-core/src/main/java/org/apache/doris/analysis/AggregateInfo.java index 8cfc113b16c873..87b3653a199fab 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/analysis/AggregateInfo.java +++ b/fe/fe-core/src/main/java/org/apache/doris/analysis/AggregateInfo.java @@ -905,7 +905,9 @@ public void materializeRequiredSlots(Analyzer analyzer, ExprSubstitutionMap smap intermediateSlotDesc.setIsMaterialized(true); } - if (!slotDesc.isMaterialized()) { + if (!slotDesc.isMaterialized() + && !(i == aggregateExprsSize - 1 && materializedSlots.isEmpty() && groupingExprs.isEmpty())) { + // we need keep at least one materialized slot in agg node continue; } diff --git a/regression-test/data/correctness_p0/test_agg_materialize.out b/regression-test/data/correctness_p0/test_agg_materialize.out new file mode 100644 index 00000000000000..1969d634862791 --- /dev/null +++ b/regression-test/data/correctness_p0/test_agg_materialize.out @@ -0,0 +1,4 @@ +-- This file is automatically generated. You should know what you did if you want to edit this +-- !select -- +3 + diff --git a/regression-test/suites/correctness_p0/test_agg_materialize.groovy b/regression-test/suites/correctness_p0/test_agg_materialize.groovy new file mode 100644 index 00000000000000..7911231ed4080c --- /dev/null +++ b/regression-test/suites/correctness_p0/test_agg_materialize.groovy @@ -0,0 +1,34 @@ +// Licensed to the Apache Software Foundation (ASF) under one +// or more contributor license agreements. See the NOTICE file +// distributed with this work for additional information +// regarding copyright ownership. The ASF licenses this file +// to you under the Apache License, Version 2.0 (the +// "License"); you may not use this file except in compliance +// with the License. You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, +// software distributed under the License is distributed on an +// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +// KIND, either express or implied. See the License for the +// specific language governing permissions and limitations +// under the License. + +// The cases is copied from https://github.com/trinodb/trino/tree/master +// /testing/trino-product-tests/src/main/resources/sql-tests/testcases/aggregate +// and modified by Doris. + +suite("test_agg_materialize") { + sql "set enable_nereids_planner=false" + qt_select """with tb1 as (select * from (select * from (select 1 k1) as t lateral view explode([1,2,3]) tmp1 as e1)t) + select count(*) from (select 1, count(*) + from tb1 + where e1 in (1, 2) + group by e1 + union all + select 1, count(*) + from tb1 + where e1 = 1)tttt + ; """ +}