Sfoglia il codice sorgente

[beeswax] Sentry support for HiveServer2 and Impala

Send impersonation proxy user all the time.
Romain Rigaux 12 anni fa
parent
commit
2481949e31
1 ha cambiato i file con 4 aggiunte e 3 eliminazioni
  1. 4 3
      apps/beeswax/src/beeswax/server/hive_server2_lib.py

+ 4 - 3
apps/beeswax/src/beeswax/server/hive_server2_lib.py

@@ -345,10 +345,11 @@ class HiveServerClient:
     if self.impersonation_enabled:
       kwargs.update({'username': 'hue'})
 
-      if self.query_server['server_name'] == 'impala':
+      if self.query_server['server_name'] == 'impala': # Only when Impala accepts it
         kwargs['configuration'].update({'impala.doas.user': user.username})
-      else:
-        kwargs['configuration'].update({'hive.server2.proxy.user': user.username})
+
+    if self.query_server['server_name'] == 'beeswax': # All the time
+      kwargs['configuration'].update({'hive.server2.proxy.user': user.username})
 
     req = TOpenSessionReq(**kwargs)
     res = self._client.OpenSession(req)