lib/kafka/cluster.rb in ruby-kafka-0.3.5 vs lib/kafka/cluster.rb in ruby-kafka-0.3.6

@@ -96,10 +96,14 @@
       raise Kafka::Error, "Failed to find group coordinator"
     end
 
     def partitions_for(topic)
       add_target_topics([topic])
+      refresh_metadata_if_necessary!
       cluster_info.partitions_for(topic)
+    rescue Kafka::ProtocolError
+      mark_as_stale!
+      raise
     end
 
     def topics
       cluster_info.topics.map(&:topic_name)
     end