Java ReservationSubmissionRequest.getReservationId方法代码示例

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Test
public void testUpdateReservation() {
  ResourceManager rm = setupResourceManager();
  ClientRMService clientService = rm.getClientRMService();
  Clock clock = new UTCClock();
  long arrival = clock.getTime();
  long duration = 60000;
  long deadline = (long) (arrival + 1.05 * duration);
  ReservationSubmissionRequest sRequest =
      submitReservationTestHelper(clientService, arrival, deadline, duration);

  ReservationDefinition rDef = sRequest.getReservationDefinition();
  ReservationRequest rr =
      rDef.getReservationRequests().getReservationResources().get(0);
  ReservationId reservationID = sRequest.getReservationId();
  rr.setNumContainers(5);
  arrival = clock.getTime();
  duration = 30000;
  deadline = (long) (arrival + 1.05 * duration);
  rr.setDuration(duration);
  rDef.setArrival(arrival);
  rDef.setDeadline(deadline);
  ReservationUpdateRequest uRequest =
      ReservationUpdateRequest.newInstance(rDef, reservationID);
  ReservationUpdateResponse uResponse = null;
  try {
    uResponse = clientService.updateReservation(uRequest);
  } catch (Exception e) {
    Assert.fail(e.getMessage());
  }
  Assert.assertNotNull(uResponse);
  System.out.println("Update reservation response: " + uResponse);

  rm.stop();
}
 

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Test
public void testListReservationsByReservationId() {
  ResourceManager rm = setupResourceManager();
  ClientRMService clientService = rm.getClientRMService();
  Clock clock = new UTCClock();
  long arrival = clock.getTime();
  long duration = 60000;
  long deadline = (long) (arrival + 1.05 * duration);
  ReservationSubmissionRequest sRequest =
      submitReservationTestHelper(clientService, arrival, deadline, duration);

  ReservationId reservationID = sRequest.getReservationId();
  ReservationListResponse response = null;
  ReservationListRequest request = ReservationListRequest.newInstance(
      ReservationSystemTestUtil.reservationQ, reservationID.toString(), -1,
      -1, false);
  try {
    response = clientService.listReservations(request);
  } catch (Exception e) {
    Assert.fail(e.getMessage());
  }
  Assert.assertNotNull(response);
  Assert.assertEquals(1, response.getReservationAllocationState().size());
  Assert.assertEquals(response.getReservationAllocationState().get(0)
      .getReservationId().getId(), reservationID.getId());
  Assert.assertEquals(response.getReservationAllocationState().get(0)
      .getResourceAllocationRequests().size(), 0);

  rm.stop();
}
 

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Test
public void testUpdateReservation() throws Exception {
  MiniYARNCluster cluster = setupMiniYARNCluster();
  YarnClient client = setupYarnClient(cluster);
  try {
    Clock clock = new UTCClock();
    long arrival = clock.getTime();
    long duration = 60000;
    long deadline = (long) (arrival + 1.05 * duration);
    ReservationSubmissionRequest sRequest =
        submitReservationTestHelper(client, arrival, deadline, duration);

    ReservationDefinition rDef = sRequest.getReservationDefinition();
    ReservationRequest rr =
        rDef.getReservationRequests().getReservationResources().get(0);
    ReservationId reservationID = sRequest.getReservationId();
    rr.setNumContainers(5);
    arrival = clock.getTime();
    duration = 30000;
    deadline = (long) (arrival + 1.05 * duration);
    rr.setDuration(duration);
    rDef.setArrival(arrival);
    rDef.setDeadline(deadline);
    ReservationUpdateRequest uRequest =
        ReservationUpdateRequest.newInstance(rDef, reservationID);
    ReservationUpdateResponse uResponse = client.updateReservation(uRequest);
    Assert.assertNotNull(uResponse);
    System.out.println("Update reservation response: " + uResponse);
  } finally {
    // clean-up
    if (client != null) {
      client.stop();
    }
    cluster.stop();
  }
}
 

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Test
public void testListReservationsByReservationId() throws Exception{
  MiniYARNCluster cluster = setupMiniYARNCluster();
  YarnClient client = setupYarnClient(cluster);
  try {
    Clock clock = new UTCClock();
    long arrival = clock.getTime();
    long duration = 60000;
    long deadline = (long) (arrival + 1.05 * duration);
    ReservationSubmissionRequest sRequest =
        submitReservationTestHelper(client, arrival, deadline, duration);

    ReservationId reservationID = sRequest.getReservationId();
    ReservationListRequest request = ReservationListRequest.newInstance(
        ReservationSystemTestUtil.reservationQ, reservationID.toString(), -1,
        -1, false);
    ReservationListResponse response = client.listReservations(request);
    Assert.assertNotNull(response);
    Assert.assertEquals(1, response.getReservationAllocationState().size());
    Assert.assertEquals(response.getReservationAllocationState().get(0)
        .getReservationId().getId(), reservationID.getId());
    Assert.assertEquals(response.getReservationAllocationState().get(0)
        .getResourceAllocationRequests().size(), 0);
  } finally {
    // clean-up
    if (client != null) {
      client.stop();
    }
    cluster.stop();
  }
}
 

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Test
public void testReservationDelete() throws Exception {
  MiniYARNCluster cluster = setupMiniYARNCluster();
  YarnClient client = setupYarnClient(cluster);
  try {
    Clock clock = new UTCClock();
    long arrival = clock.getTime();
    long duration = 60000;
    long deadline = (long) (arrival + 1.05 * duration);
    ReservationSubmissionRequest sRequest =
        submitReservationTestHelper(client, arrival, deadline, duration);

    ReservationId reservationID = sRequest.getReservationId();
    // Delete the reservation
    ReservationDeleteRequest dRequest =
        ReservationDeleteRequest.newInstance(reservationID);
    ReservationDeleteResponse dResponse = client.deleteReservation(dRequest);
    Assert.assertNotNull(dResponse);
    System.out.println("Delete reservation response: " + dResponse);

    // List reservations, search by non-existent reservationID
    ReservationListRequest request = ReservationListRequest.newInstance(
        ReservationSystemTestUtil.reservationQ, reservationID.toString(), -1,
        -1, false);

    ReservationListResponse response =  client.listReservations(request);
    Assert.assertNotNull(response);
    Assert.assertEquals(0, response.getReservationAllocationState().size());
  } finally {
    // clean-up
    if (client != null) {
      client.stop();
    }
    cluster.stop();
  }
}
 

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Override
public ReservationSubmissionResponse submitReservation(
    ReservationSubmissionRequest request) throws YarnException, IOException {
  // Check if reservation system is enabled
  checkReservationSytem(AuditConstants.SUBMIT_RESERVATION_REQUEST);
  ReservationSubmissionResponse response =
      recordFactory.newRecordInstance(ReservationSubmissionResponse.class);
  ReservationId reservationId = request.getReservationId();
  // Validate the input
  Plan plan =
      rValidator.validateReservationSubmissionRequest(reservationSystem,
          request, reservationId);

  ReservationAllocation allocation = plan.getReservationById(reservationId);

  if (allocation != null) {
    boolean isNewDefinition = !allocation.getReservationDefinition().equals(
        request.getReservationDefinition());
    if (isNewDefinition) {
      String message = "Reservation allocation already exists with the " +
          "reservation id " + reservationId.toString() + ", but a different" +
          " reservation definition was provided. Please try again with a " +
          "new reservation id, or consider updating the reservation instead.";
      throw RPCUtil.getRemoteException(message);
    } else {
      return response;
    }
  }

  // Check ACLs
  String queueName = request.getQueue();
  String user =
      checkReservationACLs(queueName,
          AuditConstants.SUBMIT_RESERVATION_REQUEST, null);
  try {
    // Try to place the reservation using the agent
    boolean result =
        plan.getReservationAgent().createReservation(reservationId, user,
            plan, request.getReservationDefinition());
    if (result) {
      // add the reservation id to valid ones maintained by reservation
      // system
      reservationSystem.setQueueForReservation(reservationId, queueName);
      // create the reservation synchronously if required
      refreshScheduler(queueName, request.getReservationDefinition(),
          reservationId.toString());
      // return the reservation id
    }
  } catch (PlanningException e) {
    RMAuditLogger.logFailure(user, AuditConstants.SUBMIT_RESERVATION_REQUEST,
        e.getMessage(), "ClientRMService",
        "Unable to create the reservation: " + reservationId);
    throw RPCUtil.getRemoteException(e);
  }
  RMAuditLogger.logSuccess(user, AuditConstants.SUBMIT_RESERVATION_REQUEST,
      "ClientRMService: " + reservationId);
  return response;
}
 

