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 AttributeSearchResult toAttributesResult(List list, GremlinQuery query) { AttributeSearchResult ret = new AttributeSearchResult(); List<String> names = new ArrayList<>(); List<List<Object>> values = new ArrayList<>(); // extract select attributes from gremlin query Option<SelectExpression> selectExpr = SelectExpressionHelper.extractSelectExpression(query.expr()); if (selectExpr.isDefined()) { List<AliasExpression> aliases = selectExpr.get().toJavaList(); if (CollectionUtils.isNotEmpty(aliases)) { for (AliasExpression alias : aliases) { names.add(alias.alias()); } ret.setName(names); } } for (Object mapObj : list) { Map map = (mapObj instanceof Map ? (Map) mapObj : null); if (MapUtils.isNotEmpty(map)) { for (Object key : map.keySet()) { Object vals = map.get(key); values.add((List<Object>) vals); } ret.setValues(values); } } return ret; }
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;
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()); }
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 AttributeSearchResult toAttributesResult(List results, GremlinQuery query) { AttributeSearchResult ret = new AttributeSearchResult(); List<String> names = (List<String>) results.get(0); List<List<Object>> values = extractValues(results.subList(1, results.size())); ret.setName(names); ret.setValues(values); return ret; }