Fixed incorrect cleanup in Task Map
git-svn-id: https://hyracks.googlecode.com/svn/branches/hyracks_dev_next@914 123451ca-8445-de46-9d55-352943316053
diff --git a/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/Joblet.java b/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/Joblet.java
index 1a98ebd..07d8ad7 100644
--- a/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/Joblet.java
+++ b/hyracks-control-nc/src/main/java/edu/uci/ics/hyracks/control/nc/Joblet.java
@@ -145,7 +145,7 @@
}
public synchronized void notifyTaskComplete(Task task) throws Exception {
- taskMap.remove(task);
+ taskMap.remove(task.getTaskAttemptId());
try {
TaskProfile taskProfile = new TaskProfile(task.getTaskAttemptId(), task.getPartitionSendProfile());
task.dumpProfile(taskProfile);
@@ -159,7 +159,7 @@
}
public synchronized void notifyTaskFailed(Task task, String details) throws Exception {
- taskMap.remove(task);
+ taskMap.remove(task.getTaskAttemptId());
try {
nodeController.getClusterController().notifyTaskFailure(jobId, task.getTaskAttemptId(),
nodeController.getId(), details);