Modified SimpleProducer to randomize the initial round robin ordering

of partitions to prevent the first message from always being published
to partition 0.

Author: alexcb

kafka/producer.py

@@ -2,6 +2,7 @@
 
 import logging
 import time
+import random
 
 from Queue import Empty
 from collections import defaultdict
@@ -197,7 +198,9 @@ def _next_partition(self, topic):
         if topic not in self.partition_cycles:
             if topic not in self.client.topic_partitions:
                 self.client.load_metadata_for_topics(topic)
-            self.partition_cycles[topic] = cycle(self.client.topic_partitions[topic])
+            randomly_ordered_partitions = self.client.topic_partitions[topic][:]
+            random.shuffle(randomly_ordered_partitions)
+            self.partition_cycles[topic] = cycle(randomly_ordered_partitions)
         return self.partition_cycles[topic].next()
 
     def send_messages(self, topic, *msg):