clean local metadata before requesting refresh in load_metadata_for_topics()

1 files changed, 7 insertions, 2 deletions

diff --git a/kafka/client.py b/kafka/client.py
index a966f92..f01c1c5 100644
--- a/kafka/client.py
+++ b/kafka/client.py
@@ -311,6 +311,13 @@ class KafkaClient(object):
         (a single partition w/o a leader, for example)
         """
         topics = [kafka_bytestring(t) for t in topics]
+
+        if topics:
+            for topic in topics:
+                self.reset_topic_metadata(topic)
+        else:
+            self.reset_all_metadata()
+
         resp = self.send_metadata_request(topics)
 
         log.debug("Broker metadata: %s", resp.brokers)
@@ -323,8 +330,6 @@ class KafkaClient(object):
             topic = topic_metadata.topic
             partitions = topic_metadata.partitions
 
-            self.reset_topic_metadata(topic)
-
             # Errors expected for new topics
             try:
                 kafka.common.check_error(topic_metadata)