From c16babb6feef9f1f9a37cdf3462f48ed5d38f453 Mon Sep 17 00:00:00 2001 From: Zhan Zhang Date: Thu, 7 May 2015 13:03:58 -0700 Subject: [PATCH] SPARK-7449: Schema fix --- .../org/apache/spark/sql/sources/DataSourceStrategy.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sql/core/src/main/scala/org/apache/spark/sql/sources/DataSourceStrategy.scala b/sql/core/src/main/scala/org/apache/spark/sql/sources/DataSourceStrategy.scala index b3d71f687a60..b91fc007b791 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/sources/DataSourceStrategy.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/sources/DataSourceStrategy.scala @@ -24,7 +24,7 @@ import org.apache.spark.sql.catalyst.planning.PhysicalOperation import org.apache.spark.sql.catalyst.plans.logical import org.apache.spark.sql.catalyst.plans.logical.LogicalPlan import org.apache.spark.sql.execution.SparkPlan -import org.apache.spark.sql.types.{UTF8String, StringType} +import org.apache.spark.sql.types.{StructType, UTF8String, StringType} import org.apache.spark.sql.{Row, Strategy, execution, sources} /** @@ -120,7 +120,7 @@ private[sql] object DataSourceStrategy extends Strategy { output: Seq[Attribute], rdd: RDD[Row]): SparkPlan = { val converted = if (relation.needConversion) { - execution.RDDConversions.rowToRowRdd(rdd, relation.schema) + execution.RDDConversions.rowToRowRdd(rdd, StructType.fromAttributes(output)) } else { rdd }