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 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()); }
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()); }
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); } }
dslResult.setCount(dslResult.getResults().size()); } else if (result.getAttributes() != null && CollectionUtils.isNotEmpty(result.getAttributes().getName())) { List<String> attrNames = result.getAttributes().getName(); for (List<Object> attrValues : result.getAttributes().getValues()) { if (attrValues == null) { continue;