IO retry unreliable remote
ActivePublic
Actions

Authored by tenma on Dec 4 2020, 3:02 PM.

Tags

None

Subscribers

None

	import logging
	from tenacity import (
	before_sleep_log,
	retry,
	stop_after_attempt,
	wait_random_exponential,
	)

	class UnreliableRemote:
	def __init__(self, responses):
	self.responses = responses
	self.counter = 0
	def request(self):
	res = self.responses[self.counter]
	self.counter += 1
	if isinstance(res, Exception):
	raise res
	return res

	def should_retry(retry_state) -> bool:
	attempt = retry_state.outcome
	if attempt.failed:
	return isinstance(attempt.exception(), IOError)
	return False

	io_retry = retry(
	retry=should_retry,
	wait=wait_random_exponential(multiplier=1),
	stop=stop_after_attempt(3),
	before_sleep=before_sleep_log(logging.getLogger(__name__), logging.DEBUG),
	)
	exc = IOError("RemoteException")
	responses = [exc, exc, exc, 3]

	def test_io_retry():
	r = UnreliableRemote(responses)
	r.request = io_retry(r.request)

	r.request()
	# DEBUG:__main__:Retrying __main__.UnreliableRemote.request in 0.8281996260225647 seconds as it raised OSError: RemoteException.
	# RetryError: RetryError[<Future at 0x7fededd759b0 state=finished raised OSError>]

IO retry unreliable remoteActivePublicActions