Basic impl

robust/exception.py

@@ -4,3 +4,7 @@ class ContinuousFailureException(Exception):
 
 class TimeoutException(Exception):
     pass
+
+
+class ConnectionCutException(Exception):
+    pass

robust/tests.py

@@ -1,6 +1,8 @@
 from pytest import raises
 from robust.tools import retry, timeout, breaker
-from robust.exception import ContinuousFailureException, TimeoutException
+from robust.exception import (ContinuousFailureException,
+                              TimeoutException,
+                              ConnectionCutException)
 
 
 class TestRetryCase(object):

robust/tools.py

@@ -59,3 +59,22 @@ def breaker(limit, revive, on_fail=None):
     After :revive: seconds it allows one connection to pass.
     If it succeeds - counter is reset, if doesn't - we wait another :revive: seconds
     """
+
+    def injector(fn):
+        counter = 0
+
+        @wraps(fn)
+        def wrapper(*args, **kwargs):
+            nonlocal counter
+            if counter > limit:
+                pass
+
+            try:
+                return fn(*args, **kwargs)
+            except Exception:
+                counter += 1
+                raise
+
+        return wrapper
+
+    return injector