YARN-8063. DistributedShellTimelinePlugin wrongly check for entityId instead of entityType. Contributed by Rohith Sharma K S.
This commit is contained in:
parent
8620d2bdf9
commit
22c5ddb7c4
|
@ -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