kenzuya/unshackle
1
0
Fork 2
mirror of https://github.com/unshackle-dl/unshackle.git synced 2026-11-12 20:15:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(vaults): size the HTTP vault connection pool for parallel lookups

Browse Source
This commit is contained in:
imSp4rky
2026-10-15 16:16:05 -06:00
parent 89f433963a
commit fae0f97102
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
unshackle/vaults/API.py
+5
View File
@@ -2,8 +2,10 @@ from typing import Iterator, Optional, Union
from uuid import UUID
from requests import Response, Session
from requests.adapters import HTTPAdapter
from unshackle.core import __version__
from unshackle.core.session import POOL_MAX_SIZE
from unshackle.core.vault import Vault
@@ -23,6 +25,9 @@ class API(Vault):
self.uri = uri.rstrip("/")
self.timeout = timeout
self.session = Session()
adapter = HTTPAdapter(pool_connections=POOL_MAX_SIZE, pool_maxsize=POOL_MAX_SIZE)
self.session.mount("https://", adapter)
self.session.mount("http://", adapter)
self.session.headers.update({"User-Agent": f"unshackle v{__version__}"})
self.session.headers.update({"Authorization": f"Bearer {token}"})
unshackle/vaults/HTTP.py
+5
View File
@@ -4,8 +4,10 @@ from typing import Iterator, Optional, Union
from uuid import UUID
from requests import Session
from requests.adapters import HTTPAdapter
from unshackle.core import __version__
from unshackle.core.session import POOL_MAX_SIZE
from unshackle.core.vault import Vault
@@ -60,6 +62,9 @@ class HTTP(Vault):
self.batch_insert = True
self.current_title = None
self.session = Session()
adapter = HTTPAdapter(pool_connections=POOL_MAX_SIZE, pool_maxsize=POOL_MAX_SIZE)
self.session.mount("https://", adapter)
self.session.mount("http://", adapter)
self.session.headers.update({"User-Agent": f"unshackle v{__version__}"})
self.api_session_id = None