Skip to content

Commit

Permalink
fix: update to use pyhive function to get compatible sasl
Browse files Browse the repository at this point in the history
  • Loading branch information
ntnhaatj committed Aug 31, 2024
1 parent 2124423 commit 8b3380e
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions dbt/adapters/spark/connections.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 8b3380e

Please sign in to comment.