Examples with AMNodeEventContainerAllocated org.apache.tez.dag.app.rm.node.AMNodeEventContainerAllocated used on opensource projects

Example 1 with AMNodeEventContainerAllocated

use of org.apache.tez.dag.app.rm.node.AMNodeEventContainerAllocated in project tez by apache.

the class TaskSchedulerManager method taskAllocated.

// TaskSchedulerAppCallback methods with schedulerId, where relevant
public synchronized void taskAllocated(int schedulerId, Object task, Object appCookie, Container container) {
    AMSchedulerEventTALaunchRequest event = (AMSchedulerEventTALaunchRequest) appCookie;
    ContainerId containerId = container.getId();
    if (appContext.getAllContainers().addContainerIfNew(container, schedulerId, event.getLauncherId(), event.getTaskCommId())) {
        appContext.getNodeTracker().nodeSeen(container.getNodeId(), schedulerId);
        sendEvent(new AMNodeEventContainerAllocated(container.getNodeId(), schedulerId, container.getId()));
    }
    TaskAttempt taskAttempt = event.getTaskAttempt();
    // taskAllocated() upcall
    assert task.equals(taskAttempt);
    if (appContext.getAllContainers().get(containerId).getState() == AMContainerState.ALLOCATED) {
        sendEvent(new AMContainerEventLaunchRequest(containerId, taskAttempt.getVertexID(), event.getContainerContext(), event.getLauncherId(), event.getTaskCommId()));
    }
    sendEvent(new AMContainerEventAssignTA(containerId, taskAttempt.getID(), event.getRemoteTaskSpec(), event.getContainerContext().getLocalResources(), event.getContainerContext().getCredentials(), event.getPriority()));
}