/** * Set the <code>allocationRequestId</code> of the request. * @see ResourceRequest#setAllocationRequestId(long) * @param allocationRequestId * <code>allocationRequestId</code> of the request * @return {@link ResourceRequestBuilder} */ @Public @Evolving public ResourceRequestBuilder allocationRequestId( long allocationRequestId) { resourceRequest.setAllocationRequestId(allocationRequestId); return this; }
@SuppressWarnings("checkstyle:parameternumber") public static ResourceRequest createResourceRequest(long id, String resource, int memory, int vCores, int priority, int containers, String labelExpression, boolean relaxLocality) throws YarnException { ResourceRequest out = createResourceRequest(resource, memory, vCores, priority, containers, labelExpression, relaxLocality); out.setAllocationRequestId(id); return out; }
public List<ResourceRequest> createReq(String[] hosts, int memory, int priority, int containers, String labelExpression, long allocationRequestId) throws Exception { List<ResourceRequest> reqs = new ArrayList<ResourceRequest>(); if (hosts != null) { for (String host : hosts) { // only add host/rack request when asked host isn't ANY if (!host.equals(ResourceRequest.ANY)) { ResourceRequest hostReq = createResourceReq(host, memory, priority, containers, labelExpression); hostReq.setAllocationRequestId(allocationRequestId); reqs.add(hostReq); ResourceRequest rackReq = createResourceReq("/default-rack", memory, priority, containers, labelExpression); rackReq.setAllocationRequestId(allocationRequestId); reqs.add(rackReq); } } } ResourceRequest offRackReq = createResourceReq(ResourceRequest.ANY, memory, priority, containers, labelExpression); offRackReq.setAllocationRequestId(allocationRequestId); reqs.add(offRackReq); return reqs; }
public static SchedulerRequestKey toSchedulerKey(Priority pri, long allocationRequestId) { ResourceRequest req = ResourceRequest.newInstance(pri, null, null, 0); req.setAllocationRequestId(allocationRequestId); return SchedulerRequestKey.create(req); }
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; }
private static ResourceRequest createResourceReqForIncrease( SchedulerRequestKey schedulerRequestKey, Resource resToIncrease, ResourceRequest rr, RMContainer rmContainer, String resourceName) { rr.setResourceName(resourceName); rr.setNumContainers(1); rr.setRelaxLocality(false); rr.setPriority(rmContainer.getContainer().getPriority()); rr.setAllocationRequestId(schedulerRequestKey.getAllocationRequestId()); rr.setCapability(resToIncrease); rr.setNodeLabelExpression(rmContainer.getNodeLabelExpression()); rr.setExecutionTypeRequest(ExecutionTypeRequest.newInstance( ExecutionType.GUARANTEED, true)); return rr; }
requests.get(0).setAllocationRequestId(0L); requests.get(1).setAllocationRequestId(1L); app.updateResourceRequests(requests);
protected ResourceRequest createResourceRequest(long id, String resource, int memory, int vCores, int priority, ExecutionType execType, int containers) { ResourceRequest req = Records.newRecord(ResourceRequest.class); req.setAllocationRequestId(id); req.setResourceName(resource); req.setCapability(Resource.newInstance(memory, vCores)); req.setPriority(Priority.newInstance(priority)); req.setExecutionTypeRequest(ExecutionTypeRequest.newInstance(execType)); req.setNumContainers(containers); return req; }
public ResourceRequest createResourceRequest(long id, String resource, int memory, int vCores, int priority, ExecutionType execType, int containers) { ResourceRequest req = Records.newRecord(ResourceRequest.class); req.setAllocationRequestId(id); req.setResourceName(resource); req.setCapability(Resource.newInstance(memory, vCores)); req.setPriority(Priority.newInstance(priority)); req.setExecutionTypeRequest(ExecutionTypeRequest.newInstance(execType)); req.setNumContainers(containers); return req; }