import org.apache.hadoop.yarn.api.protocolrecords.ReservationSubmissionRequest; //导入方法依赖的package包/类
@Test
public void testListReservationsByTimeInterval() throws Exception {
  MiniYARNCluster cluster = setupMiniYARNCluster();
  YarnClient client = setupYarnClient(cluster);
  try {
    Clock clock = new UTCClock();
    long arrival = clock.getTime();
    long duration = 60000;
    long deadline = (long) (arrival + 1.05 * duration);
    ReservationSubmissionRequest sRequest =
        submitReservationTestHelper(client, arrival, deadline, duration);

    // List reservations, search by a point in time within the reservation
    // range.
    arrival = clock.getTime();
    ReservationId reservationID = sRequest.getReservationId();
    ReservationListRequest request = ReservationListRequest.newInstance(
        ReservationSystemTestUtil.reservationQ, "", arrival + duration / 2,
        arrival + duration / 2, true);

    ReservationListResponse response = client.listReservations(request);
    Assert.assertNotNull(response);
    Assert.assertEquals(1, response.getReservationAllocationState().size());
    Assert.assertEquals(response.getReservationAllocationState().get(0)
        .getReservationId().getId(), reservationID.getId());
    // List reservations, search by time within reservation interval.
    request = ReservationListRequest.newInstance(
        ReservationSystemTestUtil.reservationQ, "", 1, Long.MAX_VALUE, true);

    response = client.listReservations(request);
    Assert.assertNotNull(response);
    Assert.assertEquals(1, response.getReservationAllocationState().size());
    Assert.assertEquals(response.getReservationAllocationState().get(0)
        .getReservationId().getId(), reservationID.getId());
    // Verify that the full resource allocations exist.
    Assert.assertTrue(response.getReservationAllocationState().get(0)
        .getResourceAllocationRequests().size() > 0);

    // Verify that the full RDL is returned.
    ReservationRequests reservationRequests =
        response.getReservationAllocationState().get(0)
            .getReservationDefinition().getReservationRequests();
    Assert.assertTrue(
        reservationRequests.getInterpreter().toString().equals("R_ALL"));
    Assert.assertTrue(reservationRequests.getReservationResources().get(0)
        .getDuration() == duration);
  } finally {
    // clean-up
    if (client != null) {
      client.stop();
    }
    cluster.stop();
  }
}