xinrong-databricks commented on a change in pull request #35415:
URL: https://github.com/apache/spark/pull/35415#discussion_r802550799
##########
File path:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/stringExpressions.scala
##########
@@ -2538,6 +2538,56 @@ case class Encode(value: Expression, charset: Expression)
newLeft: Expression, newRight: Expression): Encode = copy(value = newLeft,
charset = newRight)
}
+/**
+ * Converts the input expression to a binary value based on the supplied
format.
+ */
+// scalastyle:off line.size.limit
+@ExpressionDescription(
+ usage = """
+ _FUNC_(str[, fmt]) - Converts the input `str` to a binary value based on
the supplied `fmt`.
+ By default, the binary format for conversion is "hex" if `fmt` is
omitted.
+ """,
+ examples = """
+ Examples:
+ > SELECT _FUNC_('abc', 'utf-8');
+ abc
+ """,
+ since = "3.3.0",
+ group = "string_funcs")
+// scalastyle:on line.size.limit
+case class ToBinary(expr: Expression, format: Option[Expression], child:
Expression)
+ extends RuntimeReplaceable {
+
+ def this(expr: Expression, format: Expression) = this(expr, Option(format),
+ format.toString match {
+ case "hex" => Unhex(expr)
+ case "utf-8" => Encode(expr, Literal("UTF-8"))
+ case "base64" => UnBase64(expr)
+ case "base2" => Cast(expr, BinaryType)
+ }
+ )
+
+ def this(expr: Expression) = this(expr, None, Unhex(expr))
+
+ override def flatArguments: Iterator[Any] = Iterator(expr, format)
+ override def exprsReplaced: Seq[Expression] = expr +: format.toSeq
+
+ override def prettyName: String = "to_binary"
+ override def dataType: DataType = BinaryType
+
+ override def checkInputDataTypes(): TypeCheckResult = {
+ if (format.isEmpty || Seq(Literal("hex"), Literal("utf-8"),
Literal("base64"), Literal("base2"))
Review comment:
Good point, modified!
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
