Remove task from heart beat monitor when container finished (#588)

tony-framework · Aug 17, 2021 · f8daf80 · f8daf80
1 parent 9420681
commit f8daf80
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java b/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java
@@ -1245,6 +1245,7 @@ private void processFinishedContainer(ContainerId containerId, int exitStatus, S
 
       LOG.info("Container " + containerId + " for task " + task + " finished with exitStatus " + exitStatus + ".");
       session.onTaskCompleted(task.getJobName(), task.getTaskIndex(), exitStatus, diagnosticMessage);
+      hbMonitor.unregister(task);
 
       scheduler.registerDependencyCompleted(task.getJobName());
       if (ContainerExitStatus.SUCCESS != exitStatus) {