diff options
Diffstat (limited to 'kafka/client.py')
| -rw-r--r-- | kafka/client.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/kafka/client.py b/kafka/client.py index 99d6fec..2bd2324 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -7,12 +7,12 @@ import time import six -import kafka.common -from kafka.common import (TopicPartition, BrokerMetadata, UnknownError, - ConnectionError, FailedPayloadsError, +import kafka.errors +from kafka.errors import (UnknownError, ConnectionError, FailedPayloadsError, KafkaTimeoutError, KafkaUnavailableError, LeaderNotAvailableError, UnknownTopicOrPartitionError, NotLeaderForPartitionError, ReplicaNotAvailableError) +from kafka.structs import TopicPartition, BrokerMetadata from kafka.conn import ( collect_hosts, BrokerConnection, DEFAULT_SOCKET_TIMEOUT_SECONDS, @@ -123,7 +123,7 @@ class SimpleClient(object): # If there's a problem with finding the coordinator, raise the # provided error - kafka.common.check_error(resp) + kafka.errors.check_error(resp) # Otherwise return the BrokerMetadata return BrokerMetadata(resp.nodeId, resp.host, resp.port) @@ -389,7 +389,7 @@ class SimpleClient(object): # Or a server api error response try: - kafka.common.check_error(resp) + kafka.errors.check_error(resp) except (UnknownTopicOrPartitionError, NotLeaderForPartitionError): self.reset_topic_metadata(resp.topic) raise @@ -509,7 +509,7 @@ class SimpleClient(object): for error, topic, partitions in resp.topics: # Errors expected for new topics if error: - error_type = kafka.common.kafka_errors.get(error, UnknownError) + error_type = kafka.errors.kafka_errors.get(error, UnknownError) if error_type in (UnknownTopicOrPartitionError, LeaderNotAvailableError): log.error('Error loading topic metadata for %s: %s (%s)', topic, error_type, error) @@ -530,7 +530,7 @@ class SimpleClient(object): # Check for partition errors if error: - error_type = kafka.common.kafka_errors.get(error, UnknownError) + error_type = kafka.errors.kafka_errors.get(error, UnknownError) # If No Leader, topics_to_brokers topic_partition -> None if error_type is LeaderNotAvailableError: |