Big code re-org

1 files changed, 22 insertions, 0 deletions

diff --git a/kafka/producer.py b/kafka/producer.py
new file mode 100644
index 0000000..823d923
--- /dev/null
+++ b/kafka/producer.py
@@ -0,0 +1,22 @@
+from itertools import cycle
+import logging
+
+from kafka.common import ProduceRequest
+from kafka.protocol import create_message
+
+log = logging.getLogger("kafka")
+
+class SimpleProducer(object):
+    """
+    A simple, round-robbin producer. Each message goes to exactly one partition
+    """
+    def __init__(self, client, topic):
+        self.client = client
+        self.topic = topic
+        self.client.load_metadata_for_topics(topic)
+        self.next_partition = cycle(self.client.topic_partitions[topic])
+
+    def send_message(self, msg):
+        req = ProduceRequest(self.topic, self.next_partition.next(),
+            messages=[create_message(msg)])
+        resp = self.client.send_produce_request([req]).next()