Java Code Examples for org.apache.atlas.model.instance.AtlasEntity#getGuid()

The following examples show how to use org.apache.atlas.model.instance.AtlasEntity#getGuid() . 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: AtlasEntityGraphDiscoveryV1.java From incubator-atlas with Apache License 2.0

6 votes

@Override
public void validateAndNormalizeForUpdate(AtlasEntity entity) throws AtlasBaseException {
    List<String> messages = new ArrayList<>();

    if (! AtlasTypeUtil.isValidGuid(entity.getGuid())) {
        throw new AtlasBaseException(AtlasErrorCode.INVALID_OBJECT_ID, "invalid guid " + entity.getGuid());
    }

    AtlasEntityType type = typeRegistry.getEntityTypeByName(entity.getTypeName());

    if (type == null) {
        throw new AtlasBaseException(AtlasErrorCode.TYPE_NAME_INVALID, TypeCategory.ENTITY.name(), entity.getTypeName());
    }

    type.validateValueForUpdate(entity, entity.getTypeName(), messages);

    if (!messages.isEmpty()) {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_CRUD_INVALID_PARAMS, messages);
    }

    type.getNormalizedValueForUpdate(entity);
}

Example 2

Source File: AtlasEntityGraphDiscoveryV1.java From incubator-atlas with Apache License 2.0

6 votes

@Override
public void validateAndNormalize(AtlasEntity entity) throws AtlasBaseException {
    List<String> messages = new ArrayList<>();

    if (! AtlasTypeUtil.isValidGuid(entity.getGuid())) {
        throw new AtlasBaseException(AtlasErrorCode.INVALID_OBJECT_ID, "invalid guid " + entity.getGuid());
    }

    AtlasEntityType type = typeRegistry.getEntityTypeByName(entity.getTypeName());

    if (type == null) {
        throw new AtlasBaseException(AtlasErrorCode.TYPE_NAME_INVALID, TypeCategory.ENTITY.name(), entity.getTypeName());
    }

    type.validateValue(entity, entity.getTypeName(), messages);

    if (!messages.isEmpty()) {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_CRUD_INVALID_PARAMS, messages);
    }

    type.getNormalizedValue(entity);
}

Example 3

Source File: AtlasUserProfileDTO.java From atlas with Apache License 2.0

6 votes

@Override
public AtlasEntityWithExtInfo toEntityWithExtInfo(AtlasUserProfile obj) throws AtlasBaseException {
    AtlasEntity            entity            = toEntity(obj);
    AtlasEntityWithExtInfo entityWithExtInfo = new AtlasEntityWithExtInfo(entity);

    AtlasObjectId userProfileId = new AtlasObjectId(entity.getGuid(), AtlasUserProfileDTO.ENTITY_TYPE_NAME,
                                                    Collections.singletonMap(AtlasUserProfileDTO.PROPERTY_USER_NAME, obj.getName()));

    List<AtlasObjectId> objectIds = new ArrayList<>();

    for (AtlasUserSavedSearch ss : obj.getSavedSearches()) {
        AtlasEntity ssEntity = savedSearchDTO.toEntity(ss);

        ssEntity.setAttribute(AtlasSavedSearchDTO.PROPERTY_USER_PROFILE, userProfileId);

        entityWithExtInfo.addReferredEntity(ssEntity);

        objectIds.add(new AtlasObjectId(ssEntity.getGuid(), savedSearchDTO.getEntityType().getTypeName(), savedSearchDTO.getUniqueAttributes(ss)));
    }

    if (objectIds.size() > 0) {
        entity.setAttribute(PROPERTY_SAVED_SEARCHES, objectIds);
    }

    return entityWithExtInfo;
}

Example 4

Source File: AtlasEntityGraphDiscoveryV2.java From atlas with Apache License 2.0

6 votes

