From a4102d516be25bb80e572e79ea19da2c6343419f Mon Sep 17 00:00:00 2001
From: Peter Veentjer <alarmnmmer@gmail.com>
Date: Thu, 30 Jun 2011 21:32:32 +0300
Subject: [PATCH] more work on the roundrobin tests

---
 .../scala/akka/cluster/zookeeper/ZooKeeperBarrier.scala   | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/akka-cluster/src/main/scala/akka/cluster/zookeeper/ZooKeeperBarrier.scala b/akka-cluster/src/main/scala/akka/cluster/zookeeper/ZooKeeperBarrier.scala
index e6e44c6520..3128d525b0 100644
--- a/akka-cluster/src/main/scala/akka/cluster/zookeeper/ZooKeeperBarrier.scala
+++ b/akka-cluster/src/main/scala/akka/cluster/zookeeper/ZooKeeperBarrier.scala
@@ -65,6 +65,14 @@ class ZooKeeperBarrier(zkClient: ZkClient, name: String, node: String, count: In
     leave()
   }
 
+  /**
+   * An await does a enter/leave making this barrier a 'single' barrier instead of a double barrier.
+   */
+  def await() {
+    enter
+    leave()
+  }
+
   def enter = {
     zkClient.createEphemeral(entry)
     if (zkClient.countChildren(barrier) >= count)