Search in sources :

Example 6 with ValidationProcessor

use of uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor in project Gaffer by gchq.

the class QueryScannerTest method shouldConstructProcessorsWithNoAggregation.

@Test
public void shouldConstructProcessorsWithNoAggregation() throws OperationException, IOException {
    // Given
    final Scan scan = mock(Scan.class);
    given(scan.getAttribute(HBaseStoreConstants.VIEW)).willReturn(VIEW.toCompactJson());
    given(scan.getAttribute(HBaseStoreConstants.EXTRA_PROCESSORS)).willReturn(StringUtil.toCsv(ElementDedupeFilterProcessor.class));
    // When
    final List<GafferScannerProcessor> processors = QueryScanner.createProcessors(scan, SCHEMA_NO_AGGREGATION, serialisation);
    // Then
    assertThat(processors).hasSize(5);
    int i = 0;
    assertTrue(processors.get(i) instanceof GroupFilterProcessor);
    assertEquals(VIEW, ((GroupFilterProcessor) processors.get(i)).getView());
    i++;
    assertTrue(processors.get(i) instanceof ElementDedupeFilterProcessor);
    assertTrue(((ElementDedupeFilterProcessor) processors.get(i)).isEntities());
    assertTrue(((ElementDedupeFilterProcessor) processors.get(i)).isEdges());
    assertFalse(((ElementDedupeFilterProcessor) processors.get(i)).isDirectedEdges());
    assertFalse(((ElementDedupeFilterProcessor) processors.get(i)).isUnDirectedEdges());
    i++;
    assertTrue(processors.get(i) instanceof ValidationProcessor);
    assertEquals(SCHEMA_NO_AGGREGATION, ((ValidationProcessor) processors.get(i)).getSchema());
    i++;
    assertTrue(processors.get(i) instanceof PreAggregationFilterProcessor);
    assertEquals(VIEW, ((PreAggregationFilterProcessor) processors.get(i)).getView());
    i++;
    assertTrue(processors.get(i) instanceof PostAggregationFilterProcessor);
    assertEquals(VIEW, ((PostAggregationFilterProcessor) processors.get(i)).getView());
}
Also used : PostAggregationFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.PostAggregationFilterProcessor) GafferScannerProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GafferScannerProcessor) GroupFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GroupFilterProcessor) ElementDedupeFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ElementDedupeFilterProcessor) Scan(org.apache.hadoop.hbase.client.Scan) ValidationProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor) PreAggregationFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.PreAggregationFilterProcessor) Test(org.junit.jupiter.api.Test)

Example 7 with ValidationProcessor

use of uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor in project Gaffer by gchq.

the class QueryScanner method createProcessors.

protected static List<GafferScannerProcessor> createProcessors(final Scan scan, final Schema schema, final ElementSerialisation serialisation) {
    final List<GafferScannerProcessor> processors = new ArrayList<>();
    final Set<Class<? extends GafferScannerProcessor>> extraProcessors = getExtraProcessors(scan);
    // The view will be null if a scan of the table is done in the hbase shell
    final View view = getView(scan);
    if (null != view) {
        processors.add(new GroupFilterProcessor(view));
        if (extraProcessors.remove(ElementDedupeFilterProcessor.class)) {
            processors.add(new ElementDedupeFilterProcessor(view.hasEntities(), view.hasEdges(), getDirectedType(scan)));
        }
    }
    if (schema.isAggregationEnabled()) {
        processors.add(new StoreAggregationProcessor(serialisation, schema));
    }
    processors.add(new ValidationProcessor(schema));
    if (null != view) {
        processors.add(new PreAggregationFilterProcessor(view));
        if (schema.isAggregationEnabled()) {
            processors.add(new QueryAggregationProcessor(serialisation, schema, view));
        }
        processors.add(new PostAggregationFilterProcessor(view));
    }
    if (!extraProcessors.isEmpty()) {
        throw new RuntimeException("Unrecognised extra processors: " + extraProcessors);
    }
    return processors;
}
Also used : ArrayList(java.util.ArrayList) GroupFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GroupFilterProcessor) PreAggregationFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.PreAggregationFilterProcessor) View(uk.gov.gchq.gaffer.data.elementdefinition.view.View) StoreAggregationProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.StoreAggregationProcessor) PostAggregationFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.PostAggregationFilterProcessor) GafferScannerProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GafferScannerProcessor) ElementDedupeFilterProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ElementDedupeFilterProcessor) ValidationProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor) QueryAggregationProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.QueryAggregationProcessor)

Example 8 with ValidationProcessor

use of uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor in project Gaffer by gchq.

the class StoreScanner method createProcessors.

protected static List<GafferScannerProcessor> createProcessors(final Schema schema, final ElementSerialisation serialisation) {
    final List<GafferScannerProcessor> processors = new ArrayList<>();
    if (schema.isAggregationEnabled()) {
        processors.add(new StoreAggregationProcessor(serialisation, schema));
    }
    processors.add(new ValidationProcessor(schema));
    return processors;
}
Also used : StoreAggregationProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.StoreAggregationProcessor) GafferScannerProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GafferScannerProcessor) ArrayList(java.util.ArrayList) ValidationProcessor(uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor)

Aggregations

GafferScannerProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GafferScannerProcessor)8 ValidationProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ValidationProcessor)8 Scan (org.apache.hadoop.hbase.client.Scan)6 Test (org.junit.jupiter.api.Test)6 StoreAggregationProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.StoreAggregationProcessor)6 ElementDedupeFilterProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.ElementDedupeFilterProcessor)5 GroupFilterProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.GroupFilterProcessor)4 PostAggregationFilterProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.PostAggregationFilterProcessor)4 PreAggregationFilterProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.PreAggregationFilterProcessor)4 QueryAggregationProcessor (uk.gov.gchq.gaffer.hbasestore.coprocessor.processor.QueryAggregationProcessor)3 ArrayList (java.util.ArrayList)2 View (uk.gov.gchq.gaffer.data.elementdefinition.view.View)1