add function bootstrap_connected

Author: Wayde

kafka/consumer/group.py

@@ -377,6 +377,12 @@ def __init__(self, *topics, **configs):
             self._subscription.subscribe(topics=topics)
             self._client.set_topics(topics)
 
+    def bootstrap_connected(self):
+        """Return True if the bootstrap is connected."""
+        if self._client._bootstrap_fails > 0:
+            return False
+        return True
+
     def assign(self, partitions):
         """Manually assign a list of TopicPartitions to this consumer.
 

kafka/producer/kafka.py

@@ -402,6 +402,12 @@ def __init__(self, **configs):
         atexit.register(self._cleanup)
         log.debug("Kafka producer started")
 
+    def bootstrap_connected(self):
+        """Return True if the bootstrap is connected."""
+        if self._client._bootstrap_fails > 0:
+            return False
+        return True
+
     def _cleanup_factory(self):
         """Build a cleanup clojure that doesn't increase our ref count"""
         _self = weakref.proxy(self)