From 8060aa418138179286bd3d6eb64daac53610cadf Mon Sep 17 00:00:00 2001 From: WeichenXu Date: Fri, 12 Aug 2016 02:40:23 -0700 Subject: [PATCH] prevent_user_call_df_select_will_empty_paramlist --- sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala b/sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala index c119df83b3d71..451edfe827e30 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala @@ -1004,7 +1004,10 @@ class Dataset[T] private[sql]( * @since 2.0.0 */ @scala.annotation.varargs - def select(cols: Column*): DataFrame = withPlan { + def select(col: Column, cols: Column*): DataFrame = select((col +: cols): _*) + + @scala.annotation.varargs + private[spark] def select(cols: Column*): DataFrame = withPlan { Project(cols.map(_.named), logicalPlan) }