change .filter to .find

apache · Apr 16, 2021 · 04b24c9 · 04b24c9
1 parent baf8125
commit 04b24c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/SchemaPruning.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/SchemaPruning.scala
@@ -67,7 +67,7 @@ object SchemaPruning extends SQLConfHelper {
         val filteredRightFieldNames = rightStruct.fieldNames
           .filter(name => leftStruct.fieldNames.exists(resolver(_, name)))
         val sortedLeftFields = filteredRightFieldNames.map { fieldName =>
-          val resolvedLeftStruct = leftStruct.filter(p => resolver(p.name, fieldName)).head
+          val resolvedLeftStruct = leftStruct.find(p => resolver(p.name, fieldName)).get
           val leftFieldType = resolvedLeftStruct.dataType
           val rightFieldType = rightStruct(fieldName).dataType
           val sortedLeftFieldType = sortLeftFieldsByRight(leftFieldType, rightFieldType)