added ZKConnectError so we're not bubbling exceptions from zc.zk

amorton · Dec 15, 2011 · af353eb · af353eb
1 parent 1d070bd
commit af353eb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/brod/zk.py b/brod/zk.py
@@ -14,7 +14,7 @@
 from itertools import chain
 
 import zookeeper
-from zc.zk import ZooKeeper
+from zc.zk import ZooKeeper, FailedConnect
 
 from brod.base import BrokerPartition, FetchResult, KafkaError, MessageSet
 from brod.blocking import Kafka
@@ -23,6 +23,7 @@
 
 class NoAvailablePartitionsError(KafkaError): pass
 class ConsumerEntryNotFoundError(KafkaError): pass
+class ZKConnectError(KafkaError): pass
 
 class ZKUtil(object):
 
@@ -31,7 +32,10 @@ class ZKUtil(object):
 
     """Abstracts all Kafka-specific ZooKeeper access."""
     def __init__(self, zk_conn_str):
-        self._zk = ZooKeeper(zk_conn_str)
+        try:
+            self._zk = ZooKeeper(zk_conn_str)
+        except FailedConnect as e:
+            raise ZKConnectError(e)
 
     def close(self):
         self._zk.close()