@Override
public void validateAndNormalize(AtlasEntity entity) throws AtlasBaseException {
    List<String> messages = new ArrayList<>();

    if (!AtlasTypeUtil.isValidGuid(entity.getGuid())) {
        throw new AtlasBaseException(AtlasErrorCode.INVALID_OBJECT_ID, "invalid guid " + entity.getGuid());
    }

    AtlasEntityType type = typeRegistry.getEntityTypeByName(entity.getTypeName());

    if (type == null) {
        throw new AtlasBaseException(AtlasErrorCode.TYPE_NAME_INVALID, TypeCategory.ENTITY.name(), entity.getTypeName());
    }

    validateCustomAttributes(entity);

    validateLabels(entity.getLabels());

    type.validateValue(entity, entity.getTypeName(), messages);

    if (!messages.isEmpty()) {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_CRUD_INVALID_PARAMS, messages);
    }

    type.getNormalizedValue(entity);
}

Example 5

Source File: AtlasEntityGraphDiscoveryV2.java From atlas with Apache License 2.0

6 votes

@Override
public void validateAndNormalizeForUpdate(AtlasEntity entity) throws AtlasBaseException {
    List<String> messages = new ArrayList<>();

    if (!AtlasTypeUtil.isValidGuid(entity.getGuid())) {
        throw new AtlasBaseException(AtlasErrorCode.INVALID_OBJECT_ID, "invalid guid " + entity.getGuid());
    }

    AtlasEntityType type = typeRegistry.getEntityTypeByName(entity.getTypeName());

    if (type == null) {
        throw new AtlasBaseException(AtlasErrorCode.TYPE_NAME_INVALID, TypeCategory.ENTITY.name(), entity.getTypeName());
    }

    validateCustomAttributes(entity);

    validateLabels(entity.getLabels());

    type.validateValueForUpdate(entity, entity.getTypeName(), messages);

    if (!messages.isEmpty()) {
        throw new AtlasBaseException(AtlasErrorCode.INSTANCE_CRUD_INVALID_PARAMS, messages);
    }

    type.getNormalizedValueForUpdate(entity);
}

Example 6

Source File: EntityAuditListenerV2.java From atlas with Apache License 2.0

5 votes

@Override
public void onClassificationsUpdated(AtlasEntity entity, List<AtlasClassification> classifications) throws AtlasBaseException {
    if (CollectionUtils.isNotEmpty(classifications)) {
        MetricRecorder metric = RequestContext.get().startMetricRecord("entityAudit");

        List<EntityAuditEventV2> events = new ArrayList<>();
        String                   guid   = entity.getGuid();

        for (AtlasClassification classification : classifications) {
            if (guid.equals(classification.getEntityGuid())) {
                events.add(createEvent(entity, CLASSIFICATION_UPDATE, "Updated classification: " + AtlasType.toJson(classification)));
            } else {
                if (isPropagatedClassificationAdded(guid, classification)) {
                    events.add(createEvent(entity, PROPAGATED_CLASSIFICATION_ADD, "Added propagated classification: " + AtlasType.toJson(classification)));
                } else if (isPropagatedClassificationDeleted(guid, classification)) {
                    events.add(createEvent(entity, PROPAGATED_CLASSIFICATION_DELETE, "Deleted propagated classification: " + classification.getTypeName()));
                } else {
                    events.add(createEvent(entity, PROPAGATED_CLASSIFICATION_UPDATE, "Updated propagated classification: " + AtlasType.toJson(classification)));
                }
            }
        }

        auditRepository.putEventsV2(events);

        RequestContext.get().endMetricRecord(metric);
    }
}

Example 7

Source File: QuickStartV2.java From incubator-atlas with Apache License 2.0

5 votes

private String getTableId(String tableName) throws AtlasServiceException {
    Map<String, String> attributes = new HashMap<>();
    attributes.put(AtlasClient.REFERENCEABLE_ATTRIBUTE_NAME, tableName);

    AtlasEntity tableEntity = atlasClientV2.getEntityByAttribute(TABLE_TYPE, attributes).getEntity();
    return tableEntity.getGuid();
}

Example 8

Source File: BulkImporterImpl.java From atlas with Apache License 2.0

5 votes

