Ray序列化问题

zhnlk · Sep 9, 2020 · 61079a5 · 61079a5
1 parent b344bc0
commit 61079a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala b/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala
@@ -116,7 +116,7 @@ class Ray(override val uid: String) extends SQLAlg with VersionCompatibility wit
 
     targetLen = tempdf.rdd.partitions.length
     logInfo(s"RayMode: Final TargetLen ${targetLen}")
-    val owner = ScriptSQLExec.context().owner
+    val _owner = ScriptSQLExec.context().owner
 
     val thread = new Thread("temp-data-server-in-spark") {
       override def run(): Unit = {
@@ -125,7 +125,7 @@ class Ray(override val uid: String) extends SQLAlg with VersionCompatibility wit
         val tempSocketServerHost = tempSocketServerInDriver._host
         val tempSocketServerPort = tempSocketServerInDriver._port
         val timezoneID = session.sessionState.conf.sessionLocalTimeZone
-
+        val owner = _owner
         tempdf.rdd.mapPartitions { iter =>
 
           val host: String = if (SparkEnv.get == null || MLSQLSparkUtils.blockManager == null || MLSQLSparkUtils.blockManager.blockManagerId == null) {