Throw KafkaUnavailableError when no brokers available

Author: turtlesoupy

kafka/client.py

@@ -8,7 +8,7 @@
 from kafka.common import (ErrorMapping, TopicAndPartition,
                           ConnectionError, FailedPayloadsError,
                           BrokerResponseError, PartitionUnavailableError,
-                          KafkaRequestError)
+                          KafkaUnavailableError, KafkaRequestError)
 
 from kafka.conn import KafkaConnection
 from kafka.protocol import KafkaProtocol
@@ -78,7 +78,7 @@ def _send_broker_unaware_request(self, requestId, request):
                             "trying next server: %s" % (request, conn, e))
                 continue
 
-        raise BrokerResponseError("All servers failed to process request")
+        raise KafkaUnavailableError("All servers failed to process request")
 
     def _send_broker_aware_request(self, payloads, encoder_fn, decoder_fn):
         """

kafka/common.py

@@ -78,6 +78,10 @@ class KafkaRequestError(KafkaError):
     pass
 
 
+class KafkaUnavailableError(KafkaError):
+    pass
+
+
 class BrokerResponseError(KafkaError):
     pass