[GLUTEN-4831][VL] Support StructType in HashAggregate (#4832)

apache · Mar 7, 2024 · f9daf49 · f9daf49
1 parent 32b492d
commit f9daf49
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/backends-velox/src/test/scala/io/glutenproject/execution/TestOperator.scala b/backends-velox/src/test/scala/io/glutenproject/execution/TestOperator.scala
@@ -1145,4 +1145,15 @@ class TestOperator extends VeloxWholeStageTransformerSuite {
         |""".stripMargin
     )(df => checkFallbackOperators(df, 0))
   }
+
+  test("Support StructType in HashAggregate") {
+    runQueryAndCompare("""
+                         |select s, count(1) from (
+                         |   select named_struct('id', cast(id as int),
+                         |   'id_str', cast(id as string)) as s from range(100)
+                         |) group by s
+                         |""".stripMargin) {
+      checkOperatorMatch[HashAggregateExecTransformer]
+    }
+  }
 }
diff --git a/gluten-core/src/main/scala/io/glutenproject/execution/HashAggregateExecBaseTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/execution/HashAggregateExecBaseTransformer.scala
@@ -99,6 +99,7 @@ abstract class HashAggregateExecBaseTransformer(
         true
       case _: NumericType => true
       case _: ArrayType => true
+      case _: StructType => true
       case _: NullType => true
       case _ => false
     }