Examples with AtlasEdge org.apache.atlas.repository.graphdb.AtlasEdge used on opensource projects

Example 86 with AtlasEdge

use of org.apache.atlas.repository.graphdb.AtlasEdge in project atlas by apache.

the class AtlasJanusGraph method addEdge.

@Override
public AtlasEdge<AtlasJanusVertex, AtlasJanusEdge> addEdge(AtlasVertex<AtlasJanusVertex, AtlasJanusEdge> outVertex, AtlasVertex<AtlasJanusVertex, AtlasJanusEdge> inVertex, String edgeLabel) {
    try {
        Vertex oV = outVertex.getV().getWrappedElement();
        Vertex iV = inVertex.getV().getWrappedElement();
        Edge edge = oV.addEdge(edgeLabel, iV);
        return GraphDbObjectFactory.createEdge(this, edge);
    } catch (SchemaViolationException e) {
        throw new AtlasSchemaViolationException(e);
    }
}

Example 87 with AtlasEdge

use of org.apache.atlas.repository.graphdb.AtlasEdge in project atlas by apache.

the class AtlasRelationshipStoreV1 method getOrCreate.

@Override
public AtlasRelationship getOrCreate(AtlasRelationship relationship) throws AtlasBaseException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("==> getOrCreate({})", relationship);
    }
    validateRelationship(relationship);
    AtlasVertex end1Vertex = getVertexFromEndPoint(relationship.getEnd1());
    AtlasVertex end2Vertex = getVertexFromEndPoint(relationship.getEnd2());
    AtlasRelationship ret = null;
    // check if relationship exists
    AtlasEdge relationshipEdge = getRelationshipEdge(end1Vertex, end2Vertex, relationship.getTypeName());
    if (relationshipEdge == null) {
        validateRelationship(relationship);
        relationshipEdge = createRelationship(end1Vertex, end2Vertex, relationship);
    }
    if (relationshipEdge != null) {
        ret = entityRetriever.mapEdgeToAtlasRelationship(relationshipEdge);
    }
    if (LOG.isDebugEnabled()) {
        LOG.debug("<== getOrCreate({}): {}", relationship, ret);
    }
    return ret;
}

Example 88 with AtlasEdge

use of org.apache.atlas.repository.graphdb.AtlasEdge in project atlas by apache.

the class AtlasRelationshipStoreV1 method deleteById.

@Override
@GraphTransaction
public void deleteById(String guid) throws AtlasBaseException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("==> deleteById({})", guid);
    }
    if (StringUtils.isEmpty(guid)) {
        throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_CRUD_INVALID_PARAMS, " empty/null guid");
    }
    AtlasEdge edge = graphHelper.getEdgeForGUID(guid);
    if (edge == null) {
        throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_GUID_NOT_FOUND, guid);
    }
    if (getState(edge) == DELETED) {
        throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_ALREADY_DELETED, guid);
    }
    deleteHandler.deleteRelationships(Collections.singleton(edge));
    if (LOG.isDebugEnabled()) {
        LOG.debug("<== deleteById({}): {}", guid);
    }
}

Example 89 with AtlasEdge

use of org.apache.atlas.repository.graphdb.AtlasEdge in project atlas by apache.

the class AtlasRelationshipStoreV1 method update.

