Provide a default timeout for AvroConsumer equal to the Consumer default

kwilcox · kwilcox · commit d9d291e3eed5 · 2016-12-12T15:48:44.000-05:00
diff --git a/confluent_kafka/avro/__init__.py b/confluent_kafka/avro/__init__.py
@@ -133,14 +133,15 @@ def __init__(self, config):
         super(AvroConsumer, self).__init__(config)
         self._serializer = MessageSerializer(CachedSchemaRegistryClient(url=schem_registry_url))
 
-    def poll(self, timeout):
+    def poll(self, timeout=None):
         """
         This is an overriden method from confluent_kafka.Consumer class. This handles message
         deserialization using avro schema
 
         @:param timeout
         @:return message object with deserialized key and value as dict objects
         """
+        timeout = timeout or -1
         message = super(AvroConsumer, self).poll(timeout)
         if not message:
             return message