public static void updateVertexGuid(AtlasGraph atlasGraph, AtlasTypeRegistry typeRegistry, EntityGraphRetriever entityGraphRetriever, AtlasEntity entity) {
    String entityGuid = entity.getGuid();
    AtlasObjectId objectId = entityGraphRetriever.toAtlasObjectIdWithoutGuid(entity);

    AtlasEntityType entityType = typeRegistry.getEntityTypeByName(entity.getTypeName());
    String vertexGuid = null;
    try {
        vertexGuid = AtlasGraphUtilsV2.getGuidByUniqueAttributes(atlasGraph, entityType, objectId.getUniqueAttributes());
    } catch (AtlasBaseException e) {
        LOG.warn("Entity: {}: Does not exist!", objectId);
        return;
    }

    if (StringUtils.isEmpty(vertexGuid) || vertexGuid.equals(entityGuid)) {
        return;
    }

    AtlasVertex v = AtlasGraphUtilsV2.findByGuid(atlasGraph, vertexGuid);
    if (v == null) {
        return;
    }

    addHistoricalGuid(v, vertexGuid);
    AtlasGraphUtilsV2.setProperty(v, Constants.GUID_PROPERTY_KEY, entityGuid);

    LOG.warn("GUID Updated: Entity: {}: from: {}: to: {}", objectId, vertexGuid, entity.getGuid());
}

Example 9

Source File: HiveMetastoreHookIT.java From atlas with Apache License 2.0

5 votes

private String getColumnId(AtlasEntityWithExtInfo entityWithExtInfo, String columnName) {
    String ret = null;

    for (AtlasEntity entity : entityWithExtInfo.getReferredEntities().values()) {

        if (entity.getTypeName().equals("hive_column") && entity.getAttribute("name").equals(columnName)) {
            ret = entity.getGuid();
            break;
        }
    }

    return ret;
}

Example 10

Source File: BaseImpalaEvent.java From atlas with Apache License 2.0

5 votes

public static AtlasObjectId getObjectId(AtlasEntity entity) {
    String        qualifiedName = (String) entity.getAttribute(ATTRIBUTE_QUALIFIED_NAME);
    AtlasObjectId ret           = new AtlasObjectId(entity.getGuid(), entity.getTypeName(), Collections
        .singletonMap(ATTRIBUTE_QUALIFIED_NAME, qualifiedName));

    return ret;
}

Example 11

Source File: AtlasTypeUtil.java From atlas with Apache License 2.0

4 votes

public static AtlasObjectId getAtlasObjectId(AtlasEntity entity) {
    return new AtlasObjectId(entity.getGuid(), entity.getTypeName());
}

Example 12

Source File: RequestContext.java From atlas with Apache License 2.0

4 votes

public void cache(AtlasEntity entity) {
    if (entity != null && entity.getGuid() != null) {
        entityCache.put(entity.getGuid(), entity);
    }
}

Example 13

Source File: TestUtilsV2.java From atlas with Apache License 2.0

4 votes

