1 files changed, 10 insertions, 16 deletions
diff --git a/example.py b/example.py
index 9329312..1e4bd8b 100644
--- a/example.py
+++ b/example.py
@@ -2,27 +2,21 @@ import logging
 
 from kafka.client import KafkaClient, FetchRequest, ProduceRequest
 from kafka.consumer import SimpleConsumer
+from kafka.producer import SimpleProducer
 
-def produce_example(kafka):
-    message = kafka.create_message("testing")
-    request = ProduceRequest("my-topic", -1, [message])
-    kafka.send_message_set(request)
+def produce_example(client):
+    producer = SimpleProducer(client, "my-topic")
+    producer.send_message("test")
 
-def consume_example(kafka):
-    request = FetchRequest("my-topic", 0, 0, 1024)
-    (messages, nextRequest) = kafka.get_message_set(request)
-    for message in messages:
-        print("Got Message: %s" % (message,))
-    print(nextRequest)
-
-def produce_gz_example(kafka):
-    message = kafka.create_gzip_message("this message was gzipped", "along with this one")
-    request = ProduceRequest("my-topic", 0, [message])
-    kafka.send_message_set(request)
+def consume_example(client):
+    consumer = SimpleConsumer(client, "test-group", "my-topic")
+    for message in consumer:
+        print(message)
 
 def main():
     client = KafkaClient("localhost", 9092)
-    consumer = SimpleConsumer(client, "test-group", "my-topic")
+    produce_example(client)
+    consume_example(client)
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.DEBUG)