/// <summary> /// The /// <see cref="ResourceScheduler"/> /// is allocating data-local resources to the /// application. /// </summary> /// <param name="allocatedContainers">resources allocated to the application</param> private void AllocateNodeLocal(SchedulerNode node, Priority priority, ResourceRequest nodeLocalRequest, Container container, IList <ResourceRequest> resourceRequests) { lock (this) { // Update future requirements DecResourceRequest(node.GetNodeName(), priority, nodeLocalRequest); ResourceRequest rackLocalRequest = requests[priority][node.GetRackName()]; DecResourceRequest(node.GetRackName(), priority, rackLocalRequest); ResourceRequest offRackRequest = requests[priority][ResourceRequest.Any]; DecrementOutstanding(offRackRequest); // Update cloned NodeLocal, RackLocal and OffRack requests for recovery resourceRequests.AddItem(CloneResourceRequest(nodeLocalRequest)); resourceRequests.AddItem(CloneResourceRequest(rackLocalRequest)); resourceRequests.AddItem(CloneResourceRequest(offRackRequest)); } }
private SchedulerNode CreateNode() { SchedulerNode node = Org.Mockito.Mockito.Mock <SchedulerNode>(); Org.Mockito.Mockito.When(node.GetNodeName()).ThenReturn("somehost"); Org.Mockito.Mockito.When(node.GetRackName()).ThenReturn("somerack"); Org.Mockito.Mockito.When(node.GetNodeID()).ThenReturn(nodeId); return(node); }
public static bool IsBlacklisted(SchedulerApplicationAttempt application, SchedulerNode node, Log Log) { if (application.IsBlacklisted(node.GetNodeName())) { if (Log.IsDebugEnabled()) { Log.Debug("Skipping 'host' " + node.GetNodeName() + " for " + application.GetApplicationId () + " since it has been blacklisted"); } return(true); } if (application.IsBlacklisted(node.GetRackName())) { if (Log.IsDebugEnabled()) { Log.Debug("Skipping 'rack' " + node.GetRackName() + " for " + application.GetApplicationId () + " since it has been blacklisted"); } return(true); } return(false); }