Github user ueshin commented on a diff in the pull request:
https://github.com/apache/spark/pull/21028#discussion_r182688229
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/collectionOperations.scala
---
@@ -288,6 +288,114 @@ case class ArrayContains(left: Expression, right:
Expression)
override def prettyName: String = "array_contains"
}
+/**
+ * Checks if the two arrays contain at least one common element.
+ */
+@ExpressionDescription(
+ usage = "_FUNC_(a1, a2) - Returns true if a1 contains at least an
element present also in a2.",
+ examples = """
+ Examples:
+ > SELECT _FUNC_(array(1, 2, 3), array(3, 4, 5));
+ true
+ """, since = "2.4.0")
+case class ArraysOverlap(left: Expression, right: Expression)
+ extends BinaryExpression with ImplicitCastInputTypes {
+
+ private lazy val elementType =
inputTypes.head.asInstanceOf[ArrayType].elementType
+
+ override def dataType: DataType = BooleanType
+
+ override def inputTypes: Seq[AbstractDataType] = left.dataType match {
+ case la: ArrayType if la.sameType(right.dataType) =>
+ Seq(la, la)
+ case _ => Seq.empty
+ }
+
+ override def checkInputDataTypes(): TypeCheckResult = {
+ if (!left.dataType.isInstanceOf[ArrayType] ||
!right.dataType.isInstanceOf[ArrayType] ||
+ !left.dataType.sameType(right.dataType)) {
+ TypeCheckResult.TypeCheckFailure("Arguments must be arrays with the
same element type.")
+ } else {
+ TypeCheckResult.TypeCheckSuccess
+ }
+ }
+
+ override def nullable: Boolean = {
+ left.nullable || right.nullable ||
left.dataType.asInstanceOf[ArrayType].containsNull ||
+ right.dataType.asInstanceOf[ArrayType].containsNull
+ }
+
+ override def nullSafeEval(a1: Any, a2: Any): Any = {
+ var hasNull = false
+ val arr1 = a1.asInstanceOf[ArrayData]
+ val arr2 = a2.asInstanceOf[ArrayData]
+ if (arr1.numElements() > 0) {
+ arr1.foreach(elementType, (_, v1) =>
+ if (v1 == null) {
+ hasNull = true
+ } else {
+ arr2.foreach(elementType, (_, v2) =>
+ if (v2 == null) {
+ hasNull = true
+ } else if (v1 == v2) {
+ return true
+ }
+ )
+ }
+ )
+ } else {
--- End diff --
We can skip if the right is `containsNull == false`?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
