apache/kyuubi
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

explicitly visit spark env

Browse Source
This commit is contained in:
Kent Yao 2018-05-16 18:51:07 +08:00
parent be7639ae8b
commit ddc91c835d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/scala/org/apache/spark/KyuubiSparkUtil.scala
View File

@ -27,6 +27,7 @@ import org.apache.spark.util.{ShutdownHookManager, Utils, VersionUtils}
import org.slf4j.Logger
import yaooqinn.kyuubi.Logging
import yaooqinn.kyuubi.utils.ReflectUtils
/**
* Wrapper for [[Utils]] and [[SparkHadoopUtil]]
@ -77,7 +78,8 @@ object KyuubiSparkUtil extends Logging {
info(s"Initializing KyuubiFirstClassLoader instance with url $url as first class members")
val classLoader = new KyuubiFirstClassLoader(Array(url), getContextOrSparkClassLoader())
classLoader.loadClass("org.apache.spark.SparkEnv", true)
classLoader.loadClass("org.apache.spark.SparkEnv$", true)
val sparkEnvClass = classLoader.loadClass("org.apache.spark.SparkEnv$", true)
ReflectUtils.invokeStaticMethod(sparkEnvClass, "set", Seq(classOf[String]), null)
classLoader
}