YARN-8215. ATS v2 returns invalid YARN_CONTAINER_ALLOCATED_HOST_HTTP_ADDRESS from NM. Contributed by Rohith Sharma K S.
(cherry picked from commit 7fc09c4f72
)
This commit is contained in:
parent
ab2b4290a0
commit
a91d5c7e2c
|
@ -24,6 +24,8 @@ import java.util.HashMap;
|
|||
import java.util.Map;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
import org.apache.hadoop.yarn.webapp.util.WebAppUtils;
|
||||
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.apache.hadoop.conf.Configuration;
|
||||
|
@ -79,6 +81,7 @@ public class NMTimelinePublisher extends CompositeService {
|
|||
private NodeId nodeId;
|
||||
|
||||
private String httpAddress;
|
||||
private String httpPort;
|
||||
|
||||
private UserGroupInformation nmLoginUGI;
|
||||
|
||||
|
@ -100,6 +103,12 @@ public class NMTimelinePublisher extends CompositeService {
|
|||
UserGroupInformation.getLoginUser() :
|
||||
UserGroupInformation.getCurrentUser();
|
||||
LOG.info("Initialized NMTimelinePublisher UGI to " + nmLoginUGI);
|
||||
|
||||
String webAppURLWithoutScheme =
|
||||
WebAppUtils.getNMWebAppURLWithoutScheme(conf);
|
||||
if (webAppURLWithoutScheme.contains(":")) {
|
||||
httpPort = webAppURLWithoutScheme.split(":")[1];
|
||||
}
|
||||
super.serviceInit(conf);
|
||||
}
|
||||
|
||||
|
@ -109,6 +118,7 @@ public class NMTimelinePublisher extends CompositeService {
|
|||
// context will be updated after containerManagerImpl is started
|
||||
// hence NMMetricsPublisher is added subservice of containerManagerImpl
|
||||
this.nodeId = context.getNodeId();
|
||||
this.httpAddress = nodeId.getHost() + ":" + httpPort;
|
||||
}
|
||||
|
||||
@Override
|
||||
|
@ -330,11 +340,6 @@ public class NMTimelinePublisher extends CompositeService {
|
|||
|
||||
public void publishContainerEvent(ContainerEvent event) {
|
||||
// publish only when the desired event is received
|
||||
if (this.httpAddress == null) {
|
||||
// update httpAddress for first time. When this service started,
|
||||
// web server will not be started.
|
||||
this.httpAddress = nodeId.getHost() + ":" + context.getHttpPort();
|
||||
}
|
||||
switch (event.getType()) {
|
||||
case INIT_CONTAINER:
|
||||
publishContainerCreatedEvent(event);
|
||||
|
|
|
@ -59,7 +59,6 @@ public class TestNMTimelinePublisher {
|
|||
@SuppressWarnings("unchecked")
|
||||
final DummyTimelineClient timelineClient = new DummyTimelineClient(null);
|
||||
when(context.getNodeId()).thenReturn(NodeId.newInstance("localhost", 0));
|
||||
when(context.getHttpPort()).thenReturn(0);
|
||||
|
||||
Configuration conf = new Configuration();
|
||||
conf.setBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED, true);
|
||||
|
|
Loading…
Reference in New Issue