'commit'

DouYin/dy-downloader/control/retry_handler.py

@@ -0,0 +1,29 @@
+import asyncio
+from typing import Callable, Any, TypeVar
+from utils.logger import setup_logger
+
+logger = setup_logger('RetryHandler')
+
+T = TypeVar('T')
+
+
+class RetryHandler:
+    def __init__(self, max_retries: int = 3):
+        self.max_retries = max_retries
+        self.retry_delays = [1, 2, 5]
+
+    async def execute_with_retry(self, func: Callable[..., T], *args, **kwargs) -> T:
+        last_error = None
+
+        for attempt in range(self.max_retries):
+            try:
+                return await func(*args, **kwargs)
+            except Exception as e:
+                last_error = e
+                if attempt < self.max_retries - 1:
+                    delay = self.retry_delays[min(attempt, len(self.retry_delays) - 1)]
+                    logger.warning(f"Attempt {attempt + 1} failed: {e}, retrying in {delay}s...")
+                    await asyncio.sleep(delay)
+
+        logger.error(f"All {self.max_retries} attempts failed: {last_error}")
+        raise last_error