Merge pull request #4 from shafa-dev/max_bytes-parametr

add max_bytes got get_records method

Author: bogdan-zs

kafka_bridge_client/consumer.py

@@ -195,7 +195,9 @@ async def _subscribe(self) -> None:
             )
 
     async def get_records(
-        self, *,
+        self,
+        *,
+        max_bytes: int,
         timeout: int = 10000
     ) -> t.AsyncGenerator[t.Dict[str, t.Any], None]:
         while True:
@@ -205,7 +207,7 @@ async def get_records(
                     group_id=self._group_id,
                     name=self._consumer_name,
                 ),
-                params={'timeout': timeout},
+                params={'timeout': timeout, 'max_bytes': max_bytes},
                 headers={'Accept': 'application/vnd.kafka.binary.v2+json'},
             )
             if response.status != 200:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kafka-bridge-client"
-version = "0.1.5"
+version = "0.2.0"
 description = "Python client for Strimzi Kafka Bridge"
 authors = ["Bogdan Zaseka <zaseka.bogdan@gmail.com>"]
 license = "MIT"