Use getMethod.

viirya · viirya · commit 9fc152c21d79 · 2021-04-29T23:06:35.000-07:00
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/objects/objects.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/objects/objects.scala
@@ -325,13 +325,11 @@ case class Invoke(
 
   @transient lazy val method = targetObject.dataType match {
     case ObjectType(cls) =>
-      val m = cls.getMethods.find { m =>
-        m.getName == encodedFunctionName && m.getParameterCount == arguments.length
-      }
-      if (m.isEmpty) {
-        sys.error(s"Couldn't find $encodedFunctionName on $cls")
-      } else {
-        m
+      try {
+        Some(cls.getMethod(encodedFunctionName, argClasses: _*))
+      } catch {
+        case _: NoSuchMethodException =>
+          sys.error(s"Couldn't find $encodedFunctionName on $cls")
       }
     case _ => None
   }
