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<>());