Add pool size parameter to client constructor (influxdata#534)

vaniakov · xginn8 · commit 530b4c504c11 · 2017-11-16T21:37:30.000-05:00
* Add pool size parameter to client constructor

* come back removed newlines

* fix flake8: line too long

* Cast InfluxDBClient pool_size parameter to int

* Move pool_size parameter to the end of args to prevent tests fail
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -37,6 +37,8 @@ class InfluxDBClient(object):
     :type username: str
     :param password: password of the user, defaults to 'root'
     :type password: str
+    :param pool_size: urllib3 connection pool size, defaults to 10.
+    :type pool_size: int
     :param database: database name to connect to, defaults to None
     :type database: str
     :param ssl: use https instead of http to connect to InfluxDB, defaults to
@@ -72,6 +74,7 @@ def __init__(self,
                  use_udp=False,
                  udp_port=4444,
                  proxies=None,
+                 pool_size=10,
                  ):
         """Construct a new InfluxDBClient object."""
         self.__host = host
@@ -87,6 +90,11 @@ def __init__(self,
         self.__use_udp = use_udp
         self.__udp_port = udp_port
         self._session = requests.Session()
+        adapter = requests.adapters.HTTPAdapter(
+            pool_connections=int(pool_size),
+            pool_maxsize=int(pool_size)
+        )
+
         if use_udp:
             self.udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
 
@@ -95,6 +103,8 @@ def __init__(self,
         if ssl is True:
             self._scheme = "https"
 
+        self._session.mount(self._scheme, adapter)
+
         if proxies is None:
             self._proxies = {}
         else:
