@Override protected AssignedRequests createAssignedRequests() { AssignedRequests assignedReqs = mock(AssignedRequests.class); TaskAttemptId taskAttempt = mock(TaskAttemptId.class); when(assignedReqs.get(any(ContainerId.class))).thenReturn(taskAttempt); return assignedReqs; } }
@SuppressWarnings("unchecked") @VisibleForTesting void processFinishedContainer(ContainerStatus container) { LOG.info("Received completed container " + container.getContainerId()); TaskAttemptId attemptID = assignedRequests.get(container.getContainerId()); if (attemptID == null) { LOG.error("Container complete event for unknown container " + container.getContainerId()); } else { pendingRelease.remove(container.getContainerId()); assignedRequests.remove(attemptID); // Send the diagnostics String diagnostic = StringInterner.weakIntern(container.getDiagnostics()); eventHandler.handle(new TaskAttemptDiagnosticsUpdateEvent(attemptID, diagnostic)); // send the container completed event to Task attempt eventHandler.handle(createContainerFinishedEvent(container, attemptID)); } }
@SuppressWarnings("unchecked") @VisibleForTesting void processFinishedContainer(ContainerStatus container) { LOG.info("Received completed container " + container.getContainerId()); TaskAttemptId attemptID = assignedRequests.get(container.getContainerId()); if (attemptID == null) { LOG.error("Container complete event for unknown container " + container.getContainerId()); } else { pendingRelease.remove(container.getContainerId()); assignedRequests.remove(attemptID); // Send the diagnostics String diagnostic = StringInterner.weakIntern(container.getDiagnostics()); eventHandler.handle(new TaskAttemptDiagnosticsUpdateEvent(attemptID, diagnostic)); // send the container completed event to Task attempt eventHandler.handle(createContainerFinishedEvent(container, attemptID)); preemptionPolicy.handleCompletedContainer(attemptID); } }
ContainerId containerId = assignedRequests.get(aId); if (containerId != null) { removed = true;
ContainerId containerId = assignedRequests.get(aId); if (containerId != null) { removed = true;
TaskAttemptId attemptID = assignedRequests.get(cont.getContainerId()); if (attemptID == null) { LOG.error("Container complete event for unknown container id "
ContainerId containerId = assignedRequests.get(aId); if (containerId != null) { removed = true;
@Override public TaskAttemptId getTaskAttempt(ContainerId container) { return reqs.get(container); }