jararaca 0.3.11a14__py3-none-any.whl → 0.3.11a16__py3-none-any.whl

jararaca/utils/retry.py

@@ -0,0 +1,141 @@
+import asyncio
+import logging
+import random
+from functools import wraps
+from typing import Awaitable, Callable, Optional, ParamSpec, TypeVar
+
+logger = logging.getLogger(__name__)
+
+P = ParamSpec("P")
+T = TypeVar("T")
+
+
+class RetryConfig:
+    """Configuration for the retry mechanism."""
+
+    def __init__(
+        self,
+        max_retries: int = 5,
+        initial_delay: float = 1.0,
+        max_delay: float = 60.0,
+        backoff_factor: float = 2.0,
+        jitter: bool = True,
+    ):
+        """
+        Initialize retry configuration.
+
+        Args:
+            max_retries: Maximum number of retry attempts (default: 5)
+            initial_delay: Initial delay in seconds between retries (default: 1.0)
+            max_delay: Maximum delay in seconds between retries (default: 60.0)
+            backoff_factor: Multiplier for the delay after each retry (default: 2.0)
+            jitter: Whether to add randomness to the delay to prevent thundering herd (default: True)
+        """
+        self.max_retries = max_retries
+        self.initial_delay = initial_delay
+        self.max_delay = max_delay
+        self.backoff_factor = backoff_factor
+        self.jitter = jitter
+
+
+E = TypeVar("E", bound=Exception)
+
+
+async def retry_with_backoff(
+    fn: Callable[[], Awaitable[T]],
+    # args: P.args,
+    # kwargs: P.kwargs,
+    retry_config: Optional[RetryConfig] = None,
+    on_retry_callback: Optional[Callable[[int, E, float], None]] = None,
+    retry_exceptions: tuple[type[E], ...] = (),
+) -> T:
+    """
+    Execute a function with exponential backoff retry mechanism.
+
+    Args:
+        fn: The async function to execute with retry
+        *args: Arguments to pass to the function
+        retry_config: Configuration for the retry mechanism
+        on_retry_callback: Optional callback function called on each retry with retry count, exception, and next delay
+        retry_exceptions: Tuple of exception types that should trigger a retry
+        **kwargs: Keyword arguments to pass to the function
+
+    Returns:
+        The result of the function if successful
+
+    Raises:
+        The last exception encountered if all retries fail
+    """
+    if retry_config is None:
+        retry_config = RetryConfig()
+
+    last_exception = None
+    delay = retry_config.initial_delay
+
+    for retry_count in range(retry_config.max_retries + 1):
+        try:
+            return await fn()
+        except retry_exceptions as e:
+            last_exception = e
+
+            if retry_count >= retry_config.max_retries:
+                logger.error(
+                    f"Max retries ({retry_config.max_retries}) exceeded: {str(e)}"
+                )
+                raise
+
+            # Calculate next delay with exponential backoff
+            if retry_count > 0:  # Don't increase delay on the first failure
+                delay = min(delay * retry_config.backoff_factor, retry_config.max_delay)
+
+            # Apply jitter if configured (±25% randomness)
+            if retry_config.jitter:
+                jitter_amount = delay * 0.25
+                delay = delay + random.uniform(-jitter_amount, jitter_amount)
+                # Ensure delay doesn't go negative due to jitter
+                delay = max(delay, 0.1)
+
+            logger.warning(
+                f"Retry {retry_count+1}/{retry_config.max_retries} after error: {str(e)}. "
+                f"Retrying in {delay:.2f}s"
+            )
+
+            # Call the optional retry callback if provided
+            if on_retry_callback:
+                on_retry_callback(retry_count, e, delay)
+
+            await asyncio.sleep(delay)
+
+    # This should never be reached with the current implementation
+    if last_exception:
+        raise last_exception
+    raise RuntimeError("Unexpected error in retry logic")
+
+
+def with_retry(
+    retry_config: Optional[RetryConfig] = None,
+    retry_exceptions: tuple[type[Exception], ...] = (Exception,),
+) -> Callable[[Callable[P, Awaitable[T]]], Callable[P, Awaitable[T]]]:
+    """
+    Decorator to wrap an async function with retry logic.
+
+    Args:
+        retry_config: Configuration for the retry mechanism
+        retry_exceptions: Tuple of exception types that should trigger a retry
+
+    Returns:
+        Decorated function with retry mechanism
+    """
+
+    def decorator(fn: Callable[P, Awaitable[T]]) -> Callable[P, Awaitable[T]]:
+        @wraps(fn)
+        async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+            return await retry_with_backoff(
+                lambda: fn(*args, **kwargs),
+                retry_config=retry_config,
+                retry_exceptions=retry_exceptions,
+            )
+
+        return wrapper
+
+    return decorator

