use countdownlatch to wait for zookeeper connection

jcorwin · Jul 1, 2010 · 3508d52 · 3508d52
1 parent ecec17c
commit 3508d52
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/main/scala/com/twitter/zookeeper/ZooKeeperClient.scala b/src/main/scala/com/twitter/zookeeper/ZooKeeperClient.scala
@@ -24,18 +24,14 @@ object ZKWatch {
 class ZooKeeperClient(servers: String, sessionTimeout: Int, basePath : String, watcher: Option[ZKWatch]) extends Watcher {
   private val log = Logger.get
   private val connectionLatch = new CountDownLatch(1)
-  private var connected = false
   private val zk = new ZooKeeper(servers, sessionTimeout, this)
   connectionLatch.await()
 
   def process(event : WatchedEvent) {
     log.info("Zookeeper event: %s".format(event))
     event.getState match {
       case KeeperState.SyncConnected => {
-        if (!connected) {
-          connected = true
-          connectionLatch.countDown()
-        }
+        connectionLatch.countDown()
       }
       case _ =>
     }