mirror of
https://github.com/apache/impala.git
synced 2025-12-19 18:12:08 -05:00
ea13e74497
When Impala Shell receives an http error message (that is a message with
http code greater than or equal to 300), it may sleep for a time before
retrying. If the message contains a 'Retry-After' header that has an
integer value, then this will be used as the time for which to sleep.
The implementation is to use a new HttpError exception (similar to that
used in Impyla) which includes more information from the error message
(including the headers) so that catchers of the exception can use the
'Retry-After' header if appropriate.
TESTING:
Hand testing with a proxy that uses the 'Retry-After' header.
Added new tests that use the fault injection framework in
test_hs2_fault_injection.py
Change-Id: I2b4226e7723d585d61deb4d1d6777aac901bfd93
Reviewed-on: http://gerrit.cloudera.org:8080/16702
Reviewed-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com>
Tested-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com>
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
|
|
class RPCException(Exception):
|
|
def __init__(self, value="", exception_type=""):
|
|
self.value = value
|
|
self.exception_type = exception_type
|
|
|
|
def __str__(self):
|
|
return self.value
|
|
|
|
|
|
class QueryStateException(Exception):
|
|
def __init__(self, value=""):
|
|
self.value = value
|
|
|
|
def __str__(self):
|
|
return self.value
|
|
|
|
|
|
class DisconnectedException(Exception):
|
|
def __init__(self, value=""):
|
|
self.value = value
|
|
|
|
def __str__(self):
|
|
return self.value
|
|
|
|
|
|
class QueryCancelledByShellException(Exception): pass
|
|
|
|
|
|
class MissingThriftMethodException(Exception):
|
|
"""Thrown if a Thrift method that the client tried to call is missing."""
|
|
def __init__(self, value=""):
|
|
self.value = value
|
|
|
|
def __str__(self):
|
|
return self.value
|
|
|
|
|
|
class HttpError(Exception):
|
|
"""An error containing an http response code and a possible message body"""
|
|
def __init__(self, code, message, body, http_headers):
|
|
self.code = code
|
|
self.message = message
|
|
self.body = body
|
|
self.http_headers = http_headers
|
|
|
|
def __str__(self):
|
|
if not self.body:
|
|
return "HTTP code {}: {}".format(self.code, self.message)
|
|
else:
|
|
return "HTTP code {}: {} [{}]".format(self.code, self.message, self.body)
|