Examples of org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest

org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest

The core request sent by the ApplicationMaster to the ResourceManager to obtain resources in the cluster.

The request includes:

A response id to track duplicate responses.
Progress information.
A list of {@link ResourceRequest} to inform the ResourceManager about the application's resource requirements.
A list of unused {@link Container} which are being returned.

@see ApplicationMasterProtocol#allocate(AllocateRequest)

  }


  public AllocateResponse allocate(
      List<ResourceRequest> resourceRequest, List<ContainerId> releases)
      throws Exception {
    final AllocateRequest req =
        AllocateRequest.newInstance(++responseId, 0F, resourceRequest,
          releases, null);
    UserGroupInformation ugi =
        UserGroupInformation.createRemoteUser(attemptId.toString());
    Token<AMRMTokenIdentifier> token =

    // Request a container allocation.
    List<ResourceRequest> ask = new ArrayList<ResourceRequest>();
    ask.add(BuilderUtils.newResourceRequest(BuilderUtils.newPriority(0), "*",
        BuilderUtils.newResource(1024), 1));


    AllocateRequest allocateRequest = BuilderUtils.newAllocateRequest(
        BuilderUtils.newApplicationAttemptId(appID, 1), 0, 0F, ask,
        new ArrayList<ContainerId>());
    List<Container> allocatedContainers = scheduler.allocate(allocateRequest)
        .getAMResponse().getAllocatedContainers();


    // Modify ask to request no more.
    allocateRequest.clearAsks();


    int waitCounter = 0;
    while ((allocatedContainers == null || allocatedContainers.size() == 0)
        && waitCounter++ != 20) {
      LOG.info("Waiting for container to be allocated..");
      Thread.sleep(1000);
      allocateRequest.setResponseId(allocateRequest.getResponseId() + 1);
      allocatedContainers = scheduler.allocate(allocateRequest)
          .getAMResponse().getAllocatedContainers();
    }


    Assert.assertNotNull("Container is not allocted!", allocatedContainers);

Examples of org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest

Related Classes of org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest