Examples with ElementId uk.gov.gchq.gaffer.data.element.id.ElementId used on opensource projects

Example 31 with ElementId

use of uk.gov.gchq.gaffer.data.element.id.ElementId in project Gaffer by gchq.

the class GetElementsIT method getAllSeededVertices.

private static Collection<Object> getAllSeededVertices() {
    Set<Object> allSeededVertices = new HashSet<>();
    for (final ElementId elementId : ENTITY_SEEDS_EXIST) {
        allSeededVertices.add(((EntityId) elementId).getVertex());
    }
    for (final ElementId elementId : EDGE_SEEDS_EXIST) {
        allSeededVertices.add(((EdgeId) elementId).getSource());
        allSeededVertices.add(((EdgeId) elementId).getDestination());
    }
    for (final ElementId elementId : EDGE_DIR_SEEDS_EXIST) {
        allSeededVertices.add(((EdgeId) elementId).getSource());
        allSeededVertices.add(((EdgeId) elementId).getDestination());
    }
    return allSeededVertices;
}

Example 32 with ElementId

use of uk.gov.gchq.gaffer.data.element.id.ElementId in project Gaffer by gchq.

the class FilteringIT method testFilteringProperties.

@Test
@TraitRequirement(StoreTrait.PRE_AGGREGATION_FILTERING)
public void testFilteringProperties() throws OperationException {
    // Given
    final List<ElementId> seeds = Arrays.asList(new EntitySeed("A3"), new EdgeSeed("A5", "B5", false));
    final GetElements getElementsWithoutFiltering = new GetElements.Builder().input(seeds).build();
    final GetElements getElementsWithFiltering = new GetElements.Builder().input(seeds).view(new View.Builder().entity(TestGroups.ENTITY, new ViewElementDefinition.Builder().preAggregationFilter(new ElementFilter.Builder().select(IdentifierType.VERTEX.name()).execute(new IsEqual("A5")).build()).build()).edge(TestGroups.EDGE, new ViewElementDefinition.Builder().preAggregationFilter(new ElementFilter.Builder().select(TestPropertyNames.INT).execute(new IsLessThan(2)).build()).build()).build()).build();
    // When - without filtering
    final List<Element> resultsWithoutFiltering = Lists.newArrayList(graph.execute(getElementsWithoutFiltering, getUser()));
    // When - with filtering
    final List<Element> resultsWithFiltering = Lists.newArrayList(graph.execute(getElementsWithFiltering, getUser()));
    // Then - without filtering
    List<Element> expectedResults = Arrays.asList(getEdge("A3", "A3", false), getEdge("A3", "B3", false), getEdge("A3", "C3", false), getEdge("A3", "D3", false), getEdge("A5", "B5", false), getEdge("A3", "A3", true), getEdge("A3", "B3", true), getEdge("A3", "C3", true), getEdge("A3", "D3", true), getEntity("A3"), getEntity("A5"), getEntity("B5"));
    ElementUtil.assertElementEquals(expectedResults, resultsWithoutFiltering);
    // Then - with filtering
    List<Element> expectedFilteredResults = Arrays.asList(getEdge("A3", "A3", false), getEdge("A3", "B3", false), getEdge("A3", "D3", false), getEdge("A3", "C3", false), getEdge("A5", "B5", false), getEdge("A3", "A3", true), getEdge("A3", "B3", true), getEdge("A3", "C3", true), getEdge("A3", "D3", true), getEntity("A5"));
    ElementUtil.assertElementEquals(expectedFilteredResults, resultsWithFiltering);
}

Example 33 with ElementId

use of uk.gov.gchq.gaffer.data.element.id.ElementId in project Gaffer by gchq.

the class FilteringIT method testFilteringGroups.

@Test
@TraitRequirement(StoreTrait.PRE_AGGREGATION_FILTERING)
public void testFilteringGroups() throws OperationException {
    // Given
    final List<ElementId> seeds = Collections.singletonList(new EntitySeed("A3"));
    final GetElements getElementsWithoutFiltering = new GetElements.Builder().input(seeds).build();
    final GetElements getElementsWithFiltering = new GetElements.Builder().input(seeds).view(new View.Builder().entity(TestGroups.ENTITY).build()).build();
    // When - without filtering
    final List<Element> resultsWithoutFiltering = Lists.newArrayList(graph.execute(getElementsWithoutFiltering, getUser()));
    // When - with filtering
    final List<Element> resultsWithFiltering = Lists.newArrayList(graph.execute(getElementsWithFiltering, getUser()));
    // Then - without filtering
    assertThat(resultsWithoutFiltering).hasSize(9);
    ElementUtil.assertElementEquals(resultsWithoutFiltering, Arrays.asList(getEdge("A3", "A3", false), getEdge("A3", "B3", false), getEdge("A3", "C3", false), getEdge("A3", "D3", false), getEdge("A3", "A3", true), getEdge("A3", "B3", true), getEdge("A3", "C3", true), getEdge("A3", "D3", true), getEntity("A3")));
    // Then - with filtering
    assertThat(resultsWithFiltering).hasSize(1);
    ElementUtil.assertElementEquals(resultsWithFiltering, Arrays.asList((Element) getEntity("A3")));
}

