How to use

getEntities

method

in

org.apache.atlas.model.discovery.AtlasSearchResult

private List<ExportImportAuditEntry> toExportImportAuditEntry(AtlasSearchResult result) {
  List<ExportImportAuditEntry> ret = new ArrayList<>();
  if(CollectionUtils.isEmpty(result.getEntities())) {
    return ret;
  }
  for (AtlasEntityHeader entityHeader : result.getEntities()) {
    ExportImportAuditEntry entry = ExportImportAuditEntryDTO.from(entityHeader.getGuid(),
                                    entityHeader.getAttributes());
    if(entry == null) {
      continue;
    }
    ret.add(entry);
  }
  return ret;
}

private void search() throws Exception {
  System.out.println("\nSample DSL Queries: ");
  for (String dslQuery : getDSLQueries()) {
    try {
      AtlasSearchResult results = atlasClientV2.dslSearchWithParams(dslQuery, 10, 0);
      if (results != null) {
        List<AtlasEntityHeader> entitiesResult = results.getEntities();
        List<AtlasFullTextResult> fullTextResults = results.getFullTextResult();
        AttributeSearchResult attribResult = results.getAttributes();
        if (CollectionUtils.isNotEmpty(entitiesResult)) {
          System.out.println("query [" + dslQuery + "] returned [" + entitiesResult.size() + "] rows.");
        } else if (CollectionUtils.isNotEmpty(fullTextResults)) {
          System.out.println("query [" + dslQuery + "] returned [" + fullTextResults.size() + "] rows.");
        } else if (attribResult != null) {
          System.out.println("query [" + dslQuery + "] returned [" + attribResult.getValues().size() + "] rows.");
        } else {
          System.out.println("query [" + dslQuery + "] returned [ 0 ] rows.");
        }
      } else {
        System.out.println("query [" + dslQuery + "] failed, results:" + results);
      }
    } catch (Exception e) {
      System.out.println("query [" + dslQuery + "] execution failed!");
    }
  }
}

private List<AtlasEntityHeader> getImportedEntities(String query, String guid) throws AtlasBaseException {
  String q = StringUtils.isEmpty(guid) ? query : String.format("%s where __guid = '%s'", query, guid);
  return discoveryService.searchUsingDslQuery(q, DEFAULT_LIMIT, 0).getEntities();
}

@Override
public void scrubSearchResults(AtlasSearchResultScrubRequest request) throws AtlasAuthorizationException {
  if (LOG.isDebugEnabled()) {
    LOG.debug("==> SimpleAtlasAuthorizer.scrubSearchResults({})", request);
  }
  final AtlasSearchResult result = request.getSearchResult();
  if (CollectionUtils.isNotEmpty(result.getEntities())) {
    for (AtlasEntityHeader entity : result.getEntities()) {
      checkAccessAndScrub(entity, request);
    }
  }
  if (CollectionUtils.isNotEmpty(result.getFullTextResult())) {
    for (AtlasFullTextResult fullTextResult : result.getFullTextResult()) {
      if (fullTextResult != null) {
        checkAccessAndScrub(fullTextResult.getEntity(), request);
      }
    }
  }
  if (MapUtils.isNotEmpty(result.getReferredEntities())) {
    for (AtlasEntityHeader entity : result.getReferredEntities().values()) {
      checkAccessAndScrub(entity, request);
    }
  }
  if (LOG.isDebugEnabled()) {
    LOG.debug("<== SimpleAtlasAuthorizer.scrubSearchResults({}): {}", request, result);
  }
}

@Override
public void scrubSearchResults(AtlasSearchResultScrubRequest request) throws AtlasAuthorizationException {
  if (LOG.isDebugEnabled()) {
    LOG.debug("==> SimpleAtlasAuthorizer.scrubSearchResults({})", request);
  }
  final AtlasSearchResult result = request.getSearchResult();
  if (CollectionUtils.isNotEmpty(result.getEntities())) {
    for (AtlasEntityHeader entity : result.getEntities()) {
      checkAccessAndScrub(entity, request);
    }
  }
  if (CollectionUtils.isNotEmpty(result.getFullTextResult())) {
    for (AtlasFullTextResult fullTextResult : result.getFullTextResult()) {
      if (fullTextResult != null) {
        checkAccessAndScrub(fullTextResult.getEntity(), request);
      }
    }
  }
  if (MapUtils.isNotEmpty(result.getReferredEntities())) {
    for (AtlasEntityHeader entity : result.getReferredEntities().values()) {
      checkAccessAndScrub(entity, request);
    }
  }
  if (LOG.isDebugEnabled()) {
    LOG.debug("<== SimpleAtlasAuthorizer.scrubSearchResults({}): {}", request, result);
  }
}

