From cbd80a5bd370242a8dced2b8fbd92fd9d433fc0c Mon Sep 17 00:00:00 2001 From: BenFradet Date: Wed, 17 Jun 2015 19:30:23 +0200 Subject: [PATCH] deprecated callUdf in favor of callUDF --- .../org/apache/spark/sql/functions.scala | 21 +++++++++++++++++++ .../org/apache/spark/sql/DataFrameSuite.scala | 2 +- 2 files changed, 22 insertions(+), 1 deletion(-) diff --git a/sql/core/src/main/scala/org/apache/spark/sql/functions.scala b/sql/core/src/main/scala/org/apache/spark/sql/functions.scala index c5b77724aae17..33b4e2ddc024c 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/functions.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/functions.scala @@ -1638,6 +1638,25 @@ object functions { // scalastyle:on + /** + * Call an user-defined function. + * Example: + * {{{ + * import org.apache.spark.sql._ + * + * val df = Seq(("id1", 1), ("id2", 4), ("id3", 5)).toDF("id", "value") + * val sqlContext = df.sqlContext + * sqlContext.udf.register("simpleUdf", (v: Int) => v * v) + * df.select($"id", callUDF("simpleUdf", $"value")) + * }}} + * + * @group udf_funcs + * @since 1.5.0 + */ + def callUDF(udfName: String, cols: Column*): Column = { + UnresolvedFunction(udfName, cols.map(_.expr)) + } + /** * Call an user-defined function. * Example: @@ -1652,7 +1671,9 @@ object functions { * * @group udf_funcs * @since 1.4.0 + * @deprecated As of 1.5.0, since it was not coherent to have two functions callUdf and callUDF */ + @deprecated("Use callUDF", "1.5.0") def callUdf(udfName: String, cols: Column*): Column = { UnresolvedFunction(udfName, cols.map(_.expr)) } diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala index fa98e23e3d147..b02d66df14bfd 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala @@ -306,7 +306,7 @@ class DataFrameSuite extends QueryTest { val sqlctx = df.sqlContext sqlctx.udf.register("simpleUdf", (v: Int) => v * v) checkAnswer( - df.select($"id", callUdf("simpleUdf", $"value")), + df.select($"id", callUDF("simpleUdf", $"value")), Row("id1", 1) :: Row("id2", 16) :: Row("id3", 25) :: Nil) }