Skip to content

Commit 1a71ea6

Browse files
authored
allow max_pool_size to be defined from a caller
1 parent 0a92662 commit 1a71ea6

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

reportportal_client/service.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,7 @@ def __init__(self,
170170
is_skipped_an_issue=True,
171171
verify_ssl=True,
172172
retries=None,
173+
max_pool_size=50,
173174
**kwargs):
174175
"""Init the service class.
175176
@@ -182,6 +183,9 @@ def __init__(self,
182183
is_skipped_an_issue: option to mark skipped tests as not
183184
'To Investigate' items on Server side.
184185
verify_ssl: option to not verify ssl certificates
186+
max_pool_size: option to set the maximum number of
187+
connections to save in the pool.
188+
185189
"""
186190
self._batch_logs = []
187191
self.endpoint = endpoint
@@ -191,14 +195,13 @@ def __init__(self,
191195
self.is_skipped_an_issue = is_skipped_an_issue
192196
self.base_url_v1 = uri_join(self.endpoint, "api/v1", self.project)
193197
self.base_url_v2 = uri_join(self.endpoint, "api/v2", self.project)
194-
self.max_pool_size = 50
195198

196199
self.session = requests.Session()
197200
if retries:
198201
self.session.mount('https://', HTTPAdapter(
199-
max_retries=retries, pool_maxsize=self.max_pool_size))
202+
max_retries=retries, pool_maxsize=max_pool_size))
200203
self.session.mount('http://', HTTPAdapter(
201-
max_retries=retries, pool_maxsize=self.max_pool_size))
204+
max_retries=retries, pool_maxsize=max_pool_size))
202205
self.session.headers["Authorization"] = "bearer {0}".format(self.token)
203206
self.launch_id = None
204207
self.verify_ssl = verify_ssl

0 commit comments

Comments
 (0)