Example 34 with ElementId

use of uk.gov.gchq.gaffer.data.element.id.ElementId in project Gaffer by gchq.

the class FilteringIT method testPostAggregationFilteringIdentifiers.

@Test
@TraitRequirement({ StoreTrait.POST_AGGREGATION_FILTERING, StoreTrait.INGEST_AGGREGATION })
public void testPostAggregationFilteringIdentifiers() throws OperationException {
    // Given
    final List<ElementId> seeds = Collections.singletonList(new EntitySeed("A3"));
    final GetElements getElementsWithoutFiltering = new GetElements.Builder().input(seeds).build();
    final GetElements getElementsWithFiltering = new GetElements.Builder().input(seeds).view(new View.Builder().entity(TestGroups.ENTITY).edge(TestGroups.EDGE, new ViewElementDefinition.Builder().postAggregationFilter(new ElementFilter.Builder().select(IdentifierType.DESTINATION.name()).execute(new IsEqual("B3")).build()).build()).build()).build();
    // When - without filtering
    final List<Element> resultsWithoutFiltering = Lists.newArrayList(graph.execute(getElementsWithoutFiltering, getUser()));
    // When - with filtering
    final List<Element> resultsWithFiltering = Lists.newArrayList(graph.execute(getElementsWithFiltering, getUser()));
    // Then - without filtering
    assertThat(resultsWithoutFiltering).hasSize(9);
    ElementUtil.assertElementEquals(resultsWithoutFiltering, Arrays.asList(getEdge("A3", "A3", false), getEdge("A3", "B3", false), getEdge("A3", "C3", false), getEdge("A3", "D3", false), getEdge("A3", "A3", true), getEdge("A3", "B3", true), getEdge("A3", "C3", true), getEdge("A3", "D3", true), getEntity("A3")));
    // Then - with filtering
    assertThat(resultsWithFiltering).hasSize(3);
    ElementUtil.assertElementEquals(resultsWithFiltering, Arrays.asList(getEdge("A3", "B3", false), getEdge("A3", "B3", true), getEntity("A3")));
}

Example 35 with ElementId

use of uk.gov.gchq.gaffer.data.element.id.ElementId in project Gaffer by gchq.

the class FilteringIT method testFilteringIdentifiers.

@Test
@TraitRequirement(StoreTrait.PRE_AGGREGATION_FILTERING)
public void testFilteringIdentifiers() throws OperationException {
    // Given
    final List<ElementId> seeds = Collections.singletonList(new EntitySeed("A3"));
    final GetElements getElementsWithoutFiltering = new GetElements.Builder().input(seeds).build();
    final GetElements getElementsWithFiltering = new GetElements.Builder().input(seeds).view(new View.Builder().entity(TestGroups.ENTITY).edge(TestGroups.EDGE, new ViewElementDefinition.Builder().preAggregationFilter(new ElementFilter.Builder().select(IdentifierType.DESTINATION.name()).execute(new IsEqual("B3")).build()).build()).build()).build();
    // When - without filtering
    final List<Element> resultsWithoutFiltering = Lists.newArrayList(graph.execute(getElementsWithoutFiltering, getUser()));
    // When - with filtering
    final List<Element> resultsWithFiltering = Lists.newArrayList(graph.execute(getElementsWithFiltering, getUser()));
    // Then - without filtering
    assertThat(resultsWithoutFiltering).hasSize(9);
    ElementUtil.assertElementEquals(resultsWithoutFiltering, Arrays.asList(getEdge("A3", "A3", false), getEdge("A3", "B3", false), getEdge("A3", "C3", false), getEdge("A3", "D3", false), getEdge("A3", "A3", true), getEdge("A3", "B3", true), getEdge("A3", "C3", true), getEdge("A3", "D3", true), getEntity("A3")));
    // Then - with filtering
    assertThat(resultsWithFiltering).hasSize(3);
    ElementUtil.assertElementEquals(resultsWithFiltering, Arrays.asList(getEdge("A3", "B3", false), getEdge("A3", "B3", true), getEntity("A3")));
}