Throw KafkaUnavailableError when no brokers available
This commit is contained in:
@@ -8,7 +8,7 @@ from itertools import count
|
|||||||
from kafka.common import (ErrorMapping, TopicAndPartition,
|
from kafka.common import (ErrorMapping, TopicAndPartition,
|
||||||
ConnectionError, FailedPayloadsError,
|
ConnectionError, FailedPayloadsError,
|
||||||
BrokerResponseError, PartitionUnavailableError,
|
BrokerResponseError, PartitionUnavailableError,
|
||||||
KafkaRequestError)
|
KafkaUnavailableError, KafkaRequestError)
|
||||||
|
|
||||||
from kafka.conn import KafkaConnection
|
from kafka.conn import KafkaConnection
|
||||||
from kafka.protocol import KafkaProtocol
|
from kafka.protocol import KafkaProtocol
|
||||||
@@ -78,7 +78,7 @@ class KafkaClient(object):
|
|||||||
"trying next server: %s" % (request, conn, e))
|
"trying next server: %s" % (request, conn, e))
|
||||||
continue
|
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):
|
def _send_broker_aware_request(self, payloads, encoder_fn, decoder_fn):
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -78,6 +78,10 @@ class KafkaRequestError(KafkaError):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class KafkaUnavailableError(KafkaError):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
class BrokerResponseError(KafkaError):
|
class BrokerResponseError(KafkaError):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user