Skip to content

Commit

Permalink
Merge pull request #16 from cazeaux/dev
Browse files Browse the repository at this point in the history
fix: fixed connection management to avoid connection pool warnings
  • Loading branch information
cazeaux authored Mar 16, 2024
2 parents 8a41950 + 7db0d49 commit 0efa5df
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions custom_components/iracing/iracingapi/iracingapi.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,9 @@ class irDataClient:
def __init__(self, username=None, password=None, logger=None):
self.authenticated = False
self.session = requests.Session()
adapter = requests.adapters.HTTPAdapter(pool_connections=30, pool_maxsize=30)
self.session.mount("https://", adapter)

self.base_url = "https://members-ng.iracing.com"

self.username = username
Expand Down Expand Up @@ -117,6 +120,7 @@ def _get_resource_or_link(self, url, params=None):

raise RuntimeError("Unhandled Non-200 response", r)
data = r.json()
r.close()
if not isinstance(data, list) and "link" in data.keys():
return [data.get("link"), True]
else:
Expand All @@ -128,6 +132,7 @@ def _get_resource(self, endpoint, params=None):
if not is_link:
return resource_obj
r = self.session.get(resource_obj)
r.close()
if r.status_code != 200:
raise RuntimeError("Unhandled Non-200 response", r)
return r.json()
Expand Down

0 comments on commit 0efa5df

Please sign in to comment.