public static Map<String, AtlasEntity> createDeptEg1() {
    Map<String, AtlasEntity> deptEmpEntities = new HashMap<>();

    AtlasEntity hrDept = new AtlasEntity(DEPARTMENT_TYPE);
    AtlasEntity john = new AtlasEntity(EMPLOYEE_TYPE);

    AtlasEntity jane = new AtlasEntity("Manager");
    AtlasEntity johnAddr = new AtlasEntity("Address");
    AtlasEntity janeAddr = new AtlasEntity("Address");
    AtlasEntity julius = new AtlasEntity("Manager");
    AtlasEntity juliusAddr = new AtlasEntity("Address");
    AtlasEntity max = new AtlasEntity(EMPLOYEE_TYPE);
    AtlasEntity maxAddr = new AtlasEntity("Address");

    AtlasObjectId deptId = new AtlasObjectId(hrDept.getGuid(), hrDept.getTypeName());
    hrDept.setAttribute("name", "hr");
    john.setAttribute("name", "John");
    john.setAttribute("department", deptId);
    johnAddr.setAttribute("street", "Stewart Drive");
    johnAddr.setAttribute("city", "Sunnyvale");
    john.setAttribute("address", johnAddr);

    john.setAttribute("birthday",new Date(1950, 5, 15));
    john.setAttribute("hasPets", true);
    john.setAttribute("numberOfCars", 1);
    john.setAttribute("houseNumber", 153);
    john.setAttribute("carMileage", 13364);
    john.setAttribute("shares", 15000);
    john.setAttribute("salary", 123345.678);
    john.setAttribute("age", 50);
    john.setAttribute("numberOfStarsEstimate", new BigInteger("1000000000000000000000"));
    john.setAttribute("approximationOfPi", new BigDecimal("3.141592653589793238462643383279502884197169399375105820974944592307816406286"));

    jane.setAttribute("name", "Jane");
    jane.setAttribute("department", deptId);
    janeAddr.setAttribute("street", "Great America Parkway");
    janeAddr.setAttribute("city", "Santa Clara");
    jane.setAttribute("address", janeAddr);
    janeAddr.setAttribute("street", "Great America Parkway");

    julius.setAttribute("name", "Julius");
    julius.setAttribute("department", deptId);
    juliusAddr.setAttribute("street", "Madison Ave");
    juliusAddr.setAttribute("city", "Newtonville");
    julius.setAttribute("address", juliusAddr);
    julius.setAttribute("subordinates", Collections.emptyList());

    AtlasObjectId janeId = getAtlasObjectId(jane);
    AtlasObjectId johnId = getAtlasObjectId(john);

    //TODO - Change to MANAGER_TYPE for JULIUS
    AtlasObjectId maxId = new AtlasObjectId(max.getGuid(), EMPLOYEE_TYPE);
    AtlasObjectId juliusId = new AtlasObjectId(julius.getGuid(), EMPLOYEE_TYPE);

    max.setAttribute("name", "Max");
    max.setAttribute("department", deptId);
    maxAddr.setAttribute("street", "Ripley St");
    maxAddr.setAttribute("city", "Newton");
    max.setAttribute("address", maxAddr);
    max.setAttribute("manager", janeId);
    max.setAttribute("mentor", juliusId);
    max.setAttribute("birthday",new Date(1979, 3, 15));
    max.setAttribute("hasPets", true);
    max.setAttribute("age", 36);
    max.setAttribute("numberOfCars", 2);
    max.setAttribute("houseNumber", 17);
    max.setAttribute("carMileage", 13);
    max.setAttribute("shares", Long.MAX_VALUE);
    max.setAttribute("salary", Double.MAX_VALUE);
    max.setAttribute("numberOfStarsEstimate", new BigInteger("1000000000000000000000000000000"));
    max.setAttribute("approximationOfPi", new BigDecimal("3.1415926535897932"));

    john.setAttribute("manager", janeId);
    john.setAttribute("mentor", maxId);
    hrDept.setAttribute("employees", Arrays.asList(johnId, janeId, juliusId, maxId));

    jane.setAttribute("subordinates", Arrays.asList(johnId, maxId));

    deptEmpEntities.put(jane.getGuid(), jane);
    deptEmpEntities.put(john.getGuid(), john);
    deptEmpEntities.put(julius.getGuid(), julius);
    deptEmpEntities.put(max.getGuid(), max);
    deptEmpEntities.put(deptId.getGuid(), hrDept);
    return deptEmpEntities;
}

Example 14

Source File: AtlasTypeUtil.java From atlas with Apache License 2.0

4 votes

public static AtlasObjectId getObjectId(AtlasEntity entity) {
    String qualifiedName = (String) entity.getAttribute(ATTRIBUTE_QUALIFIED_NAME);
    AtlasObjectId ret = new AtlasObjectId(entity.getGuid(), entity.getTypeName(), Collections.singletonMap(ATTRIBUTE_QUALIFIED_NAME, qualifiedName));

    return ret;
}

Example 15

Source File: NiFiFlowPath.java From nifi with Apache License 2.0

4 votes

public void setExEntity(AtlasEntity exEntity) {
    this.exEntity = exEntity;
    this.atlasGuid = exEntity.getGuid();
}

Example 16

Source File: QuickStartV2.java From atlas with Apache License 2.0

4 votes

private String getTableId(String tableName) throws AtlasServiceException {
    Map<String, String> attributes  = Collections.singletonMap(REFERENCEABLE_ATTRIBUTE_NAME, tableName + CLUSTER_SUFFIX);
    AtlasEntity         tableEntity = atlasClientV2.getEntityByAttribute(TABLE_TYPE, attributes).getEntity();

    return tableEntity.getGuid();
}

Example 17

Source File: TestUtilsV2.java From incubator-atlas with Apache License 2.0

4 votes

public static Map<String, AtlasEntity> createDeptEg1() {
    Map<String, AtlasEntity> deptEmpEntities = new HashMap<>();

    AtlasEntity hrDept = new AtlasEntity(DEPARTMENT_TYPE);
    AtlasEntity john = new AtlasEntity(EMPLOYEE_TYPE);

    AtlasEntity jane = new AtlasEntity("Manager");
    AtlasEntity johnAddr = new AtlasEntity("Address");
    AtlasEntity janeAddr = new AtlasEntity("Address");
    AtlasEntity julius = new AtlasEntity("Manager");
    AtlasEntity juliusAddr = new AtlasEntity("Address");
    AtlasEntity max = new AtlasEntity(EMPLOYEE_TYPE);
    AtlasEntity maxAddr = new AtlasEntity("Address");

    AtlasObjectId deptId = new AtlasObjectId(hrDept.getGuid(), hrDept.getTypeName());
    hrDept.setAttribute("name", "hr");
    john.setAttribute("name", "John");
    john.setAttribute("department", deptId);
    johnAddr.setAttribute("street", "Stewart Drive");
    johnAddr.setAttribute("city", "Sunnyvale");
    john.setAttribute("address", johnAddr);

    john.setAttribute("birthday",new Date(1950, 5, 15));
    john.setAttribute("hasPets", true);
    john.setAttribute("numberOfCars", 1);
    john.setAttribute("houseNumber", 153);
    john.setAttribute("carMileage", 13364);
    john.setAttribute("shares", 15000);
    john.setAttribute("salary", 123345.678);
    john.setAttribute("age", 50);
    john.setAttribute("numberOfStarsEstimate", new BigInteger("1000000000000000000000"));
    john.setAttribute("approximationOfPi", new BigDecimal("3.141592653589793238462643383279502884197169399375105820974944592307816406286"));

    jane.setAttribute("name", "Jane");
    jane.setAttribute("department", deptId);
    janeAddr.setAttribute("street", "Great America Parkway");
    janeAddr.setAttribute("city", "Santa Clara");
    jane.setAttribute("address", janeAddr);
    janeAddr.setAttribute("street", "Great America Parkway");

    julius.setAttribute("name", "Julius");
    julius.setAttribute("department", deptId);
    juliusAddr.setAttribute("street", "Madison Ave");
    juliusAddr.setAttribute("city", "Newtonville");
    julius.setAttribute("address", juliusAddr);
    julius.setAttribute("subordinates", ImmutableList.of());

    AtlasObjectId janeId = AtlasTypeUtil.getAtlasObjectId(jane);
    AtlasObjectId johnId = AtlasTypeUtil.getAtlasObjectId(john);

    //TODO - Change to MANAGER_TYPE for JULIUS
    AtlasObjectId maxId = new AtlasObjectId(max.getGuid(), EMPLOYEE_TYPE);
    AtlasObjectId juliusId = new AtlasObjectId(julius.getGuid(), EMPLOYEE_TYPE);

    max.setAttribute("name", "Max");
    max.setAttribute("department", deptId);
    maxAddr.setAttribute("street", "Ripley St");
    maxAddr.setAttribute("city", "Newton");
    max.setAttribute("address", maxAddr);
    max.setAttribute("manager", janeId);
    max.setAttribute("mentor", juliusId);
    max.setAttribute("birthday",new Date(1979, 3, 15));
    max.setAttribute("hasPets", true);
    max.setAttribute("age", 36);
    max.setAttribute("numberOfCars", 2);
    max.setAttribute("houseNumber", 17);
    max.setAttribute("carMileage", 13);
    max.setAttribute("shares", Long.MAX_VALUE);
    max.setAttribute("salary", Double.MAX_VALUE);
    max.setAttribute("numberOfStarsEstimate", new BigInteger("1000000000000000000000000000000"));
    max.setAttribute("approximationOfPi", new BigDecimal("3.1415926535897932"));

    john.setAttribute("manager", janeId);
    john.setAttribute("mentor", maxId);
    hrDept.setAttribute("employees", ImmutableList.of(johnId, janeId, juliusId, maxId));

    jane.setAttribute("subordinates", ImmutableList.of(johnId, maxId));

    deptEmpEntities.put(jane.getGuid(), jane);
    deptEmpEntities.put(john.getGuid(), john);
    deptEmpEntities.put(julius.getGuid(), julius);
    deptEmpEntities.put(max.getGuid(), max);
    deptEmpEntities.put(deptId.getGuid(), hrDept);
    return deptEmpEntities;
}

