YARN-9311. Fix TestRMRestart hangs. Contributed by Prabhu Joseph.

This commit is contained in:
Rohith Sharma K S 2019-02-27 10:28:16 +05:30
parent 625e93713b
commit 8eae260af5

View File

@ -100,6 +100,7 @@
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStore.RMState; import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStore.RMState;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreAMRMTokenEvent; import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreAMRMTokenEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreEvent; import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreProxyCAEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreRMDTEvent; import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreRMDTEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreRMDTMasterKeyEvent; import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreRMDTMasterKeyEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.records.ApplicationAttemptStateData; import org.apache.hadoop.yarn.server.resourcemanager.recovery.records.ApplicationAttemptStateData;
@ -1646,7 +1647,8 @@ protected void handleStoreEvent(RMStateStoreEvent event) {
// Skip if synchronous updation of DTToken // Skip if synchronous updation of DTToken
if (!(event instanceof RMStateStoreAMRMTokenEvent) if (!(event instanceof RMStateStoreAMRMTokenEvent)
&& !(event instanceof RMStateStoreRMDTEvent) && !(event instanceof RMStateStoreRMDTEvent)
&& !(event instanceof RMStateStoreRMDTMasterKeyEvent)) { && !(event instanceof RMStateStoreRMDTMasterKeyEvent)
&& !(event instanceof RMStateStoreProxyCAEvent)) {
while (wait); while (wait);
} }
super.handleStoreEvent(event); super.handleStoreEvent(event);