made suggested changes

apache · Jul 16, 2024 · 0f1ea9d · 0f1ea9d
1 parent 0287b4f
commit 0f1ea9d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/aggregate/Mode.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/aggregate/Mode.scala
@@ -76,8 +76,7 @@ case class Mode(
       data: ArrayData): Seq[Any] = {
     (0 until data.numElements()).map { i =>
       data.get(i, a.elementType) match {
-        case k: UTF8String if a.elementType.isInstanceOf[StringType] &&
-          !a.elementType.asInstanceOf[StringType].supportsBinaryEquality
+        case k: UTF8String if !UnsafeRowUtils.isBinaryStable(a.elementType)
           => CollationFactory.getCollationKey(k, a.elementType.asInstanceOf[StringType].collationId)
         case k if a.elementType.isInstanceOf[StructType] =>
           recursivelyGetBufferForStructType(
@@ -106,7 +105,7 @@ case class Mode(
 
   private def isSpecificStringTypeMatch(field: StructField, fieldName: String): Boolean =
     field.dataType.isInstanceOf[StringType] &&
-      !field.dataType.asInstanceOf[StringType].supportsBinaryEquality &&
+      !UnsafeRowUtils.isBinaryStable(field.dataType) &&
       field.name == fieldName
 
   private def isSpecificStructTypeMatch(field: StructField, fieldName: String): Boolean =