protected void setRequestLimit(Priority priority, Resource capability, int limit) { if (limit < 0) { limit = Integer.MAX_VALUE; } ResourceRequest newReqLimit = ResourceRequest.newInstance(priority, ResourceRequest.ANY, capability, limit); ResourceRequest oldReqLimit = requestLimits.put(newReqLimit, newReqLimit); if (oldReqLimit == null || oldReqLimit.getNumContainers() < limit) { requestLimitsToUpdate.add(newReqLimit); } }
protected void setRequestLimit(Priority priority, Resource capability, int limit) { if (limit < 0) { limit = Integer.MAX_VALUE; } ResourceRequest newReqLimit = ResourceRequest.newInstance(priority, ResourceRequest.ANY, capability, limit); ResourceRequest oldReqLimit = requestLimits.put(newReqLimit, newReqLimit); if (oldReqLimit == null || oldReqLimit.getNumContainers() < limit) { requestLimitsToUpdate.add(newReqLimit); } }
protected void setRequestLimit(Priority priority, Resource capability, int limit) { if (limit < 0) { limit = Integer.MAX_VALUE; } ResourceRequest newReqLimit = ResourceRequest.newInstance(priority, ResourceRequest.ANY, capability, limit); ResourceRequest oldReqLimit = requestLimits.put(newReqLimit, newReqLimit); if (oldReqLimit == null || oldReqLimit.getNumContainers() < limit) { requestLimitsToUpdate.add(newReqLimit); } }
@Public @Stable public static ResourceRequest newInstance(Priority priority, String hostName, Resource capability, int numContainers) { return newInstance(priority, hostName, capability, numContainers, true); }
@Public @Stable public static ResourceRequest newInstance(Priority priority, String hostName, Resource capability, int numContainers, boolean relaxLocality) { return newInstance(priority, hostName, capability, numContainers, relaxLocality, null); }
@Public @Stable public static ResourceRequest newInstance(Priority priority, String hostName, Resource capability, int numContainers) { return newInstance(priority, hostName, capability, numContainers, true); }
@Public @Stable public static ResourceRequest newInstance(Priority priority, String hostName, Resource capability, int numContainers, boolean relaxLocality) { return newInstance(priority, hostName, capability, numContainers, relaxLocality, null); }
public static SchedulerRequestKey toSchedulerKey(Priority pri) { return SchedulerRequestKey.create( ResourceRequest.newInstance(pri, null, null, 0)); }
private List<ResourceRequest> cloneAsks() { List<ResourceRequest> askList = new ArrayList<ResourceRequest>(ask.size()); for(ResourceRequest r : ask) { // create a copy of ResourceRequest as we might change it while the // RPC layer is using it to send info across ResourceRequest rr = ResourceRequest.newInstance(r.getPriority(), r.getResourceName(), r.getCapability(), r.getNumContainers(), r.getRelaxLocality(), r.getNodeLabelExpression()); askList.add(rr); } return askList; }
public ResourceRequest cloneResourceRequest(ResourceRequest request) { ResourceRequest newRequest = ResourceRequest.newInstance( request.getPriority(), request.getResourceName(), request.getCapability(), 1, request.getRelaxLocality()); return newRequest; } }
public ResourceRequest cloneResourceRequest(ResourceRequest request) { ResourceRequest newRequest = ResourceRequest.newInstance( request.getPriority(), request.getResourceName(), request.getCapability(), 1, request.getRelaxLocality()); return newRequest; } }
public MockRMApp(int newid, long time, RMAppState newState) { finish = time; id = MockApps.newAppID(newid); state = newState; amReqs = Collections.singletonList(ResourceRequest.newInstance( Priority.UNDEFINED, "0.0.0.0", Resource.newInstance(0, 0), 1)); }
private ResourceRequest createResourceRequest(String resource, boolean relaxLocality, String nodeLabel) { return ResourceRequest.newInstance(Priority.newInstance(0), resource, Resource.newInstance(1, 1), 1, relaxLocality, nodeLabel); } }
public static SchedulerRequestKey toSchedulerKey(Priority pri, long allocationRequestId) { ResourceRequest req = ResourceRequest.newInstance(pri, null, null, 0); req.setAllocationRequestId(allocationRequestId); return SchedulerRequestKey.create(req); }
public static SchedulerRequestKey toSchedulerKey(int pri) { return SchedulerRequestKey.create(ResourceRequest.newInstance( Priority.newInstance(pri), null, null, 0)); }
public class MockRMNullStateStore extends NullRMStateStore { @SuppressWarnings("rawtypes") @Override protected EventHandler getRMStateStoreEventHandler() { return rmStateStoreEventHandler; } }
public RMApp submitApp(int masterMemory, Set<String> appTags) throws Exception { Resource resource = Resource.newInstance(masterMemory, 0); ResourceRequest amResourceRequest = ResourceRequest.newInstance( Priority.newInstance(0), ResourceRequest.ANY, resource, 1); return submitApp(Collections.singletonList(amResourceRequest), "", UserGroupInformation.getCurrentUser().getShortUserName(), null, false, null, super.getConfig().getInt(YarnConfiguration.RM_AM_MAX_ATTEMPTS, YarnConfiguration.DEFAULT_RM_AM_MAX_ATTEMPTS), null, null, true, false, false, null, 0, null, true, Priority.newInstance(0), null, null, null, appTags); }
private static ResourceRequest newResourceRequest(int priority, long allocReqId, String rName, Resource resource, int numContainers, boolean relaxLoc, ExecutionType eType) { ResourceRequest rr = ResourceRequest.newInstance( Priority.newInstance(priority), rName, resource, numContainers, relaxLoc, null, ExecutionTypeRequest.newInstance(eType, true)); rr.setAllocationRequestId(allocReqId); return rr; }
@Test public void testEqualsOnExecutionTypeRequest() { ResourceRequest resourceRequestA = ResourceRequest.newInstance(Priority.newInstance(0), "localhost", Resource.newInstance(1024, 1), 1, false, "", ExecutionTypeRequest.newInstance(ExecutionType.GUARANTEED, true)); ResourceRequest resourceRequestB = ResourceRequest.newInstance(Priority.newInstance(0), "localhost", Resource.newInstance(1024, 1), 1, false, "", ExecutionTypeRequest.newInstance(ExecutionType.GUARANTEED, false)); Assert.assertFalse(resourceRequestA.equals(resourceRequestB)); } }
@Test public void testRMAppSubmitResource() throws Exception { asContext.setResource(Resources.createResource(1024)); asContext.setAMContainerResourceRequests(null); RMApp app = testRMAppSubmit(); // setResource Assert.assertEquals(Collections.singletonList( ResourceRequest.newInstance(RMAppAttemptImpl.AM_CONTAINER_PRIORITY, ResourceRequest.ANY, Resources.createResource(1024), 1, true, "")), app.getAMResourceRequests()); }