Java Code Examples for org.apache.hadoop.yarn.api.records.timeline.TimelineEntity#getDomainId()

The following examples show how to use org.apache.hadoop.yarn.api.records.timeline.TimelineEntity#getDomainId() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: MemoryTimelineStore.java From hadoop with Apache License 2.0

4 votes

@Override
public synchronized TimelineEntities getEntities(String entityType, Long limit,
    Long windowStart, Long windowEnd, String fromId, Long fromTs,
    NameValuePair primaryFilter, Collection<NameValuePair> secondaryFilters,
    EnumSet<Field> fields, CheckAcl checkAcl) throws IOException {
  if (limit == null) {
    limit = DEFAULT_LIMIT;
  }
  if (windowStart == null) {
    windowStart = Long.MIN_VALUE;
  }
  if (windowEnd == null) {
    windowEnd = Long.MAX_VALUE;
  }
  if (fields == null) {
    fields = EnumSet.allOf(Field.class);
  }

  Iterator<TimelineEntity> entityIterator = null;
  if (fromId != null) {
    TimelineEntity firstEntity = entities.get(new EntityIdentifier(fromId,
        entityType));
    if (firstEntity == null) {
      return new TimelineEntities();
    } else {
      entityIterator = new TreeSet<TimelineEntity>(entities.values())
          .tailSet(firstEntity, true).iterator();
    }
  }
  if (entityIterator == null) {
    entityIterator = new PriorityQueue<TimelineEntity>(entities.values())
        .iterator();
  }

  List<TimelineEntity> entitiesSelected = new ArrayList<TimelineEntity>();
  while (entityIterator.hasNext()) {
    TimelineEntity entity = entityIterator.next();
    if (entitiesSelected.size() >= limit) {
      break;
    }
    if (!entity.getEntityType().equals(entityType)) {
      continue;
    }
    if (entity.getStartTime() <= windowStart) {
      continue;
    }
    if (entity.getStartTime() > windowEnd) {
      continue;
    }
    if (fromTs != null && entityInsertTimes.get(new EntityIdentifier(
        entity.getEntityId(), entity.getEntityType())) > fromTs) {
      continue;
    }
    if (primaryFilter != null &&
        !matchPrimaryFilter(entity.getPrimaryFilters(), primaryFilter)) {
      continue;
    }
    if (secondaryFilters != null) { // AND logic
      boolean flag = true;
      for (NameValuePair secondaryFilter : secondaryFilters) {
        if (secondaryFilter != null && !matchPrimaryFilter(
            entity.getPrimaryFilters(), secondaryFilter) &&
            !matchFilter(entity.getOtherInfo(), secondaryFilter)) {
          flag = false;
          break;
        }
      }
      if (!flag) {
        continue;
      }
    }
    if (entity.getDomainId() == null) {
      entity.setDomainId(DEFAULT_DOMAIN_ID);
    }
    if (checkAcl == null || checkAcl.check(entity)) {
      entitiesSelected.add(entity);
    }
  }
  List<TimelineEntity> entitiesToReturn = new ArrayList<TimelineEntity>();
  for (TimelineEntity entitySelected : entitiesSelected) {
    entitiesToReturn.add(maskFields(entitySelected, fields));
  }
  Collections.sort(entitiesToReturn);
  TimelineEntities entitiesWrapper = new TimelineEntities();
  entitiesWrapper.setEntities(entitiesToReturn);
  return entitiesWrapper;
}

Example 2

Source File: TimelineDataManager.java From hadoop with Apache License 2.0

4 votes

private static void addDefaultDomainIdIfAbsent(TimelineEntity entity) {
  // be compatible with the timeline data created before 2.6
  if (entity.getDomainId() == null) {
    entity.setDomainId(DEFAULT_DOMAIN_ID);
  }
}

Example 3

Source File: MemoryTimelineStore.java From big-c with Apache License 2.0

4 votes

@Override
public synchronized TimelineEntities getEntities(String entityType, Long limit,
    Long windowStart, Long windowEnd, String fromId, Long fromTs,
    NameValuePair primaryFilter, Collection<NameValuePair> secondaryFilters,
    EnumSet<Field> fields, CheckAcl checkAcl) throws IOException {
  if (limit == null) {
    limit = DEFAULT_LIMIT;
  }
  if (windowStart == null) {
    windowStart = Long.MIN_VALUE;
  }
  if (windowEnd == null) {
    windowEnd = Long.MAX_VALUE;
  }
  if (fields == null) {
    fields = EnumSet.allOf(Field.class);
  }

  Iterator<TimelineEntity> entityIterator = null;
  if (fromId != null) {
    TimelineEntity firstEntity = entities.get(new EntityIdentifier(fromId,
        entityType));
    if (firstEntity == null) {
      return new TimelineEntities();
    } else {
      entityIterator = new TreeSet<TimelineEntity>(entities.values())
          .tailSet(firstEntity, true).iterator();
    }
  }
  if (entityIterator == null) {
    entityIterator = new PriorityQueue<TimelineEntity>(entities.values())
        .iterator();
  }

  List<TimelineEntity> entitiesSelected = new ArrayList<TimelineEntity>();
  while (entityIterator.hasNext()) {
    TimelineEntity entity = entityIterator.next();
    if (entitiesSelected.size() >= limit) {
      break;
    }
    if (!entity.getEntityType().equals(entityType)) {
      continue;
    }
    if (entity.getStartTime() <= windowStart) {
      continue;
    }
    if (entity.getStartTime() > windowEnd) {
      continue;
    }
    if (fromTs != null && entityInsertTimes.get(new EntityIdentifier(
        entity.getEntityId(), entity.getEntityType())) > fromTs) {
      continue;
    }
    if (primaryFilter != null &&
        !matchPrimaryFilter(entity.getPrimaryFilters(), primaryFilter)) {
      continue;
    }
    if (secondaryFilters != null) { // AND logic
      boolean flag = true;
      for (NameValuePair secondaryFilter : secondaryFilters) {
        if (secondaryFilter != null && !matchPrimaryFilter(
            entity.getPrimaryFilters(), secondaryFilter) &&
            !matchFilter(entity.getOtherInfo(), secondaryFilter)) {
          flag = false;
          break;
        }
      }
      if (!flag) {
        continue;
      }
    }
    if (entity.getDomainId() == null) {
      entity.setDomainId(DEFAULT_DOMAIN_ID);
    }
    if (checkAcl == null || checkAcl.check(entity)) {
      entitiesSelected.add(entity);
    }
  }
  List<TimelineEntity> entitiesToReturn = new ArrayList<TimelineEntity>();
  for (TimelineEntity entitySelected : entitiesSelected) {
    entitiesToReturn.add(maskFields(entitySelected, fields));
  }
  Collections.sort(entitiesToReturn);
  TimelineEntities entitiesWrapper = new TimelineEntities();
  entitiesWrapper.setEntities(entitiesToReturn);
  return entitiesWrapper;
}

Example 4

Source File: TimelineDataManager.java From big-c with Apache License 2.0

4 votes

private static void addDefaultDomainIdIfAbsent(TimelineEntity entity) {
  // be compatible with the timeline data created before 2.6
  if (entity.getDomainId() == null) {
    entity.setDomainId(DEFAULT_DOMAIN_ID);
  }
}