Use KafkaProducer / KafkaConsumer in example.py

1 files changed, 12 insertions, 13 deletions

diff --git a/example.py b/example.py
index 062761b..a1a1e1e 100755
--- a/example.py
+++ b/example.py
@@ -1,21 +1,18 @@
 #!/usr/bin/env python
 import threading, logging, time
 
-from kafka.client import KafkaClient
-from kafka.consumer import SimpleConsumer
-from kafka.producer import SimpleProducer
+from kafka import KafkaConsumer, KafkaProducer
+
 
 class Producer(threading.Thread):
     daemon = True
 
     def run(self):
-        client = KafkaClient("localhost:9092")
-        producer = SimpleProducer(client)
+        producer = KafkaProducer(bootstrap_servers='localhost:9092')
 
         while True:
-            producer.send_messages('my-topic', "test")
-            producer.send_messages('my-topic', "\xc2Hola, mundo!")
-
+            producer.send('my-topic', b"test")
+            producer.send('my-topic', b"\xc2Hola, mundo!")
             time.sleep(1)
 
 
@@ -23,11 +20,13 @@ class Consumer(threading.Thread):
     daemon = True
 
     def run(self):
-        client = KafkaClient("localhost:9092")
-        consumer = SimpleConsumer(client, "test-group", "my-topic")
+        consumer = KafkaConsumer(bootstrap_servers='localhost:9092',
+                                 auto_offset_reset='earliest')
+        consumer.subscribe(['my-topic'])
 
         for message in consumer:
-            print(message)
+            print (message)
+
 
 def main():
     threads = [
@@ -38,11 +37,11 @@ def main():
     for t in threads:
         t.start()
 
-    time.sleep(5)
+    time.sleep(10)
 
 if __name__ == "__main__":
     logging.basicConfig(
         format='%(asctime)s.%(msecs)s:%(name)s:%(thread)d:%(levelname)s:%(process)d:%(message)s',
-        level=logging.DEBUG
+        level=logging.INFO
         )
     main()