in src/ab/plugins/db/hive.py [0:0]
def __init__(self, host, port, db, username, password=None):
self.sampler_class = Sampler
self.host = host
self.port = port
self.db = db
self.username = username
self.password = password
if password is not None:
auth = 'LDAP'
kerberos_service_name = None
else:
auth = 'KERBEROS'
kerberos_service_name = 'hive'
thrift_transport = get_thrift_transport(host=host, port=port,
username=username,
auth=auth,
password=password,
kerberos_service_name=kerberos_service_name,
kerberos_host=app.config.get('HUAWEI_KERBEROS_HOST'),
socket_timeout_in_ms=app.config.HIVE_TIMEOUT * 1000,
)
self.connect_args = {'port': None, 'database': db, 'thrift_transport': thrift_transport}
self.engine = db_conn_pool.get_engine("hive://", connect_args=self.connect_args)