def _register_for_operations(config, session, service_name):
# There's certainly a tradeoff for registering the retry config
# for the operations when the service is created. In practice,
# there aren't a whole lot of per operation retry configs so
# this is ok for now.
for key in config:
if key == '__default__':
continue
handler = retryhandler.create_retry_handler(config, key)
unique_id = 'retry-config-%s-%s' % (service_name, key)
session.register('needs-retry.%s.%s' % (service_name, key),
handler, unique_id=unique_id)
def _register_for_operations(config, session, service_name):
# There's certainly a tradeoff for registering the retry config
# for the operations when the service is created. In practice,
# there aren't a whole lot of per operation retry configs so
# this is ok for now.
for key in config:
if key == '__default__':
continue
handler = retryhandler.create_retry_handler(config, key)
unique_id = 'retry-config-%s-%s' % (service_name, key)
session.register('needs-retry.%s.%s' % (service_name, key),
handler,
unique_id=unique_id)
def register_retries_for_service(service, **kwargs):
loader = service.session.get_component('data_loader')
config = _load_retry_config(loader, service.endpoint_prefix)
if not config:
return
logger.debug("Registering retry handlers for service: %s", service)
session = service.session
handler = retryhandler.create_retry_handler(
config, service.endpoint_prefix)
unique_id = 'retry-config-%s' % service.endpoint_prefix
session.register('needs-retry.%s' % service.endpoint_prefix,
handler, unique_id=unique_id)
_register_for_operations(config, session,
service_name=service.endpoint_prefix)
def register_retries_for_service(service, **kwargs):
loader = service.session.get_component('data_loader')
config = _load_retry_config(loader, service.endpoint_prefix)
if not config:
return
logger.debug("Registering retry handlers for service: %s", service)
session = service.session
handler = retryhandler.create_retry_handler(config,
service.endpoint_prefix)
unique_id = 'retry-config-%s' % service.endpoint_prefix
session.register('needs-retry.%s' % service.endpoint_prefix,
handler,
unique_id=unique_id)
_register_for_operations(config,
session,
service_name=service.endpoint_prefix)
