Преглед изворни кода

HUE-1107 [beeswax] Hive Server 2 kerberos support

1. use sasl Thrift
Romain Rigaux пре 12 година
родитељ
комит
1d3a308
1 измењених фајлова са 6 додато и 2 уклоњено
  1. 6 2
      apps/beeswax/src/beeswax/server/hive_server2_lib.py

+ 6 - 2
apps/beeswax/src/beeswax/server/hive_server2_lib.py

@@ -30,6 +30,7 @@ from beeswax import conf
 from beeswax.models import Session, HiveServerQueryHandle, HiveServerQueryHistory
 from beeswax.server.dbms import Table, NoSuchObjectException, DataTable,\
   QueryServerException
+from beeswax.server.beeswax_lib import BeeswaxClient
 
 
 LOG = logging.getLogger(__name__)
@@ -226,16 +227,19 @@ class HiveServerTColumnDesc:
 
 
 class HiveServerClient:
-  """Thrift service client."""
 
   def __init__(self, query_server, user):
     self.query_server = query_server
     self.user = user
 
+    use_sasl, kerberos_principal_short_name = BeeswaxClient.get_security(query_server)
+
     self._client = thrift_util.get_client(TCLIService.Client,
                                           query_server['server_host'],
                                           query_server['server_port'],
-                                          service_name='Hive Server 2',
+                                          service_name=query_server['server_name'],
+                                          kerberos_principal=kerberos_principal_short_name,
+                                          use_sasl=use_sasl,
                                           timeout_seconds=conf.BEESWAX_SERVER_CONN_TIMEOUT.get())