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)