YARN-6695. Fixed NPE in publishing appFinished events to ATSv2.

Contributed by Prabhu Joseph
This commit is contained in:
Eric Yang 2019-04-18 12:29:37 -04:00
parent b979fdde99
commit df76cdc895
3 changed files with 56 additions and 12 deletions

View File

@ -473,10 +473,15 @@ public class TimelineServiceV2Publisher extends AbstractSystemMetricsPublisher {
}
TimelineCollector timelineCollector =
rmTimelineCollectorManager.get(appId);
TimelineEntities entities = new TimelineEntities();
entities.addEntity(entity);
timelineCollector.putEntities(entities,
UserGroupInformation.getCurrentUser());
if (timelineCollector != null) {
TimelineEntities entities = new TimelineEntities();
entities.addEntity(entity);
timelineCollector.putEntities(entities,
UserGroupInformation.getCurrentUser());
} else {
LOG.debug("Cannot find active collector while publishing entity "
+ entity);
}
} catch (IOException e) {
LOG.error("Error when publishing entity " + entity);
LOG.debug("Error when publishing entity {}", entity, e);

View File

@ -28,12 +28,17 @@ import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import org.apache.log4j.AppenderSkeleton;
import org.apache.log4j.Logger;
import org.apache.log4j.spi.LoggingEvent;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileContext;
import org.apache.hadoop.fs.Path;
@ -293,6 +298,48 @@ public class TestSystemMetricsPublisherForV2 {
TimelineServiceHelper.invertLong(containerId.getContainerId()));
}
@Test(timeout = 10000)
public void testPutEntityWhenNoCollector() throws Exception {
// Validating the logs as DrainDispatcher won't throw exception
class TestAppender extends AppenderSkeleton {
private final List<LoggingEvent> log = new ArrayList<>();
@Override
public boolean requiresLayout() {
return false;
}
@Override
protected void append(final LoggingEvent loggingEvent) {
log.add(loggingEvent);
}
@Override
public void close() {
}
public List<LoggingEvent> getLog() {
return new ArrayList<>(log);
}
}
TestAppender appender = new TestAppender();
final Logger logger = Logger.getRootLogger();
logger.addAppender(appender);
try {
RMApp app = createRMApp(ApplicationId.newInstance(0, 1));
metricsPublisher.appCreated(app, app.getStartTime());
dispatcher.await();
for (LoggingEvent event : appender.getLog()) {
assertFalse("Dispatcher Crashed",
event.getRenderedMessage().contains("Error in dispatcher thread"));
}
} finally {
logger.removeAppender(appender);
}
}
private RMApp createAppAndRegister(ApplicationId appId) {
RMApp app = createRMApp(appId);

View File

@ -330,14 +330,6 @@ Following are the basic configurations to start Timeline service v.2:
<name>yarn.system-metrics-publisher.enabled</name>
<value>true</value>
</property>
<property>
<description>The setting that controls whether yarn container events are
published to the timeline service or not by RM. This configuration setting
is for ATS V2.</description>
<name>yarn.rm.system-metrics-publisher.emit-container-events</name>
<value>true</value>
</property>
```
If using an aux services manifest instead of setting aux services through the Configuration, ensure that the manifest services array includes the timeline\_collector service as follows: