Java TimelineEntity.getEntityType方法代码示例

import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity; //导入方法依赖的package包/类
public TimelinePutResponse putEntities(
    TimelineEntity... entities) throws IOException, YarnException {
  TimelineEntities entitiesContainer = new TimelineEntities();
  for (TimelineEntity entity : entities) {
    if (entity.getEntityId() == null || entity.getEntityType() == null) {
      throw new YarnException("Incomplete entity without entity id/type");
    }
    entitiesContainer.addEntity(entity);
  }
  ClientResponse resp = doPosting(entitiesContainer, null);
  return resp.getEntity(TimelinePutResponse.class);
}
 

import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity; //导入方法依赖的package包/类
public boolean checkAccess(UserGroupInformation callerUGI,
    ApplicationAccessType applicationAccessType,
    TimelineEntity entity) throws YarnException, IOException {
  if (LOG.isDebugEnabled()) {
    LOG.debug("Verifying the access of "
        + (callerUGI == null ? null : callerUGI.getShortUserName())
        + " on the timeline entity "
        + new EntityIdentifier(entity.getEntityId(), entity.getEntityType()));
  }

  if (!adminAclsManager.areACLsEnabled()) {
    return true;
  }

  // find domain owner and acls
  AccessControlListExt aclExt = aclExts.get(entity.getDomainId());
  if (aclExt == null) {
    aclExt = loadDomainFromTimelineStore(entity.getDomainId());
  }
  if (aclExt == null) {
    throw new YarnException("Domain information of the timeline entity "
        + new EntityIdentifier(entity.getEntityId(), entity.getEntityType())
        + " doesn't exist.");
  }
  String owner = aclExt.owner;
  AccessControlList domainACL = aclExt.acls.get(applicationAccessType);
  if (domainACL == null) {
    if (LOG.isDebugEnabled()) {
      LOG.debug("ACL not found for access-type " + applicationAccessType
          + " for domain " + entity.getDomainId() + " owned by "
          + owner + ". Using default ["
          + YarnConfiguration.DEFAULT_YARN_APP_ACL + "]");
    }
    domainACL =
        new AccessControlList(YarnConfiguration.DEFAULT_YARN_APP_ACL);
  }

  if (callerUGI != null
      && (adminAclsManager.isAdmin(callerUGI) ||
          callerUGI.getShortUserName().equals(owner) ||
          domainACL.isUserAllowed(callerUGI))) {
    return true;
  }
  return false;
}
 

import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity; //导入方法依赖的package包/类
private TimelinePutResponse doPostEntities(
    TimelineEntities entities,
    UserGroupInformation callerUGI) throws YarnException, IOException {
  if (entities == null) {
    return new TimelinePutResponse();
  }
  metrics.incrPostEntitiesTotal(entities.getEntities().size());
  TimelineEntities entitiesToPut = new TimelineEntities();
  List<TimelinePutResponse.TimelinePutError> errors =
      new ArrayList<TimelinePutResponse.TimelinePutError>();
  for (TimelineEntity entity : entities.getEntities()) {

    // if the domain id is not specified, the entity will be put into
    // the default domain
    if (entity.getDomainId() == null ||
        entity.getDomainId().length() == 0) {
      entity.setDomainId(DEFAULT_DOMAIN_ID);
    }
    if (entity.getEntityId() == null || entity.getEntityType() == null) {
      throw new BadRequestException("Incomplete entity without entity"
          + " id/type");
    }
    // check if there is existing entity
    TimelineEntity existingEntity = null;
    try {
      existingEntity =
          store.getEntity(entity.getEntityId(), entity.getEntityType(),
              EnumSet.of(Field.PRIMARY_FILTERS));
      if (existingEntity != null) {
        addDefaultDomainIdIfAbsent(existingEntity);
        if (!existingEntity.getDomainId().equals(entity.getDomainId())) {
          throw new YarnException("The domain of the timeline entity "
            + "{ id: " + entity.getEntityId() + ", type: "
            + entity.getEntityType() + " } is not allowed to be changed from "
            + existingEntity.getDomainId() + " to " + entity.getDomainId());
        }
      }
      if (!timelineACLsManager.checkAccess(
          callerUGI, ApplicationAccessType.MODIFY_APP, entity)) {
        throw new YarnException(callerUGI
            + " is not allowed to put the timeline entity "
            + "{ id: " + entity.getEntityId() + ", type: "
            + entity.getEntityType() + " } into the domain "
            + entity.getDomainId() + ".");
      }
    } catch (Exception e) {
      // Skip the entity which already exists and was put by others
      LOG.warn("Skip the timeline entity: { id: " + entity.getEntityId()
          + ", type: "+ entity.getEntityType() + " }", e);
      TimelinePutResponse.TimelinePutError error =
          new TimelinePutResponse.TimelinePutError();
      error.setEntityId(entity.getEntityId());
      error.setEntityType(entity.getEntityType());
      error.setErrorCode(
          TimelinePutResponse.TimelinePutError.ACCESS_DENIED);
      errors.add(error);
      continue;
    }

    entitiesToPut.addEntity(entity);
  }

  TimelinePutResponse response = store.put(entitiesToPut);
  // add the errors of timeline system filter key conflict
  response.addErrors(errors);
  return response;
}