Adding server principal and principal

apache · Sep 29, 2024 · 7de8d7d · 7de8d7d
1 parent 4a1d02e
commit 7de8d7d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/kyuubi-hive-jdbc/src/main/java/org/apache/kyuubi/jdbc/hive/Utils.java b/kyuubi-hive-jdbc/src/main/java/org/apache/kyuubi/jdbc/hive/Utils.java
@@ -364,6 +364,20 @@ public static JdbcConnectionParams extractURLComponents(String uri, Properties i
       }
     }
 
+    if (!connParams.getSessionVars().containsKey(AUTH_KYUUBI_SERVER_PRINCIPAL)) {
+      if (info.containsKey(AUTH_KYUUBI_SERVER_PRINCIPAL)) {
+        connParams
+            .getSessionVars()
+            .put(AUTH_KYUUBI_SERVER_PRINCIPAL, info.getProperty(AUTH_KYUUBI_SERVER_PRINCIPAL));
+      }
+    }
+
+    if (!connParams.getSessionVars().containsKey(AUTH_PRINCIPAL)) {
+      if (info.containsKey(AUTH_PRINCIPAL)) {
+        connParams.getSessionVars().put(AUTH_PRINCIPAL, info.getProperty(AUTH_PRINCIPAL));
+      }
+    }
+
     if (info.containsKey(AUTH_TYPE)) {
       connParams.getSessionVars().put(AUTH_TYPE, info.getProperty(AUTH_TYPE));
     }