Skip to content

Commit

Permalink
get address from blockmanager
Browse files Browse the repository at this point in the history
allwefantasy committed Sep 9, 2020
1 parent 2b6be62 commit f6952a1
Showing 2 changed files with 5 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -8,4 +8,8 @@ object MLSQLSparkUtils {
SparkEnv.get.rpcEnv
}

def blockManager = {
SparkEnv.get.blockManager
}

}
2 changes: 1 addition & 1 deletion streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala
Original file line number Diff line number Diff line change
@@ -133,7 +133,7 @@ class Ray(override val uid: String) extends SQLAlg with VersionCompatibility wit
Map("PY_EXECUTE_USER" -> owner))
NetTool.localHostName()
}
else MLSQLSparkUtils.rpcEnv().address.host
else MLSQLSparkUtils.blockManager.blockManagerId.host

val socketRunner = new SparkSocketRunner("serveToStreamWithArrow", host, timezoneID)
val commonTaskContext = new SparkContextImp(TaskContext.get(), null)

0 comments on commit f6952a1

Please sign in to comment.