{jararaca-0.3.11a14.dist-info → jararaca-0.3.11a16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: jararaca
-Version: 0.3.11a14
+Version: 0.3.11a16
 Summary: A simple and fast API framework for Python
 Author: Lucas S
 Author-email: me@luscasleo.dev

{jararaca-0.3.11a14.dist-info → jararaca-0.3.11a16.dist-info}/RECORD

@@ -3,7 +3,7 @@ jararaca/__main__.py,sha256=-O3vsB5lHdqNFjUtoELDF81IYFtR-DSiiFMzRaiSsv4,67
 jararaca/broker_backend/__init__.py,sha256=GzEIuHR1xzgCJD4FE3harNjoaYzxHMHoEL0_clUaC-k,3528
 jararaca/broker_backend/mapper.py,sha256=vTsi7sWpNvlga1PWPFg0rCJ5joJ0cdzykkIc2Tuvenc,696
 jararaca/broker_backend/redis_broker_backend.py,sha256=a7DHchy3NAiD71Ix8SwmQOUnniu7uup-Woa4ON_4J7I,5786
-jararaca/cli.py,sha256=1mFBAWE_eYNc7PpRBe9cHciU6MEJrjjuRCTQZBxYx2U,19901
+jararaca/cli.py,sha256=zkWRcqllY_C0sIR7h_crlptq2cA6sxRM4nvMMLBaANs,25946
 jararaca/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/core/providers.py,sha256=wktH84FK7c1s2wNq-fudf1uMfi3CQBR0neU2czJ_L0U,434
@@ -20,7 +20,7 @@ jararaca/messagebus/interceptors/aiopika_publisher_interceptor.py,sha256=_DEHwIH
 jararaca/messagebus/interceptors/publisher_interceptor.py,sha256=ojy1bRhqMgrkQljcGGS8cd8-8pUjL8ZHjIUkdmaAnNM,1325
 jararaca/messagebus/message.py,sha256=U6cyd2XknX8mtm0333slz5fanky2PFLWCmokAO56vvU,819
 jararaca/messagebus/publisher.py,sha256=JTkxdKbvxvDWT8nK8PVEyyX061vYYbKQMxRHXrZtcEY,2173
-jararaca/messagebus/worker.py,sha256=18oD-6Ip_rOa90p53EcEPlvkXho3SWrC40l4OVSIsE4,22356
+jararaca/messagebus/worker.py,sha256=CrSIejWMGII4_JK0aH4jxdj0oBJX4hSXY0SmVa6KURA,54187
 jararaca/microservice.py,sha256=rRIimfeP2-wf289PKoUbk9wrSdA0ga_qWz5JNgQ5IE0,9667
 jararaca/observability/decorators.py,sha256=MOIr2PttPYYvRwEdfQZEwD5RxKHOTv8UEy9n1YQVoKw,2281
 jararaca/observability/interceptor.py,sha256=U4ZLM0f8j6Q7gMUKKnA85bnvD-Qa0ii79Qa_X8KsXAQ,1498
@@ -66,8 +66,9 @@ jararaca/tools/app_config/interceptor.py,sha256=HV8h4AxqUc_ACs5do4BSVlyxlRXzx7Hq
 jararaca/tools/typescript/interface_parser.py,sha256=35xbOrZDQDyTXdMrVZQ8nnFw79f28lJuLYNHAspIqi8,30492
 jararaca/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jararaca/utils/rabbitmq_utils.py,sha256=ytdAFUyv-OBkaVnxezuJaJoLrmN7giZgtKeet_IsMBs,10918
-jararaca-0.3.11a14.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-jararaca-0.3.11a14.dist-info/METADATA,sha256=QctSa-5oHXPeay_yQplmGs8d_kwMjQS_yLmQrmk4cp4,4998
-jararaca-0.3.11a14.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-jararaca-0.3.11a14.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
-jararaca-0.3.11a14.dist-info/RECORD,,
+jararaca/utils/retry.py,sha256=DzPX_fXUvTqej6BQ8Mt2dvLo9nNlTBm7Kx2pFZ26P2Q,4668
+jararaca-0.3.11a16.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+jararaca-0.3.11a16.dist-info/METADATA,sha256=E5OUx4jCVKdki-auNqJUVcOflGaKTt37W3FAtKYS7ow,4998
+jararaca-0.3.11a16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+jararaca-0.3.11a16.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
+jararaca-0.3.11a16.dist-info/RECORD,,