fix build

Justin Uang · Justin Uang · commit 83774685de7d · 2019-01-14T17:57:02.000-05:00
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/adaptive/QueryStage.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/adaptive/QueryStage.scala
@@ -61,7 +61,7 @@ abstract class QueryStage extends UnaryExecNode {
     }
     val broadcastFutures = broadcastQueryStages.map { queryStage =>
       Future {
-        SQLExecution.withExecutionId(sqlContext.sparkContext, executionId) {
+        SQLExecution.withExecutionId(sqlContext.sparkSession, executionId) {
           queryStage.prepareBroadcast()
         }
       }(QueryStage.executionContext)

Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,7 @@ abstract class QueryStage extends UnaryExecNode {`
`61`	`61`	`}`
`62`	`62`	`val broadcastFutures = broadcastQueryStages.map { queryStage =>`
`63`	`63`	`Future {`
`64`		`- SQLExecution.withExecutionId(sqlContext.sparkContext, executionId) {`
	`64`	`+ SQLExecution.withExecutionId(sqlContext.sparkSession, executionId) {`
`65`	`65`	`queryStage.prepareBroadcast()`
`66`	`66`	`}`
`67`	`67`	`}(QueryStage.executionContext)`