[NDAP-614] Kyuubi가 spark.ndap.acl.user 에 JDBC 사용자 ID를 전달하도록 수정 (#11)

nexr · Jul 3, 2024 · 86edd1f · 86edd1f
1 parent af716b1
commit 86edd1f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/kyuubi-server/src/main/scala/org/apache/kyuubi/engine/spark/SparkProcessBuilder.scala b/kyuubi-server/src/main/scala/org/apache/kyuubi/engine/spark/SparkProcessBuilder.scala
@@ -77,6 +77,13 @@ class SparkProcessBuilder(
     buffer += executable
     buffer += CLASS
     buffer += mainClass
+    /**
+     * add current user to spark.yarn.tags
+     * allConf is immutable, so add to conf at here
+     */
+    conf.set(
+      SparkProcessBuilder.TAG_KEY,
+      conf.getOption(SparkProcessBuilder.TAG_KEY).map(_ + ",").getOrElse("") + s"$proxyUser")
 
     var allConf = conf.getAll
 
@@ -96,7 +103,9 @@ class SparkProcessBuilder(
       case None =>
         setSparkUserName(proxyUser, buffer)
         buffer += PROXY_USER
-        buffer += proxyUser
+        buffer += s"hive"
+        buffer += CONF
+        buffer += s"spark.ndap.acl.user=$proxyUser"
       case Some(name) =>
         setSparkUserName(name, buffer)
     }