From e76faebc9589654e83c8244ef9aff88391e56b80 Mon Sep 17 00:00:00 2001 From: Vinod Kumar Vavilapalli Date: Sun, 9 Nov 2014 18:56:06 -0800 Subject: [PATCH] YARN-2834. Fixed ResourceManager to ignore token-renewal failures on recovery consistent with the (somewhat incorrect) behaviour in the non-recovery case. Contributed by Jian He. --- hadoop-yarn-project/CHANGES.txt | 4 ++ .../resourcemanager/rmapp/RMAppImpl.java | 27 ++------ .../rmapp/attempt/RMAppAttemptImpl.java | 4 +- .../TestWorkPreservingRMRestart.java | 67 ++++++++++++++++--- .../rmapp/TestRMAppTransitions.java | 28 -------- 5 files changed, 70 insertions(+), 60 deletions(-) diff --git a/hadoop-yarn-project/CHANGES.txt b/hadoop-yarn-project/CHANGES.txt index 8abdb9cb2ae..b76e28f19ad 100644 --- a/hadoop-yarn-project/CHANGES.txt +++ b/hadoop-yarn-project/CHANGES.txt @@ -925,6 +925,10 @@ Release 2.6.0 - 2014-11-15 YARN-2830. Add backwords compatible ContainerId.newInstance constructor. (jeagles via acmurthy) + YARN-2834. Fixed ResourceManager to ignore token-renewal failures on recovery + consistent with the (somewhat incorrect) behaviour in the non-recovery case. + (Jian He via vinodkv) + Release 2.5.2 - UNRELEASED INCOMPATIBLE CHANGES diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/RMAppImpl.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/RMAppImpl.java index 9b10872641d..ad92cc463a7 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/RMAppImpl.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/RMAppImpl.java @@ -714,7 +714,7 @@ public void handle(RMAppEvent event) { } @Override - public void recover(RMState state) throws Exception{ + public void recover(RMState state) { ApplicationState appState = state.getApplicationState().get(getApplicationId()); this.recoveredFinalState = appState.getState(); LOG.info("Recovering app: " + getApplicationId() + " with " + @@ -830,14 +830,7 @@ private static final class RMAppRecoveredTransition implements public RMAppState transition(RMAppImpl app, RMAppEvent event) { RMAppRecoverEvent recoverEvent = (RMAppRecoverEvent) event; - try { - app.recover(recoverEvent.getRMState()); - } catch (Exception e) { - String msg = app.applicationId + " failed to recover. " + e.getMessage(); - failToRecoverApp(app, event, msg, e); - return RMAppState.FINAL_SAVING; - } - + app.recover(recoverEvent.getRMState()); // The app has completed. if (app.recoveredFinalState != null) { app.recoverAppAttempts(); @@ -852,10 +845,10 @@ public RMAppState transition(RMAppImpl app, RMAppEvent event) { app.getApplicationId(), app.parseCredentials(), app.submissionContext.getCancelTokensWhenComplete(), app.getUser()); } catch (Exception e) { - String msg = "Failed to renew delegation token on recovery for " - + app.applicationId + e.getMessage(); - failToRecoverApp(app, event, msg, e); - return RMAppState.FINAL_SAVING; + String msg = "Failed to renew token for " + app.applicationId + + " on recovery : " + e.getMessage(); + app.diagnostics.append(msg); + LOG.error(msg, e); } } @@ -892,14 +885,6 @@ public RMAppState transition(RMAppImpl app, RMAppEvent event) { // Thus we return ACCECPTED state on recovery. return RMAppState.ACCEPTED; } - - private void failToRecoverApp(RMAppImpl app, RMAppEvent event, String msg, - Exception e) { - app.diagnostics.append(msg); - LOG.error(msg, e); - app.rememberTargetTransitionsAndStoreState(event, new FinalTransition( - RMAppState.FAILED), RMAppState.FAILED, RMAppState.FAILED); - } } private static final class AddApplicationToSchedulerTransition extends diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/attempt/RMAppAttemptImpl.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/attempt/RMAppAttemptImpl.java index d3fe1510175..0d7e33451fd 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/attempt/RMAppAttemptImpl.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/attempt/RMAppAttemptImpl.java @@ -789,7 +789,7 @@ public ApplicationResourceUsageReport getApplicationResourceUsageReport() { } @Override - public void recover(RMState state) throws Exception { + public void recover(RMState state) { ApplicationState appState = state.getApplicationState().get(getAppAttemptId().getApplicationId()); ApplicationAttemptState attemptState = @@ -823,7 +823,7 @@ public void transferStateFromPreviousAttempt(RMAppAttempt attempt) { } private void recoverAppAttemptCredentials(Credentials appAttemptTokens, - RMAppAttemptState state) throws IOException { + RMAppAttemptState state) { if (appAttemptTokens == null || state == RMAppAttemptState.FAILED || state == RMAppAttemptState.FINISHED || state == RMAppAttemptState.KILLED) { diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestWorkPreservingRMRestart.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestWorkPreservingRMRestart.java index 2f0a839e9e9..1cefcf851fb 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestWorkPreservingRMRestart.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestWorkPreservingRMRestart.java @@ -18,15 +18,15 @@ package org.apache.hadoop.yarn.server.resourcemanager; -import java.io.File; -import java.io.FileWriter; -import java.io.IOException; -import java.io.PrintWriter; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertNull; import static org.junit.Assert.assertTrue; +import java.io.File; +import java.io.FileWriter; +import java.io.IOException; +import java.io.PrintWriter; import java.net.UnknownHostException; import java.util.ArrayList; import java.util.Arrays; @@ -35,9 +35,10 @@ import java.util.Map; import java.util.Set; +import org.apache.hadoop.fs.CommonConfigurationKeysPublic; import org.apache.hadoop.metrics2.lib.DefaultMetricsSystem; +import org.apache.hadoop.security.Credentials; import org.apache.hadoop.security.UserGroupInformation; -import org.apache.hadoop.service.Service; import org.apache.hadoop.test.GenericTestUtils; import org.apache.hadoop.yarn.api.protocolrecords.AllocateResponse; import org.apache.hadoop.yarn.api.records.ApplicationAttemptId; @@ -50,9 +51,7 @@ import org.apache.hadoop.yarn.api.records.ResourceRequest; import org.apache.hadoop.yarn.conf.YarnConfiguration; import org.apache.hadoop.yarn.server.api.protocolrecords.NMContainerStatus; -import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FSParentQueue; -import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FairSchedulerConfiguration; -import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.policies.DominantResourceFairnessPolicy; +import org.apache.hadoop.yarn.server.resourcemanager.TestRMRestart.TestSecurityMockRM; import org.apache.hadoop.yarn.server.resourcemanager.recovery.MemoryRMStateStore; import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp; import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppState; @@ -71,9 +70,13 @@ import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacitySchedulerConfiguration; import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.LeafQueue; import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.ParentQueue; -import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FairScheduler; import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FSAppAttempt; +import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FSParentQueue; +import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FairScheduler; +import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.FairSchedulerConfiguration; +import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fair.policies.DominantResourceFairnessPolicy; import org.apache.hadoop.yarn.server.resourcemanager.scheduler.fifo.FifoScheduler; +import org.apache.hadoop.yarn.server.resourcemanager.security.DelegationTokenRenewer; import org.apache.hadoop.yarn.util.ControlledClock; import org.apache.hadoop.yarn.util.SystemClock; import org.apache.hadoop.yarn.util.resource.DominantResourceCalculator; @@ -1011,4 +1014,50 @@ public void testRetriedFinishApplicationMasterRequest() am0.unregisterAppAttempt(false); } + @Test (timeout = 30000) + public void testAppFailedToRenewTokenOnRecovery() throws Exception { + conf.set(CommonConfigurationKeysPublic.HADOOP_SECURITY_AUTHENTICATION, + "kerberos"); + conf.setInt(YarnConfiguration.RM_AM_MAX_ATTEMPTS, 1); + UserGroupInformation.setConfiguration(conf); + MemoryRMStateStore memStore = new MemoryRMStateStore(); + memStore.init(conf); + MockRM rm1 = new TestSecurityMockRM(conf, memStore); + rm1.start(); + MockNM nm1 = + new MockNM("127.0.0.1:1234", 8192, rm1.getResourceTrackerService()); + nm1.registerNode(); + RMApp app1 = rm1.submitApp(200); + MockAM am1 = MockRM.launchAndRegisterAM(app1, rm1, nm1); + + MockRM rm2 = new TestSecurityMockRM(conf, memStore) { + protected DelegationTokenRenewer createDelegationTokenRenewer() { + return new DelegationTokenRenewer() { + @Override + public void addApplicationSync(ApplicationId applicationId, + Credentials ts, boolean shouldCancelAtEnd, String user) + throws IOException { + throw new IOException("Token renew failed !!"); + } + }; + } + }; + nm1.setResourceTrackerService(rm2.getResourceTrackerService()); + rm2.start(); + NMContainerStatus containerStatus = + TestRMRestart.createNMContainerStatus(am1.getApplicationAttemptId(), 1, + ContainerState.RUNNING); + nm1.registerNode(Arrays.asList(containerStatus), null); + + // am re-register + rm2.waitForState(app1.getApplicationId(), RMAppState.ACCEPTED); + am1.setAMRMProtocol(rm2.getApplicationMasterService(), rm2.getRMContext()); + am1.registerAppAttempt(true); + rm2.waitForState(app1.getApplicationId(), RMAppState.RUNNING); + + // Because the token expired, am could crash. + nm1.nodeHeartbeat(am1.getApplicationAttemptId(), 1, ContainerState.COMPLETE); + rm2.waitForState(am1.getApplicationAttemptId(), RMAppAttemptState.FAILED); + rm2.waitForState(app1.getApplicationId(), RMAppState.FAILED); + } } diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/TestRMAppTransitions.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/TestRMAppTransitions.java index ecb6b5caff6..bbfb0ee05b0 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/TestRMAppTransitions.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/rmapp/TestRMAppTransitions.java @@ -539,34 +539,6 @@ public void testAppRecoverPath() throws IOException { testCreateAppSubmittedRecovery(sub); } - @Test (timeout = 30000) - public void testAppRecoverToFailed() throws IOException { - LOG.info("--- START: testAppRecoverToFailed ---"); - ApplicationSubmissionContext sub = - Records.newRecord(ApplicationSubmissionContext.class); - ContainerLaunchContext clc = - Records.newRecord(ContainerLaunchContext.class); - Credentials credentials = new Credentials(); - DataOutputBuffer dob = new DataOutputBuffer(); - credentials.writeTokenStorageToStream(dob); - ByteBuffer securityTokens = - ByteBuffer.wrap(dob.getData(), 0, dob.getLength()); - clc.setTokens(securityTokens); - sub.setAMContainerSpec(clc); - - RMApp application = createNewTestApp(sub); - // NEW => FINAL_SAVING, event RMAppEventType.RECOVER - RMState state = new RMState(); - RMAppEvent event = - new RMAppRecoverEvent(application.getApplicationId(), state); - // NPE will throw on recovery. - application.handle(event); - assertAppState(RMAppState.FINAL_SAVING, application); - sendAppUpdateSavedEvent(application); - rmDispatcher.await(); - assertAppState(RMAppState.FAILED, application); - } - @Test (timeout = 30000) public void testAppNewKill() throws IOException { LOG.info("--- START: testAppNewKill ---");