Github user ueshin commented on a diff in the pull request:
https://github.com/apache/spark/pull/21037#discussion_r182298195
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/collectionOperations.scala
---
@@ -353,3 +353,61 @@ case class ArrayMax(child: Expression) extends
UnaryExpression with ImplicitCast
override def prettyName: String = "array_max"
}
+
+
+/**
+ * Returns the position of the first occurrence of element in the given
array as long.
+ * Returns 0 if substr could not be found in str. Returns null if either
of the arguments are null
+ *
+ * NOTE: that this is not zero based, but 1-based index. The first
character in str has index 1.
+ */
+@ExpressionDescription(
+ usage = """
+ _FUNC_(array, element) - Returns the (1-based) index of the first
element of the array as long.
+ """,
+ examples = """
+ Examples:
+ > SELECT _FUNC_(array(3, 2, 1), 1);
+ 3
+ """,
+ since = "2.4.0")
+case class ArrayPosition(left: Expression, right: Expression)
+ extends BinaryExpression with ImplicitCastInputTypes {
+
+ override def dataType: DataType = LongType
+ override def inputTypes: Seq[AbstractDataType] =
+ Seq(ArrayType, left.dataType.asInstanceOf[ArrayType].elementType)
+
+ override def nullable: Boolean = {
+ left.nullable || right.nullable
+ }
--- End diff --
We don't need this here because this is the same as the one in
`BinaryExpression`.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
