internal 1.1.6__tar.gz → 1.1.8__tar.gz

{internal-1.1.6 → internal-1.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: internal
-Version: 1.1.6
+Version: 1.1.8
 Summary: 
 Author: Ray
 Author-email: ray@cruisys.com

{internal-1.1.6 → internal-1.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "internal"
-version = "1.1.6"
+version = "1.1.8"
 description = ""
 authors = [{ name = "Ray", email = "ray@cruisys.com" }]
 readme = "README.md"

{internal-1.1.6 → internal-1.1.8}/src/internal/base_config.py

@@ -11,9 +11,10 @@ class BaseConfig(BaseSettings):
     LOGGER_REQUEST_ENABLE: bool = True
 
     # Request
-    REQUEST_VERIFY_SSL: bool = True
+    REQUEST_VERIFY_SSL: bool = False
     REQUEST_PROXY: str = ''
-    REQUEST_RETRY: int = 5
+    REQUEST_RETRY_COUNT: int = 3
+    REQUEST_RETRY_DELAY_INITIAL_SECONDS: int = 1
     REQUEST_CONN_POOL_TIMEOUT: float = 5
     REQUEST_CONN_TIMEOUT: float = 5
     REQUEST_WRITE_TIMEOUT: float = 5

{internal-1.1.6 → internal-1.1.8}/src/internal/http/requests.py

@@ -11,9 +11,31 @@ from ..const import CORRELATION_ID_HEADER_KEY_NAME
 from ..exception.internal_exception import GatewayTimeoutException, BadGatewayException
 
 
+async def invoke_request(timeout: httpx.Timeout, method: str, url: str, app: FastAPI, **kwargs):
+    try:
+        async with httpx.AsyncClient(timeout=timeout, verify=app.state.config.REQUEST_VERIFY_SSL) as client:
+            if "json" in kwargs:
+                kwargs["json"] = jsonable_encoder(kwargs["json"])
+
+            app.state.logger.info(f"invoke_request() request, url: {method} {url} \nkwargs: {kwargs}")
+            response = await client.request(method, url, **kwargs)
+            app.state.logger.info(
+                f"invoke_request() response, url: {method} {url} \nkwargs: {kwargs} \n\nresponse.status_code: {response.status_code} \nresponse.text: {response.text}"
+            )
+            return response
+    except httpx.TimeoutException as exc:
+        app.state.logger.warn(f"invoke_request(), TimeoutException, exc: {exc}, url: {url}, method: {method}, kwargs: {kwargs}")
+        raise GatewayTimeoutException(str(exc)) from exc
+    except Exception as exc:
+        app.state.logger.warn(
+            f"invoke_request(), Exception, exc: {exc}, url: {url}, method: {method}, kwargs: {kwargs}")
+        raise BadGatewayException(str(exc))
+
+
 async def async_request(app: FastAPI, method, url, current_user: dict = None,
                         request_conn_pool_timeout: float = 0, request_conn_timeout: float = 0,
                         request_write_timeout: float = 0, response_read_timeout: float = 0,
+                        request_retry_count: int = 0, request_retry_delay: float = 0,
                         **kwargs):
     if request_conn_pool_timeout <= 0:
         request_conn_pool_timeout = app.state.config.REQUEST_CONN_POOL_TIMEOUT
@@ -23,6 +45,10 @@ async def async_request(app: FastAPI, method, url, current_user: dict = None,
         request_write_timeout = app.state.config.REQUEST_WRITE_TIMEOUT
     if response_read_timeout <= 0:
         response_read_timeout = app.state.config.RESPONSE_READ_TIMEOUT
+    if request_retry_count <= 0:
+        request_retry_count = app.state.config.REQUEST_RETRY_COUNT
+    if request_retry_delay <= 0:
+        request_retry_delay = app.state.config.REQUEST_RETRY_DELAY_INITIAL_SECONDS
 
     timeout = httpx.Timeout(connect=request_conn_timeout, read=response_read_timeout,
                             write=request_write_timeout, pool=request_conn_pool_timeout)
@@ -42,24 +68,30 @@ async def async_request(app: FastAPI, method, url, current_user: dict = None,
                 "Authorization": f"Bearer {current_user.get('access_token')}"
             }
 
-    try:
-        async with httpx.AsyncClient(timeout=timeout, verify=False) as client:
-            if "json" in kwargs:
-                kwargs["json"] = jsonable_encoder(kwargs["json"])
+    if request_retry_count <= 0:
+        response = await invoke_request(timeout, method, url, app, **kwargs)
+        return response
+    else:
+        retries = 0
+        current_delay = float(request_retry_delay)
 
-            app.state.logger.info(f"async_request() request, url: {method} {url} \nkwargs: {kwargs}")
-            response = await client.request(method, url, **kwargs)
-            app.state.logger.info(
-                f"async_request() response, url: {method} {url} \nkwargs: {kwargs} \n\nresponse.status_code: {response.status_code} \nresponse.text: {response.text}")
-            return response
-    except httpx.TimeoutException as exc:
-        app.state.logger.warn(
-            f"async_request(), TimeoutException, exc: {exc}, url: {url}, method: {method}, kwargs: {kwargs}")
-        raise GatewayTimeoutException(str(exc))
-    except Exception as exc:
-        app.state.logger.warn(
-            f"async_request(), Exception, exc: {exc}, url: {url}, method: {method}, kwargs: {kwargs}")
-        raise BadGatewayException(str(exc))
+        while retries <= request_retry_count:
+            app.state.logger.warn(f"嘗試送請求 (第 {retries + 1} 次嘗試)...")
+            try:
+                # 使用 await 關鍵字等待異步請求完成
+                response = await invoke_request(timeout, method, url, app, **kwargs)
+                return response
+            except GatewayTimeoutException as e:
+                if retries < request_retry_count:
+                    # 計算下一次的延遲時間：current_delay * 2^retries + 隨機抖動
+                    sleep_time = current_delay * (2 ** retries) + random.uniform(0, 0.5)
+                    app.state.logger.warn(f"等待 {sleep_time:.2f} 秒後重試...")
+                    # 使用 asyncio.sleep 進行異步等待，不會阻塞主執行緒
+                    await asyncio.sleep(sleep_time)
+                    retries += 1
+                else:
+                    app.state.logger.warn(f"已達到最大重試次數 ({request_retry_count})，放棄發送請求。")
+                    raise  # 重新拋出最後一個異常
 
 
 async def invoke_webhook_message_api(app: FastAPI, message: str):