YARN-8063. DistributedShellTimelinePlugin wrongly check for entityId instead of entityType. Contributed by Rohith Sharma K S.
(cherry picked from commit 22c5ddb7c4
)
This commit is contained in:
parent
59ab86311c
commit
e10e6c9012
|
@ -52,7 +52,7 @@ public class DistributedShellTimelinePlugin extends TimelineEntityGroupPlugin {
|
|||
@Override
|
||||
public Set<TimelineEntityGroupId> getTimelineEntityGroupId(String entityId,
|
||||
String entityType) {
|
||||
if (ApplicationMaster.DSEntity.DS_CONTAINER.toString().equals(entityId)) {
|
||||
if (ApplicationMaster.DSEntity.DS_CONTAINER.toString().equals(entityType)) {
|
||||
ContainerId containerId = ContainerId.fromString(entityId);
|
||||
ApplicationId appId = containerId.getApplicationAttemptId()
|
||||
.getApplicationId();
|
||||
|
|
|
@ -494,6 +494,15 @@ public class TestDistributedShell {
|
|||
Assert.assertEquals(2, entities.getEntities().size());
|
||||
Assert.assertEquals(entities.getEntities().get(0).getEntityType()
|
||||
.toString(), ApplicationMaster.DSEntity.DS_CONTAINER.toString());
|
||||
|
||||
String entityId = entities.getEntities().get(0).getEntityId();
|
||||
org.apache.hadoop.yarn.api.records.timeline.TimelineEntity entity =
|
||||
yarnCluster.getApplicationHistoryServer().getTimelineStore()
|
||||
.getEntity(entityId,
|
||||
ApplicationMaster.DSEntity.DS_CONTAINER.toString(), null);
|
||||
Assert.assertNotNull(entity);
|
||||
Assert.assertEquals(entityId, entity.getEntityId());
|
||||
|
||||
if (haveDomain) {
|
||||
Assert.assertEquals(domain.getId(),
|
||||
entities.getEntities().get(0).getDomainId());
|
||||
|
|
Loading…
Reference in New Issue