Examples with ExtractProperty uk.gov.gchq.gaffer.data.element.function.ExtractProperty used on opensource projects

Example 6 with ExtractProperty

use of uk.gov.gchq.gaffer.data.element.function.ExtractProperty in project Gaffer by gchq.

the class KeyFunctionMatchTest method shouldJsonSerialiseAndDeserialiseWithKeyFunctions.

@Test
public void shouldJsonSerialiseAndDeserialiseWithKeyFunctions() throws SerialisationException {
    // given
    KeyFunctionMatch match = new KeyFunctionMatch.Builder().firstKeyFunction(new FunctionComposite(Lists.newArrayList(new DivideBy(20), new FirstItem()))).secondKeyFunction(new ExtractProperty("count")).build();
    // when / then
    String expected = "{\n" + "  \"class\" : \"uk.gov.gchq.gaffer.store.operation.handler.join.match.KeyFunctionMatch\",\n" + "  \"firstKeyFunction\" : {\n" + "    \"class\" : \"uk.gov.gchq.koryphe.function.FunctionComposite\",\n" + "    \"functions\" : [ {\n" + "      \"class\" : \"uk.gov.gchq.koryphe.impl.function.DivideBy\",\n" + "      \"by\" : 20\n" + "    }, {\n" + "      \"class\" : \"uk.gov.gchq.koryphe.impl.function.FirstItem\"\n" + "    } ]\n" + "  },\n" + "  \"secondKeyFunction\" : {\n" + "    \"class\" : \"uk.gov.gchq.gaffer.data.element.function.ExtractProperty\",\n" + "    \"name\" : \"count\"\n" + "  }\n" + "}";
    assertEquals(expected, new String(JSONSerialiser.serialise(match, true)));
    assertEquals(match, JSONSerialiser.deserialise(expected, KeyFunctionMatch.class));
}

Example 7 with ExtractProperty

use of uk.gov.gchq.gaffer.data.element.function.ExtractProperty in project Gaffer by gchq.

the class GetWalksIT method shouldNotFilterWalksWhenNoPredicateSupplied.

@Test
public void shouldNotFilterWalksWhenNoPredicateSupplied() throws Exception {
    final Conditional conditional = new Conditional();
    conditional.setTransform(new OperationChain.Builder().first(new Map.Builder<>().first(new ExtractWalkEntities()).then(new IterableConcat()).build()).then(new ForEach.Builder<>().operation(new Map.Builder<>().first(new ExtractProperty(TestPropertyNames.PROP_1)).build()).build()).build());
    final Iterable<Walk> walks = executeGetWalksApplyingConditional(conditional);
    assertThat(getPaths(walks)).isEqualTo("AED,ABC");
}

Example 8 with ExtractProperty

use of uk.gov.gchq.gaffer.data.element.function.ExtractProperty in project Gaffer by gchq.

the class JoinIT method shouldRightSideFullJoinUsingKeyFunctionMatch.

