From 8b3380ef6108f8045e2d5fa903f0f2bc176af2fa Mon Sep 17 00:00:00 2001 From: ntnhaatj Date: Sat, 31 Aug 2024 16:33:20 +0700 Subject: [PATCH] fix: update to use pyhive function to get compatible sasl --- dbt/adapters/spark/connections.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/dbt/adapters/spark/connections.py b/dbt/adapters/spark/connections.py index 0405eaf5b..e13e90e2d 100644 --- a/dbt/adapters/spark/connections.py +++ b/dbt/adapters/spark/connections.py @@ -613,12 +613,11 @@ def build_ssl_transport( password = "x" def sasl_factory() -> SASLClient: + from pyhive.hive import get_installed_sasl if sasl_auth == "GSSAPI": - sasl_client = SASLClient(host, kerberos_service_name, mechanism=sasl_auth) + sasl_client = get_installed_sasl(host, sasl_auth, kerberos_service_name) elif sasl_auth == "PLAIN": - sasl_client = SASLClient( - host, mechanism=sasl_auth, username=username, password=password - ) + sasl_client = get_installed_sasl(host, sasl_auth, username=username, password=password) else: raise AssertionError return sasl_client