Examples with TimelineEvent org.apache.hadoop.yarn.api.records.timelineservice.TimelineEvent used on opensource projects

Example 56 with TimelineEvent

use of org.apache.hadoop.yarn.api.records.timelineservice.TimelineEvent in project hadoop by apache.

the class TestMRTimelineEventHandling method verifyEntity.

/**
   * Verifies entity by reading the entity file written via FS impl.
   * @param entityFile File to be read.
   * @param eventId Event to be checked.
   * @param chkMetrics If event is not null, this flag determines if metrics
   *     exist when the event is encountered. If event is null, we merely check
   *     if metrics exist in the entity file.
   * @param chkCfg If event is not null, this flag determines if configs
   *     exist when the event is encountered. If event is null, we merely check
   *     if configs exist in the entity file.
   * @param cfgsToVerify a set of configs which should exist in the entity file.
   * @throws IOException
   */
private void verifyEntity(File entityFile, String eventId, boolean chkMetrics, boolean chkCfg, Set<String> cfgsToVerify) throws IOException {
    BufferedReader reader = null;
    String strLine;
    try {
        reader = new BufferedReader(new FileReader(entityFile));
        while ((strLine = reader.readLine()) != null) {
            if (strLine.trim().length() > 0) {
                org.apache.hadoop.yarn.api.records.timelineservice.TimelineEntity entity = FileSystemTimelineReaderImpl.getTimelineRecordFromJSON(strLine.trim(), org.apache.hadoop.yarn.api.records.timelineservice.TimelineEntity.class);
                if (eventId == null) {
                    // its found. Same applies to configs.
                    if (chkMetrics && entity.getMetrics().size() > 0) {
                        return;
                    }
                    if (chkCfg && entity.getConfigs().size() > 0) {
                        if (cfgsToVerify == null) {
                            return;
                        } else {
                            // not).
                            for (Iterator<String> itr = cfgsToVerify.iterator(); itr.hasNext(); ) {
                                String config = itr.next();
                                if (entity.getConfigs().containsKey(config)) {
                                    itr.remove();
                                }
                            }
                            // All the required configs have been verified, so return.
                            if (cfgsToVerify.isEmpty()) {
                                return;
                            }
                        }
                    }
                } else {
                    for (TimelineEvent event : entity.getEvents()) {
                        if (event.getId().equals(eventId)) {
                            if (chkMetrics) {
                                assertTrue(entity.getMetrics().size() > 0);
                            }
                            if (chkCfg) {
                                assertTrue(entity.getConfigs().size() > 0);
                                if (cfgsToVerify != null) {
                                    for (String cfg : cfgsToVerify) {
                                        assertTrue(entity.getConfigs().containsKey(cfg));
                                    }
                                }
                            }
                            return;
                        }
                    }
                }
            }
        }
        if (cfgsToVerify != null) {
            assertTrue(cfgsToVerify.isEmpty());
            return;
        }
        fail("Expected event : " + eventId + " not found in the file " + entityFile);
    } finally {
        reader.close();
    }
}