Repository: spark Updated Branches: refs/heads/master 4bb2b3698 -> d4c397a64
[SPARK-11325] [SQL] Alias 'alias' in Scala's DataFrame API Author: Nong Li <non...@gmail.com> Closes #9286 from nongli/spark-11325. Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/d4c397a6 Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/d4c397a6 Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/d4c397a6 Branch: refs/heads/master Commit: d4c397a64af4cec899fdaa3e617ed20333cc567d Parents: 4bb2b36 Author: Nong Li <non...@gmail.com> Authored: Mon Oct 26 18:27:02 2015 -0700 Committer: Yin Huai <yh...@databricks.com> Committed: Mon Oct 26 18:27:02 2015 -0700 ---------------------------------------------------------------------- .../main/scala/org/apache/spark/sql/DataFrame.scala | 14 ++++++++++++++ .../scala/org/apache/spark/sql/DataFrameSuite.scala | 7 +++++++ 2 files changed, 21 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/d4c397a6/sql/core/src/main/scala/org/apache/spark/sql/DataFrame.scala ---------------------------------------------------------------------- diff --git a/sql/core/src/main/scala/org/apache/spark/sql/DataFrame.scala b/sql/core/src/main/scala/org/apache/spark/sql/DataFrame.scala index 25ad3bb..32d9b0b 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/DataFrame.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/DataFrame.scala @@ -699,6 +699,20 @@ class DataFrame private[sql]( def as(alias: Symbol): DataFrame = as(alias.name) /** + * Returns a new [[DataFrame]] with an alias set. Same as `as`. + * @group dfops + * @since 1.6.0 + */ + def alias(alias: String): DataFrame = as(alias) + + /** + * (Scala-specific) Returns a new [[DataFrame]] with an alias set. Same as `as`. + * @group dfops + * @since 1.6.0 + */ + def alias(alias: Symbol): DataFrame = as(alias) + + /** * Selects a set of column based expressions. * {{{ * df.select($"colA", $"colB" + 1) http://git-wip-us.apache.org/repos/asf/spark/blob/d4c397a6/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala ---------------------------------------------------------------------- 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 f4c7aa3..59565a6 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 @@ -105,6 +105,13 @@ class DataFrameSuite extends QueryTest with SharedSQLContext { assert(testData.head(2).head.schema === testData.schema) } + test("dataframe alias") { + val df = Seq(Tuple1(1)).toDF("c").as("t") + val dfAlias = df.alias("t2") + df.col("t.c") + dfAlias.col("t2.c") + } + test("simple explode") { val df = Seq(Tuple1("a b c"), Tuple1("d e")).toDF("words") --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org For additional commands, e-mail: commits-h...@spark.apache.org