Merge pull request confluentinc#28 from confluentinc/README_impr_cons_example

edenhill · web-flow · commit 9bfdc20d7d77 · 2016-08-09T08:35:42.000+02:00
Improve consumer example to not exit when end of partition is reached
diff --git a/README.md b/README.md
@@ -30,7 +30,7 @@ p.flush()
 **High-level Consumer:**
 
 ```python
-from confluent_kafka import Consumer
+from confluent_kafka import Consumer, KafkaError
 
 c = Consumer({'bootstrap.servers': 'mybroker', 'group.id': 'mygroup',
               'default.topic.config': {'auto.offset.reset': 'smallest'}})
@@ -40,7 +40,8 @@ while running:
     msg = c.poll()
     if not msg.error():
         print('Received message: %s' % msg.value().decode('utf-8'))
-    else:
+    elif msg.error().code() != KafkaError._PARTITION_EOF:
+        print(msg.error())
         running = False
 c.close()
 ```
