Adding initial support for gzip compression

1 files changed, 11 insertions, 8 deletions

diff --git a/example.py b/example.py
index c58751b..0204af1 100644
--- a/example.py
+++ b/example.py
@@ -1,27 +1,30 @@
 import logging
 
 from kafka import KafkaClient, FetchRequest, ProduceRequest
-from kafka import create_message_from_string
 
 def produce_example(kafka):
-    message = create_message_from_string("test")
+    message = kafka.create_message_from_string("testing")
     request = ProduceRequest("my-topic", 0, [message])
-    print("Sending %s" % str(request))
     kafka.send_message_set(request)
 
 def consume_example(kafka):
-    request = FetchRequest("my-topic", 0, 0, 64)
-    print("Sending %s" % str(request))
+    request = FetchRequest("my-topic", 0, 0, 1024)
     (messages, nextRequest) = kafka.get_message_set(request)
-    print("Got %d messages:" % len(messages))
     for message in messages:
-        print("\t%s" % message.payload)
-    print("Next request %s" % str(nextRequest))
+        print("Got Message: %s" % (message,))
+    print(nextRequest)
+
+def produce_gz_example(kafka):
+    message = kafka.create_gzipped_message("this message was gzipped", "along with this one")
+    request = ProduceRequest("my-topic", 0, [message])
+    kafka.send_message_set(request)
 
 def main():
     kafka = KafkaClient("localhost", 9092)
     produce_example(kafka)
+    produce_gz_example(kafka)
     consume_example(kafka)
+    kafka.close()
 
 
 if __name__ == "__main__":