Browse Source

超时机制和连接池

AnlaAnla 1 month ago
parent
commit
422a55d19f
1 changed files with 4 additions and 1 deletions
  1. 4 1
      app/api/auto_import.py

+ 4 - 1
app/api/auto_import.py

@@ -187,7 +187,10 @@ async def auto_import_script_api(
 
     is_reflect_str = "true" if is_reflect_card else "false"
 
-    async with aiohttp.ClientSession() as session:
+    timeout = aiohttp.ClientTimeout(total=180, connect=5, sock_connect=5, sock_read=120)
+    connector = aiohttp.TCPConnector(limit=20, force_close=True)
+
+    async with aiohttp.ClientSession(timeout=timeout, connector=connector) as session:
         try:
             main_bytes_data = {k: (await v.read(), v.filename) for k, v in valid_main_files.items()}
             gray_bytes_data = {k: (await v.read(), v.filename) for k, v in valid_gray_files.items()}