@Test
public void shouldRightSideFullJoinUsingKeyFunctionMatch() throws OperationException {
    // Given
    final Map map = new Map.Builder<>().input(Lists.newArrayList(2L, 1L, 2L, 3L)).first(new Identity()).build();
    final ArrayList<Entity> input = Lists.newArrayList(getJoinEntity(TestGroups.ENTITY, 1), getJoinEntity(TestGroups.ENTITY_2, 2), getJoinEntity(TestGroups.ENTITY_2, 2), getJoinEntity(TestGroups.ENTITY_5, 5));
    final Join<Object> leftJoin = new Join.Builder<>().flatten(false).matchKey(MatchKey.RIGHT).joinType(JoinType.FULL).operation(map).matchMethod(new KeyFunctionMatch(new ToLong(), new ExtractProperty("count"))).input(input).build();
    // When
    final Iterable<? extends MapTuple> results = graph.execute(leftJoin, user);
    final List<java.util.Map> loadedResults = new ArrayList<>();
    results.forEach(e -> loadedResults.add(e.getValues()));
    // Then
    assertThat(loadedResults).hasSize(4);
    assertThat(loadedResults.get(0)).containsEntry(MatchKey.LEFT.name(), Lists.newArrayList(getJoinEntity(TestGroups.ENTITY_2, 2), getJoinEntity(TestGroups.ENTITY_2, 2))).containsEntry(MatchKey.RIGHT.name(), 2L);
    assertThat(loadedResults.get(1)).containsEntry(MatchKey.LEFT.name(), Lists.newArrayList(getJoinEntity(TestGroups.ENTITY, 1))).containsEntry(MatchKey.RIGHT.name(), 1L);
    assertThat(loadedResults.get(2)).containsEntry(MatchKey.LEFT.name(), Lists.newArrayList(getJoinEntity(TestGroups.ENTITY_2, 2), getJoinEntity(TestGroups.ENTITY_2, 2))).containsEntry(MatchKey.RIGHT.name(), 2L);
    assertThat(loadedResults.get(3)).containsEntry(MatchKey.LEFT.name(), Lists.newArrayList()).containsEntry(MatchKey.RIGHT.name(), 3L);
}

Example 9 with ExtractProperty

use of uk.gov.gchq.gaffer.data.element.function.ExtractProperty in project Gaffer by gchq.

the class JoinIT method shouldRightSideInnerJoinUsingKeyFunctionMatch.

@Test
public void shouldRightSideInnerJoinUsingKeyFunctionMatch() throws OperationException {
    // Given
    final Map map = new Map.Builder<>().input(Lists.newArrayList(4L)).first(new Identity()).build();
    final ArrayList<Entity> input = Lists.newArrayList(getJoinEntity(TestGroups.ENTITY, 4), getJoinEntity(TestGroups.ENTITY_2, 4), getJoinEntity(TestGroups.ENTITY_3, 4));
    final Join<Object> rightJoin = new Join.Builder<>().flatten(true).matchKey(MatchKey.RIGHT).joinType(JoinType.INNER).operation(map).matchMethod(new KeyFunctionMatch(new Identity(), new ExtractProperty("count"))).input(input).build();
    // When
    final Iterable<? extends MapTuple> rightResults = graph.execute(rightJoin, user);
    final List<java.util.Map> loadedResults = new ArrayList<>();
    rightResults.forEach(e -> loadedResults.add(e.getValues()));
    // Then
    assertThat(loadedResults).hasSize(3);
    assertThat(loadedResults.get(0)).containsEntry(MatchKey.LEFT.name(), getJoinEntity(TestGroups.ENTITY, 4)).containsEntry(MatchKey.RIGHT.name(), 4L);
    assertThat(loadedResults.get(1)).containsEntry(MatchKey.LEFT.name(), getJoinEntity(TestGroups.ENTITY_2, 4)).containsEntry(MatchKey.RIGHT.name(), 4L);
    assertThat(loadedResults.get(2)).containsEntry(MatchKey.LEFT.name(), getJoinEntity(TestGroups.ENTITY_3, 4)).containsEntry(MatchKey.RIGHT.name(), 4L);
}

Example 10 with ExtractProperty

use of uk.gov.gchq.gaffer.data.element.function.ExtractProperty in project Gaffer by gchq.

the class GetWalksIT method getWalksThatPassPredicateTest.

private Iterable<Walk> getWalksThatPassPredicateTest(final Predicate predicate) throws Exception {
    final Conditional conditional = new Conditional();
    conditional.setTransform(new OperationChain.Builder().first(new Map.Builder<>().first(new ExtractWalkEntities()).then(new IterableConcat()).build()).then(new ForEach.Builder<>().operation(new Map.Builder<>().first(new ExtractProperty(TestPropertyNames.PROP_1)).build()).build()).build());
    conditional.setPredicate(predicate);
    return executeGetWalksApplyingConditional(conditional);
}