From ac93c9c1898751263d98eb1a41513135082b96ac Mon Sep 17 00:00:00 2001
From: leventov <leventov@ya.ru>
Date: Sat, 24 Jan 2015 16:01:38 +0700
Subject: [PATCH] Fixed a couple of bugs in AbstractChronicleMap iterations

---
 .../java/net/openhft/chronicle/map/AbstractChronicleMap.java  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/net/openhft/chronicle/map/AbstractChronicleMap.java b/src/main/java/net/openhft/chronicle/map/AbstractChronicleMap.java
index eb38d48db..87c68c5da 100644
--- a/src/main/java/net/openhft/chronicle/map/AbstractChronicleMap.java
+++ b/src/main/java/net/openhft/chronicle/map/AbstractChronicleMap.java
@@ -295,6 +295,7 @@ abstract class AbstractChronicleMap<K, V> extends AbstractMap<K, V>
                     c.updateLock().lock();
                     c.initSegment();
                     c.hashLookup.forEach(this);
+                    return;
                 } finally {
                     context = null;
                 }
@@ -466,7 +467,8 @@ abstract class AbstractChronicleMap<K, V> extends AbstractMap<K, V>
             try {
                 if (!c.containsKey()) // for replicated map
                     return false;
-                shouldBreak = predicate.test(c);
+                shouldBreak = !predicate.test(c);
+                c.closeKey0();
                 // release all exclusive locks: possibly if context.remove() is performed
                 // in the callback, or acquired manually
                 while (c.writeLockCount > 0) {