Retry Client#partitions_for up to two times

Author: dasch

lib/kafka/client.rb

@@ -700,7 +700,22 @@ def has_topic?(topic)
     # @param topic [String]
     # @return [Integer] the number of partitions in the topic.
     def partitions_for(topic)
-      @cluster.partitions_for(topic).count
+      attempt = 1
+
+      begin
+        @cluster.partitions_for(topic).count
+      rescue LeaderNotAvailable
+        if attempt >= 3
+          raise
+        else
+          attempt += 1
+          @logger.warn "Leader not available, retrying in 1s..."
+
+          sleep 1
+
+          retry
+        end
+      end
     end
 
     # Counts the number of replicas for a topic's partition