Added chunk size parameter

Panos · Panos · commit 7dddb0baed32 · 2017-02-16T15:40:51.000Z
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -308,7 +308,8 @@ def query(self,
               expected_response_code=200,
               database=None,
               raise_errors=True,
-              chunked=False):
+              chunked=False,
+              chunk_size=0):
         """Send a query to InfluxDB.
 
         :param query: the actual query string
@@ -333,6 +334,9 @@ def query(self,
             is returned as opposed to a list.
         :type chunked: bool
 
+        :param chunk_size: Size of each chunk to tell InfluxDB to use.
+        :type chunk_size: int
+
         :returns: the queried data
         :rtype: :class:`~.ResultSet`
         """
@@ -355,6 +359,8 @@ def query(self,
 
         if chunked or 'chunked' in params:
             params['chunked'] = 'true'
+            if chunk_size > 0:
+                params['chunk_size'] = chunk_size
             return self._read_chunked_response(response)
 
         data = response.json()
diff --git a/influxdb/tests/client_test.py b/influxdb/tests/client_test.py
@@ -813,7 +813,7 @@ def test_chunked_response(self):
                 text=example_response
             )
             response = list(self.cli.query('show series limit 4 offset 0',
-                                           chunked=True))
+                                           chunked=True, chunk_size=4))
             self.assertTrue(len(response) == 4)
             self.assertEqual(response[0].raw, ResultSet(
                 {"statement_id": 0,

Original file line number	Diff line number	Diff line change
`@@ -813,7 +813,7 @@ def test_chunked_response(self):`
`813`	`813`	`text=example_response`
`814`	`814`	`)`
`815`	`815`	`response = list(self.cli.query('show series limit 4 offset 0',`
`816`		`- chunked=True))`
	`816`	`+ chunked=True, chunk_size=4))`
`817`	`817`	`self.assertTrue(len(response) == 4)`
`818`	`818`	`self.assertEqual(response[0].raw, ResultSet(`
`819`	`819`	`{"statement_id": 0,`