Example 18

Source File: AtlasTypeUtil.java From incubator-atlas with Apache License 2.0

4 votes

public static AtlasObjectId getAtlasObjectId(AtlasEntity entity) {
    return new AtlasObjectId(entity.getGuid(), entity.getTypeName());
}

Example 19

Source File: EntityAuditListenerV2.java From atlas with Apache License 2.0

4 votes

private EntityAuditEventV2 createEvent(AtlasEntity entity, EntityAuditActionV2 action, String details) {
    return new EntityAuditEventV2(entity.getGuid(), RequestContext.get().getRequestTime(),
                                  RequestContext.get().getUser(), action, details, entity);
}

Example 20

Source File: EntityNotificationListenerV2.java From atlas with Apache License 2.0

4 votes

private AtlasEntityHeader toNotificationHeader(AtlasEntity entity) {
    AtlasEntityHeader ret         = new AtlasEntityHeader(entity.getTypeName(), entity.getGuid(), new HashMap<>());
    Object            name        = entity.getAttribute(NAME);
    Object            displayText = name != null ? name : entity.getAttribute(QUALIFIED_NAME);

    ret.setGuid(entity.getGuid());
    ret.setStatus(entity.getStatus());
    ret.setIsIncomplete(entity.getIsIncomplete());

    setAttribute(ret, NAME, name);
    setAttribute(ret, DESCRIPTION, entity.getAttribute(DESCRIPTION));
    setAttribute(ret, OWNER, entity.getAttribute(OWNER));
    setAttribute(ret, CREATE_TIME, entity.getAttribute(CREATE_TIME));

    if (displayText != null) {
        ret.setDisplayText(displayText.toString());
    }

    AtlasEntityType entityType = typeRegistry.getEntityTypeByName(entity.getTypeName());

    if (entityType != null) {
        for (AtlasAttribute attribute : entityType.getAllAttributes().values()) {
            if (attribute.getAttributeDef().getIsUnique() || attribute.getAttributeDef().getIncludeInNotification()) {
                Object attrValue = entity.getAttribute(attribute.getName());

                if (attrValue != null) {
                    ret.setAttribute(attribute.getName(), attrValue);
                }
            }
        }

        if (CollectionUtils.isNotEmpty(entity.getClassifications())) {
            List<AtlasClassification> classifications     = new ArrayList<>(entity.getClassifications().size());
            List<String>              classificationNames = new ArrayList<>(entity.getClassifications().size());

            for (AtlasClassification classification : getAllClassifications(entity.getClassifications())) {
                classifications.add(classification);
                classificationNames.add(classification.getTypeName());
            }

            ret.setClassifications(classifications);
            ret.setClassificationNames(classificationNames);
        }
    }

    return ret;
}