Examples with VertexWrapper org.apache.atlas.catalog.VertexWrapper used on opensource projects

Example 1 with VertexWrapper

use of org.apache.atlas.catalog.VertexWrapper in project incubator-atlas by apache.

the class TraitRelation method traverse.

@Override
public Collection<RelationSet> traverse(VertexWrapper vWrapper) {
    Vertex v = vWrapper.getVertex();
    Collection<VertexWrapper> vertices = new ArrayList<>();
    for (Edge e : v.getEdges(Direction.OUT)) {
        if (e.getLabel().startsWith(v.<String>getProperty(Constants.ENTITY_TYPE_PROPERTY_KEY))) {
            VertexWrapper trait = new TermVertexWrapper(e.getVertex(Direction.IN));
            if (!trait.getPropertyKeys().contains("available_as_tag") && !isDeleted(trait.getVertex())) {
                vertices.add(trait);
            }
        }
    }
    return Collections.singletonList(new RelationSet("traits", vertices));
}

Example 2 with VertexWrapper

use of org.apache.atlas.catalog.VertexWrapper in project incubator-atlas by apache.

the class AtlasEntityTagQuery method getQueryPipe.

@Override
protected Pipe getQueryPipe() {
    GremlinPipeline p;
    if (guid.equals("*")) {
        p = new GremlinPipeline().has(Constants.ENTITY_TEXT_PROPERTY_KEY).hasNot(Constants.ENTITY_TYPE_PROPERTY_KEY, "Taxonomy").outE();
    } else {
        p = new GremlinPipeline().has(Constants.GUID_PROPERTY_KEY, guid).outE();
    }
    //todo: this is basically the same pipeline used in TagRelation.asPipe()
    p.add(new FilterFunctionPipe<>(new PipeFunction<Edge, Boolean>() {

        @Override
        public Boolean compute(Edge edge) {
            String type = edge.getVertex(Direction.OUT).getProperty(Constants.ENTITY_TYPE_PROPERTY_KEY);
            VertexWrapper v = new TermVertexWrapper(edge.getVertex(Direction.IN));
            return edge.getLabel().startsWith(type) && v.getPropertyKeys().contains("available_as_tag");
        }
    }));
    return p.inV();
}

Example 3 with VertexWrapper

use of org.apache.atlas.catalog.VertexWrapper in project incubator-atlas by apache.

the class TermResourceDefinition method getHierarchyProjection.

private Projection getHierarchyProjection() {
    final String projectionName = "hierarchy";
    return new Projection(projectionName, Projection.Cardinality.SINGLE, new TransformFunctionPipe<>(new PipeFunction<VertexWrapper, Collection<ProjectionResult>>() {

        @Override
        public Collection<ProjectionResult> compute(VertexWrapper start) {
            Map<String, Object> map = new TreeMap<>(new ResourceComparator());
            TermPath termPath = new TermPath(start.getVertex().<String>getProperty(Constants.ENTITY_TYPE_PROPERTY_KEY));
            map.put("path", termPath.getPath());
            map.put("short_name", termPath.getShortName());
            map.put("taxonomy", termPath.getTaxonomyName());
            return Collections.singleton(new ProjectionResult(projectionName, start, Collections.singleton(map)));
        }
    }));
}

Example 4 with VertexWrapper

use of org.apache.atlas.catalog.VertexWrapper in project incubator-atlas by apache.

the class TermResourceDefinition method getSubTermProjection.

private Projection getSubTermProjection() {
    //todo: combine with other term projections
    final String termsProjectionName = "terms";
    return new Projection(termsProjectionName, Projection.Cardinality.SINGLE, new TransformFunctionPipe<>(new PipeFunction<VertexWrapper, Collection<ProjectionResult>>() {

        @Override
        public Collection<ProjectionResult> compute(VertexWrapper start) {
            Map<String, Object> map = new TreeMap<>(new ResourceComparator());
            StringBuilder sb = new StringBuilder();
            sb.append("v1/taxonomies/");
            TermPath termPath = new TermPath(start.getVertex().<String>getProperty(Constants.ENTITY_TYPE_PROPERTY_KEY));
            String[] paths = termPath.getPathSegments();
            sb.append(termPath.getTaxonomyName());
            for (String path : paths) {
                //todo: shouldn't need to check for null or empty after TermPath addition
                if (path != null && !path.isEmpty()) {
                    sb.append("/terms/");
                    sb.append(path);
                }
            }
            sb.append("/terms");
            map.put("href", sb.toString());
            return Collections.singleton(new ProjectionResult(termsProjectionName, start, Collections.singleton(map)));
        }
    }));
}

Example 5 with VertexWrapper

use of org.apache.atlas.catalog.VertexWrapper in project incubator-atlas by apache.

the class AlwaysQueryExpressionTest method testGetProperties.

@Test
public void testGetProperties() {
    VertexWrapper v = createStrictMock(VertexWrapper.class);
    replay(v);
    QueryExpression expression = new AlwaysQueryExpression();
    assertTrue(expression.getProperties().isEmpty());
    verify(v);
}