private void assertSearchResult(AtlasSearchResult searchResult, int expected, String query) {
  assertNotNull(searchResult);
  if(expected == 0) {
    assertTrue(searchResult.getAttributes() == null || CollectionUtils.isEmpty(searchResult.getAttributes().getValues()));
    assertNull(searchResult.getEntities(), query);
  } else if(searchResult.getEntities() != null) {
    assertEquals(searchResult.getEntities().size(), expected, query);
  } else {
    assertNotNull(searchResult.getAttributes());
    assertNotNull(searchResult.getAttributes().getValues());
    assertEquals(searchResult.getAttributes().getValues().size(), expected, query);
  }
}

private void assertSearchResult(AtlasSearchResult searchResult, int expected, String query) {
  assertNotNull(searchResult);
  if(expected == 0) {
    assertTrue(searchResult.getAttributes() == null || CollectionUtils.isEmpty(searchResult.getAttributes().getValues()));
    assertNull(searchResult.getEntities(), query);
  } else if(searchResult.getEntities() != null) {
    assertEquals(searchResult.getEntities().size(), expected, query);
  } else {
    assertNotNull(searchResult.getAttributes());
    assertNotNull(searchResult.getAttributes().getValues());
    assertEquals(searchResult.getAttributes().getValues().size(), expected, query);
  }
}

private void assertSearchResult(AtlasSearchResult searchResult, FieldValueValidator expected) {
  assertNotNull(searchResult);
  assertNull(searchResult.getEntities());
  assertEquals(searchResult.getAttributes().getName().size(), expected.getFieldNamesCount());
  for (int i = 0; i < searchResult.getAttributes().getName().size(); i++) {
    String s = searchResult.getAttributes().getName().get(i);
    assertEquals(s, expected.fieldNames[i]);
  }
  assertEquals(searchResult.getAttributes().getValues().size(), expected.values.size());
}

if (result.getEntities() == null || result.getEntities().isEmpty()) {
  LOG.warn("DSL {} returned no entities", query);
  success = false;
} else if (result.getEntities().size() != expected) {
  LOG.warn("DSL {} returned unexpected number of entities. Expected {} Actual {}", query, expected, result.getEntities().size());
  success = false;
} else {

if (result.getEntities() == null || result.getEntities().isEmpty()) {
  LOG.warn("DSL {} returned no entities", query);
  success = false;
} else if (result.getEntities().size() != expected) {
  LOG.warn("DSL {} returned unexpected number of entities. Expected {} Actual {}", query, expected, result.getEntities().size());
  success = false;
} else {

private void assertSearchResult(AtlasSearchResult searchResult, FieldValueValidator expected) {
  assertNotNull(searchResult);
  assertNull(searchResult.getEntities());
  assertEquals(searchResult.getAttributes().getName().size(), expected.getFieldNamesCount());
  for (int i = 0; i < searchResult.getAttributes().getName().size(); i++) {
    String s = searchResult.getAttributes().getName().get(i);
    assertEquals(s, expected.fieldNames[i]);
  }
  assertEquals(searchResult.getAttributes().getValues().size(), expected.values.size());
}

dslResult.setCount(0);
if (CollectionUtils.isNotEmpty(result.getEntities())) {
  for (AtlasEntityHeader entityHeader : result.getEntities()) {
    Referenceable entity = getEntity(entityHeader.getGuid());

  if (ret.getEntities().size() == resultSize) {
    break;
if (ret.getEntities() != null && ret.getEntities().size() == resultSize) {
  break;

if (ret.getEntities() == null) {
  ret.setEntities(new ArrayList<>());