From ddc91c835dc9eef050d640b0483aba9ef08bd73f Mon Sep 17 00:00:00 2001 From: Kent Yao Date: Wed, 16 May 2018 18:51:07 +0800 Subject: [PATCH] explicitly visit spark env --- src/main/scala/org/apache/spark/KyuubiSparkUtil.scala | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala b/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala index af5c5ed0c..1283e76a4 100644 --- a/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala +++ b/src/main/scala/org/apache/spark/KyuubiSparkUtil.scala @@ -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 }