mirror of
https://github.com/apache/superset.git
synced 2026-04-14 05:34:38 +00:00
9114d86ecd
* Initial hive implementation * Fix select star query for hive. * Exclude generated code. * Address code coverage and linting. * Exclude generated code from coveralls. * Fix lint errors * Move TCLIService to it's own repo. * Address comments * Implement special postgres case,
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
from pyhive import hive
|
|
from pythrifthiveapi.TCLIService import ttypes
|
|
|
|
|
|
# TODO: contribute back to pyhive.
|
|
def fetch_logs(self, max_rows=1024,
|
|
orientation=ttypes.TFetchOrientation.FETCH_NEXT):
|
|
"""Mocked. Retrieve the logs produced by the execution of the query.
|
|
Can be called multiple times to fetch the logs produced after
|
|
the previous call.
|
|
:returns: list<str>
|
|
:raises: ``ProgrammingError`` when no query has been started
|
|
.. note::
|
|
This is not a part of DB-API.
|
|
"""
|
|
try:
|
|
req = ttypes.TGetLogReq(operationHandle=self._operationHandle)
|
|
logs = self._connection.client.GetLog(req)
|
|
return logs
|
|
except ttypes.TApplicationException as e: # raised if Hive is used
|
|
if self._state == self._STATE_NONE:
|
|
raise hive.ProgrammingError("No query yet")
|
|
logs = []
|
|
while True:
|
|
req = ttypes.TFetchResultsReq(
|
|
operationHandle=self._operationHandle,
|
|
orientation=ttypes.TFetchOrientation.FETCH_NEXT,
|
|
maxRows=self.arraysize,
|
|
fetchType=1 # 0: results, 1: logs
|
|
)
|
|
response = self._connection.client.FetchResults(req)
|
|
hive._check_status(response)
|
|
assert not (
|
|
response.results.rows, 'expected data in columnar format'
|
|
)
|
|
assert len(response.results.columns) == 1, response.results.columns
|
|
new_logs = hive._unwrap_column(response.results.columns[0])
|
|
logs += new_logs
|
|
if not new_logs:
|
|
break
|
|
return logs
|