Examples with AtlasFullTextResult org.apache.atlas.model.discovery.AtlasSearchResult.AtlasFullTextResult used on opensource projects

Example 6 with AtlasFullTextResult

use of org.apache.atlas.model.discovery.AtlasSearchResult.AtlasFullTextResult in project incubator-atlas by apache.

the class EntityDiscoveryService method getIndexQueryResults.

private List<AtlasFullTextResult> getIndexQueryResults(AtlasIndexQuery query, QueryParams params, boolean excludeDeletedEntities) throws AtlasBaseException {
    List<AtlasFullTextResult> ret = new ArrayList<>();
    Iterator<Result> iter = query.vertices();
    while (iter.hasNext() && ret.size() < params.limit()) {
        Result idxQueryResult = iter.next();
        AtlasVertex vertex = idxQueryResult.getVertex();
        if (skipDeletedEntities(excludeDeletedEntities, vertex)) {
            continue;
        }
        String guid = vertex != null ? vertex.getProperty(Constants.GUID_PROPERTY_KEY, String.class) : null;
        if (guid != null) {
            AtlasEntityHeader entity = entityRetriever.toAtlasEntityHeader(vertex);
            Double score = idxQueryResult.getScore();
            ret.add(new AtlasFullTextResult(entity, score));
        }
    }
    return ret;
}