def get(user, query_server=None):
# Avoid circular dependency
from beeswax.server.hive_server2_lib import HiveServerClientCompatible, HiveServerClient
from beeswax.server.beeswax_lib import BeeswaxClient
if query_server is None:
query_server = get_query_server(support_ddl=True)
if SERVER_INTERFACE.get() == HIVE_SERVER2:
return Dbms(HiveServerClientCompatible(HiveServerClient(query_server, user)), QueryHistory.SERVER_TYPE[1][0])
else:
return Dbms(BeeswaxClient(query_server, user), QueryHistory.SERVER_TYPE[0][0])
def get(user, query_server=None):
# Avoid circular dependency
from beeswax.server.hive_server2_lib import HiveServerClientCompatible, HiveServerClient
from beeswax.server.beeswax_lib import BeeswaxClient
if query_server is None:
query_server = get_query_server_config()
if query_server['server_interface'] == HIVE_SERVER2:
return Dbms(HiveServerClientCompatible(HiveServerClient(query_server, user)), QueryHistory.SERVER_TYPE[1][0])
else:
return Dbms(BeeswaxClient(query_server, user), QueryHistory.SERVER_TYPE[0][0])
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=query_server['server_name'],
kerberos_principal=kerberos_principal_short_name,
use_sasl=use_sasl,
timeout_seconds=conf.BEESWAX_SERVER_CONN_TIMEOUT.get())
def __init__(self, client_class, 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(client_class.Client,
query_server['server_host'],
query_server['server_port'],
service_name='Impala',
kerberos_principal=kerberos_principal_short_name,
use_sasl=use_sasl,
timeout_seconds=conf.BEESWAX_SERVER_CONN_TIMEOUT.get())