@Override
@GraphTransaction
public AtlasRelationship update(AtlasRelationship relationship) throws AtlasBaseException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("==> update({})", relationship);
    }
    String guid = relationship.getGuid();
    if (StringUtils.isEmpty(guid)) {
        throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_GUID_NOT_FOUND, guid);
    }
    AtlasEdge edge = graphHelper.getEdgeForGUID(guid);
    String edgeType = AtlasGraphUtilsV1.getTypeName(edge);
    AtlasVertex end1Vertex = edge.getOutVertex();
    AtlasVertex end2Vertex = edge.getInVertex();
    // update shouldn't change endType
    if (StringUtils.isNotEmpty(relationship.getTypeName()) && !StringUtils.equalsIgnoreCase(edgeType, relationship.getTypeName())) {
        throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_UPDATE_TYPE_CHANGE_NOT_ALLOWED, guid, edgeType, relationship.getTypeName());
    }
    // update shouldn't change ends
    if (relationship.getEnd1() != null) {
        String updatedEnd1Guid = relationship.getEnd1().getGuid();
        if (updatedEnd1Guid == null) {
            AtlasVertex updatedEnd1Vertex = getVertexFromEndPoint(relationship.getEnd1());
            updatedEnd1Guid = updatedEnd1Vertex == null ? null : AtlasGraphUtilsV1.getIdFromVertex(updatedEnd1Vertex);
        }
        if (updatedEnd1Guid != null) {
            String end1Guid = AtlasGraphUtilsV1.getIdFromVertex(end1Vertex);
            if (!StringUtils.equalsIgnoreCase(relationship.getEnd1().getGuid(), end1Guid)) {
                throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_UPDATE_END_CHANGE_NOT_ALLOWED, edgeType, guid, end1Guid, relationship.getEnd1().getGuid());
            }
        }
    }
    // update shouldn't change ends
    if (relationship.getEnd2() != null) {
        String updatedEnd2Guid = relationship.getEnd2().getGuid();
        if (updatedEnd2Guid == null) {
            AtlasVertex updatedEnd2Vertex = getVertexFromEndPoint(relationship.getEnd2());
            updatedEnd2Guid = updatedEnd2Vertex == null ? null : AtlasGraphUtilsV1.getIdFromVertex(updatedEnd2Vertex);
        }
        if (updatedEnd2Guid != null) {
            String end2Guid = AtlasGraphUtilsV1.getIdFromVertex(end2Vertex);
            if (!StringUtils.equalsIgnoreCase(relationship.getEnd2().getGuid(), end2Guid)) {
                throw new AtlasBaseException(AtlasErrorCode.RELATIONSHIP_UPDATE_END_CHANGE_NOT_ALLOWED, AtlasGraphUtilsV1.getTypeName(edge), guid, end2Guid, relationship.getEnd2().getGuid());
            }
        }
    }
    validateRelationship(end1Vertex, end2Vertex, edgeType, relationship.getAttributes());
    AtlasRelationship ret = updateRelationship(edge, relationship);
    if (LOG.isDebugEnabled()) {
        LOG.debug("<== update({}): {}", relationship, ret);
    }
    return ret;
}

Example 90 with AtlasEdge

use of org.apache.atlas.repository.graphdb.AtlasEdge in project atlas by apache.

the class AtlasRelationshipStoreV1 method createRelationshipEdge.

private AtlasEdge createRelationshipEdge(AtlasVertex fromVertex, AtlasVertex toVertex, AtlasRelationship relationship) throws RepositoryException, AtlasBaseException {
    String relationshipLabel = getRelationshipEdgeLabel(fromVertex, toVertex, relationship.getTypeName());
    PropagateTags tagPropagation = getRelationshipTagPropagation(fromVertex, toVertex, relationship);
    AtlasEdge ret = graphHelper.getOrCreateEdge(fromVertex, toVertex, relationshipLabel);
    if (LOG.isDebugEnabled()) {
        LOG.debug("Created relationship edge from [{}] --> [{}] using edge label: [{}]", getTypeName(fromVertex), getTypeName(toVertex), relationshipLabel);
    }
    // map additional properties to relationship edge
    if (ret != null) {
        final String guid = UUID.randomUUID().toString();
        AtlasGraphUtilsV1.setProperty(ret, Constants.ENTITY_TYPE_PROPERTY_KEY, relationship.getTypeName());
        AtlasGraphUtilsV1.setProperty(ret, Constants.RELATIONSHIP_GUID_PROPERTY_KEY, guid);
        AtlasGraphUtilsV1.setProperty(ret, Constants.VERSION_PROPERTY_KEY, getRelationshipVersion(relationship));
        AtlasGraphUtilsV1.setProperty(ret, Constants.RELATIONSHIPTYPE_TAG_PROPAGATION_KEY, tagPropagation.name());
        // propagate tags
        entityRetriever.addTagPropagation(ret, tagPropagation);
    }
    return ret;
}