YARN-5905. Update the RM webapp host that is reported as part of Federation membership to current primary RM's IP.

This commit is contained in:
Subru Krishnan 2016-11-22 18:30:40 -08:00 committed by Carlo Curino
parent 575137f41c
commit e0f0e7afed
2 changed files with 12 additions and 3 deletions

View File

@ -177,8 +177,8 @@ public class FederationStateStoreService extends AbstractService
config.getSocketAddr(YarnConfiguration.RM_ADMIN_ADDRESS, config.getSocketAddr(YarnConfiguration.RM_ADMIN_ADDRESS,
YarnConfiguration.DEFAULT_RM_ADMIN_ADDRESS, YarnConfiguration.DEFAULT_RM_ADMIN_ADDRESS,
YarnConfiguration.DEFAULT_RM_ADMIN_PORT)); YarnConfiguration.DEFAULT_RM_ADMIN_PORT));
String webAppAddress = String webAppAddress = getServiceAddress(NetUtils
WebAppUtils.getResolvedRemoteRMWebAppURLWithoutScheme(config); .createSocketAddr(WebAppUtils.getRMWebAppURLWithScheme(config)));
SubClusterInfo subClusterInfo = SubClusterInfo.newInstance(subClusterId, SubClusterInfo subClusterInfo = SubClusterInfo.newInstance(subClusterId,
amRMAddress, clientRMAddress, rmAdminAddress, webAppAddress, amRMAddress, clientRMAddress, rmAdminAddress, webAppAddress,

View File

@ -19,6 +19,7 @@ package org.apache.hadoop.yarn.server.resourcemanager.federation;
import java.io.IOException; import java.io.IOException;
import java.io.StringReader; import java.io.StringReader;
import java.net.UnknownHostException;
import javax.xml.bind.JAXBException; import javax.xml.bind.JAXBException;
@ -157,12 +158,20 @@ public class TestFederationRMStateStoreService {
} }
private String checkSubClusterInfo(SubClusterState state) private String checkSubClusterInfo(SubClusterState state)
throws YarnException { throws YarnException, UnknownHostException {
Assert.assertNotNull(stateStore.getSubCluster(request)); Assert.assertNotNull(stateStore.getSubCluster(request));
SubClusterInfo response = SubClusterInfo response =
stateStore.getSubCluster(request).getSubClusterInfo(); stateStore.getSubCluster(request).getSubClusterInfo();
Assert.assertEquals(state, response.getState()); Assert.assertEquals(state, response.getState());
Assert.assertTrue(response.getLastHeartBeat() >= lastHearbeatTS); Assert.assertTrue(response.getLastHeartBeat() >= lastHearbeatTS);
String expectedAddress =
(response.getClientRMServiceAddress().split(":"))[0];
Assert.assertEquals(expectedAddress,
(response.getAMRMServiceAddress().split(":"))[0]);
Assert.assertEquals(expectedAddress,
(response.getRMAdminServiceAddress().split(":"))[0]);
Assert.assertEquals(expectedAddress,
(response.getRMWebServiceAddress().split(":"))[0]);
lastHearbeatTS = response.getLastHeartBeat(); lastHearbeatTS = response.getLastHeartBeat();
return response.getCapability(); return response.getCapability();
} }