diff options
Diffstat (limited to 'kafka/producer.py')
| -rw-r--r-- | kafka/producer.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/kafka/producer.py b/kafka/producer.py index 823d923..93d6e3b 100644 --- a/kafka/producer.py +++ b/kafka/producer.py @@ -13,10 +13,11 @@ class SimpleProducer(object): def __init__(self, client, topic): self.client = client self.topic = topic - self.client.load_metadata_for_topics(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() + resp = self.client.send_produce_request([req])[0] + assert resp.error == 0 |