import org.apache.hadoop.yarn.exceptions.InvalidApplicationMasterRequestException; //导入依赖的package包/类
private void throwApplicationDoesNotExistInCacheException(
    ApplicationAttemptId appAttemptId)
    throws InvalidApplicationMasterRequestException {
  String message = "Application doesn't exist in cache "
      + appAttemptId;
  LOG.error(message);
  throw new InvalidApplicationMasterRequestException(message);
}
 

import org.apache.hadoop.yarn.exceptions.InvalidApplicationMasterRequestException; //导入依赖的package包/类
@Test(timeout=1200000)
public void testFinishApplicationMasterBeforeRegistering() throws Exception {
  MockRM rm = new MockRM(conf);
  try {
    rm.start();
    // Register node1
    MockNM nm1 = rm.registerNode("127.0.0.1:1234", 6 * GB);
    // Submit an application
    RMApp app1 = rm.submitApp(2048);
    MockAM am1 = MockRM.launchAM(app1, rm, nm1);
    FinishApplicationMasterRequest req =
        FinishApplicationMasterRequest.newInstance(
            FinalApplicationStatus.FAILED, "", "");
    Throwable cause = null;
    try {
      am1.unregisterAppAttempt(req, false);
    } catch (Exception e) {
      cause = e.getCause();
    }
    Assert.assertNotNull(cause);
    Assert
        .assertTrue(cause instanceof InvalidApplicationMasterRequestException);
    Assert.assertNotNull(cause.getMessage());
    Assert
        .assertTrue(cause
            .getMessage()
            .contains(
                "Application Master is trying to unregister before registering for:"));
  } finally {
    if (rm != null) {
      rm.stop();
    }
  }
}
 

import org.apache.hadoop.yarn.exceptions.InvalidApplicationMasterRequestException; //导入依赖的package包/类
@Test
public void testARRMResponseId() throws Exception {

  MockNM nm1 = rm.registerNode("h1:1234", 5000);

  RMApp app = rm.submitApp(2000);

  // Trigger the scheduling so the AM gets 'launched'
  nm1.nodeHeartbeat(true);

  RMAppAttempt attempt = app.getCurrentAppAttempt();
  MockAM am = rm.sendAMLaunched(attempt.getAppAttemptId());

  am.registerAppAttempt();
  
  AllocateRequest allocateRequest =
      AllocateRequest.newInstance(0, 0F, null, null, null);

  AllocateResponse response =
      allocate(attempt.getAppAttemptId(), allocateRequest);
  Assert.assertEquals(1, response.getResponseId());
  Assert.assertTrue(response.getAMCommand() == null);
  allocateRequest =
      AllocateRequest.newInstance(response.getResponseId(), 0F, null, null,
        null);
  
  response = allocate(attempt.getAppAttemptId(), allocateRequest);
  Assert.assertEquals(2, response.getResponseId());
  /* try resending */
  response = allocate(attempt.getAppAttemptId(), allocateRequest);
  Assert.assertEquals(2, response.getResponseId());
  
  /** try sending old request again **/
  allocateRequest = AllocateRequest.newInstance(0, 0F, null, null, null);

  try {
    allocate(attempt.getAppAttemptId(), allocateRequest);
    Assert.fail();
  } catch (Exception e) {
    Assert.assertTrue(e.getCause() instanceof InvalidApplicationMasterRequestException);
  }
}
 

import org.apache.hadoop.yarn.exceptions.InvalidApplicationMasterRequestException; //导入依赖的package包/类
@Override
public FinishApplicationMasterResponse finishApplicationMaster(
    FinishApplicationMasterRequest request) throws YarnException,
    IOException {

  ApplicationAttemptId applicationAttemptId = authorizeRequest();

  AllocateResponseLock lock = responseMap.get(applicationAttemptId);
  if (lock == null) {
    throwApplicationDoesNotExistInCacheException(applicationAttemptId);
  }

  // Allow only one thread in AM to do finishApp at a time.
  synchronized (lock) {
    if (!hasApplicationMasterRegistered(applicationAttemptId)) {
      String message =
          "Application Master is trying to unregister before registering for: "
              + applicationAttemptId.getApplicationId();
      LOG.error(message);
      RMAuditLogger.logFailure(
          this.rmContext.getRMApps()
              .get(applicationAttemptId.getApplicationId()).getUser(),
          AuditConstants.UNREGISTER_AM, "", "ApplicationMasterService",
          message, applicationAttemptId.getApplicationId(),
          applicationAttemptId);
      throw new InvalidApplicationMasterRequestException(message);
    }
    
    this.amLivelinessMonitor.receivedPing(applicationAttemptId);

    RMApp rmApp =
        rmContext.getRMApps().get(applicationAttemptId.getApplicationId());

    if (rmApp.isAppFinalStateStored()) {
      return FinishApplicationMasterResponse.newInstance(true);
    }

    rmContext.getDispatcher().getEventHandler().handle(
        new RMAppAttemptUnregistrationEvent(applicationAttemptId, request
            .getTrackingUrl(), request.getFinalApplicationStatus(), request
            .getDiagnostics()));

    // For UnmanagedAMs, return true so they don't retry
    return FinishApplicationMasterResponse.newInstance(
        rmApp.getApplicationSubmissionContext().getUnmanagedAM());
  }
}