commit c583202796c456edd43d0e709497af61e93df84a
author Madhusudan.C.S <madhusudancs@gmail.com> Wed Jun 05 16:52:45 2013 -0700
committer Madhusudan.C.S <madhusudancs@gmail.com> Wed Jun 05 17:45:27 2013 -0700
tree 9d788e2548970fca899cf51f31bd45962fed12aa
parent 01ded08db557c5a15501f428e0be4312e2a7d68d

Do not remove the hash map entries while iterating.

hyracks/hyracks-control/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/dataset/DatasetPartitionManager.java

diff --git a/hyracks/hyracks-control/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/dataset/DatasetPartitionManager.java b/hyracks/hyracks-control/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/dataset/DatasetPartitionManager.java
index 1299593..32b310b 100644
--- a/hyracks/hyracks-control/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/dataset/DatasetPartitionManager.java
+++ b/hyracks/hyracks-control/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/dataset/DatasetPartitionManager.java
@@ -205,7 +205,7 @@
     @Override
     public synchronized void close() {
         for (Entry<JobId, IDatasetStateRecord> entry : partitionResultStateMap.entrySet()) {
-            deinitState(entry.getKey());
+            deinit(entry.getKey());
         }
         deallocatableRegistry.close();
     }
@@ -217,7 +217,12 @@
 
     @Override
     public void deinitState(JobId jobId) {
-        ResultSetMap rsIdMap = (ResultSetMap) partitionResultStateMap.remove(jobId);
+        deinit(jobId);
+        partitionResultStateMap.remove(jobId);
+    }
+
+    private void deinit(JobId jobId) {
+        ResultSetMap rsIdMap = (ResultSetMap) partitionResultStateMap.get(jobId);
         if (rsIdMap != null) {
             for (ResultSetId rsId : rsIdMap.keySet()) {
                 ResultState[] resultStates = rsIdMap.get(rsId);