53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""Special cased retries.
|
|
|
|
These are additional retry cases we still have to handle from the legacy
|
|
retry handler. They don't make sense as part of the standard mode retry
|
|
module. Ideally we should be able to remove this module.
|
|
|
|
"""
|
|
import logging
|
|
from binascii import crc32
|
|
|
|
from botocore.retries.base import BaseRetryableChecker
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# TODO: This is an ideal candidate for the retryable trait once that's
|
|
# available.
|
|
class RetryIDPCommunicationError(BaseRetryableChecker):
|
|
|
|
_SERVICE_NAME = 'sts'
|
|
|
|
def is_retryable(self, context):
|
|
service_name = context.operation_model.service_model.service_name
|
|
if service_name != self._SERVICE_NAME:
|
|
return False
|
|
error_code = context.get_error_code()
|
|
return error_code == 'IDPCommunicationError'
|
|
|
|
|
|
class RetryDDBChecksumError(BaseRetryableChecker):
|
|
|
|
_CHECKSUM_HEADER = 'x-amz-crc32'
|
|
_SERVICE_NAME = 'dynamodb'
|
|
|
|
def is_retryable(self, context):
|
|
service_name = context.operation_model.service_model.service_name
|
|
if service_name != self._SERVICE_NAME:
|
|
return False
|
|
if context.http_response is None:
|
|
return False
|
|
checksum = context.http_response.headers.get(self._CHECKSUM_HEADER)
|
|
if checksum is None:
|
|
return False
|
|
actual_crc32 = crc32(context.http_response.content) & 0xFFFFFFFF
|
|
if actual_crc32 != int(checksum):
|
|
logger.debug(
|
|
"DynamoDB crc32 checksum does not match, "
|
|
"expected: %s, actual: %s",
|
|
checksum,
|
|
actual_crc32,
|
|
)
|
|
return True
|