Skip to content

Commit 72399ad

Browse files
committed
feat: Pass timeout to read and write queries
1 parent bf1c1df commit 72399ad

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

influxdb_client/client/query_api.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -146,8 +146,10 @@ def query_data_frame_stream(self, query: str, org=None, data_frame_index: List[s
146146
if org is None:
147147
org = self._influxdb_client.org
148148

149+
request_timeout = self._influxdb_client.api_client.configuration.timeout
149150
response = self._query_api.post_query(org=org, query=self._create_query(query, self.default_dialect, params),
150-
async_req=False, _preload_content=False, _return_http_data_only=False)
151+
async_req=False, _preload_content=False, _return_http_data_only=False,
152+
_request_timeout=request_timeout)
151153

152154
_parser = FluxCsvParser(response=response, serialization_mode=FluxSerializationMode.dataFrame,
153155
data_frame_index=data_frame_index)

influxdb_client/client/write_api.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -368,8 +368,9 @@ def _http(self, batch_item: _BatchItem):
368368

369369
retry = self._write_options.to_retry_strategy()
370370

371+
request_timeout = self._influxdb_client.api_client.configuration.timeout
371372
self._post_write(False, batch_item.key.bucket, batch_item.key.org, batch_item.data,
372-
batch_item.key.precision, urlopen_kw={'retries': retry})
373+
batch_item.key.precision, urlopen_kw={'retries': retry}, _request_timeout=request_timeout)
373374

374375
logger.debug("Write request finished %s", batch_item)
375376

0 commit comments

Comments
 (0)