fix share state behavior (#437)

pingcap · Sep 5, 2018 · 88a73ec · 88a73ec
1 parent 5b15211
commit 88a73ec
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/sql/TiSparkSession.scala b/core/src/main/scala/org/apache/spark/sql/TiSparkSession.scala
@@ -30,8 +30,8 @@ import org.apache.spark.{SparkConf, SparkContext}
 
 class TiSparkSession(
   @transient override val sparkContext: SparkContext,
-  @transient private val existingSharedState: Option[SharedState]
-) extends SparkSession(sparkContext) { self =>
+  @transient private val tiExistingSharedState: Option[SharedState]
+) extends SparkSession(sparkContext) {
 
   private[sql] def this(sc: SparkContext) {
     this(sc, None)
@@ -40,6 +40,11 @@ class TiSparkSession(
   @transient
   override lazy val sessionState: SessionState = new TiSessionState(this)
 
+  @transient
+  override private[sql] lazy val sharedState: SharedState = {
+    tiExistingSharedState.getOrElse(new SharedState(sparkContext))
+  }
+
   override def newSession(): SparkSession = new TiSparkSession(sparkContext, Some(sharedState))
 
   TiUtils.sessionInitialize(