[KYUUBI #6147][TRINO] Using the overlay conf as session conf

apache · Mar 8, 2024 · 4ee7c04 · 4ee7c04
1 parent 6fc7552
commit 4ee7c04
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...trino-engine/src/main/scala/org/apache/kyuubi/engine/trino/session/TrinoSessionImpl.scala b/...trino-engine/src/main/scala/org/apache/kyuubi/engine/trino/session/TrinoSessionImpl.scala
@@ -49,7 +49,11 @@ class TrinoSessionImpl(
     sessionManager: SessionManager)
   extends AbstractSession(protocol, user, password, ipAddress, conf, sessionManager) {
 
-  val sessionConf: KyuubiConf = sessionManager.getConf
+  val sessionConf: KyuubiConf = {
+    val engineConf = sessionManager.getConf
+    conf.foreach(kv => engineConf.set(kv._1, kv._2))
+    engineConf
+  }
 
   override val handle: SessionHandle =
     conf.get(KYUUBI_SESSION_HANDLE_KEY).map(SessionHandle.fromUUID).getOrElse(SessionHandle())