private static FinalizingFieldAccessPostAggregator buildDecorated( String name, String fieldName, Map<String, AggregatorFactory> aggregators ) { FinalizingFieldAccessPostAggregator ret = new FinalizingFieldAccessPostAggregator(name, fieldName); return ret.decorate(aggregators); } }
@Test(expected = UnsupportedOperationException.class) public void testComputeWithoutFinalizing() { String aggName = "rows"; Aggregator agg = new CountAggregator(); agg.aggregate(); agg.aggregate(); agg.aggregate(); Map<String, Object> metricValues = new HashMap<>(); metricValues.put(aggName, agg.get()); FinalizingFieldAccessPostAggregator postAgg = new FinalizingFieldAccessPostAggregator("final_rows", aggName); Assert.assertEquals(new Long(3L), postAgg.compute(metricValues)); }
@Override public FinalizingFieldAccessPostAggregator decorate(final Map<String, AggregatorFactory> aggregators) { final Comparator<Object> theComparator; final Function<Object, Object> theFinalizer; if (aggregators != null && aggregators.containsKey(fieldName)) { //noinspection unchecked theComparator = aggregators.get(fieldName).getComparator(); } else { //noinspection unchecked theComparator = (Comparator) Comparators.naturalNullsFirst(); } if (aggregators != null && aggregators.containsKey(fieldName)) { theFinalizer = aggregators.get(fieldName)::finalizeComputation; } else { theFinalizer = Function.identity(); } return new FinalizingFieldAccessPostAggregator( name, fieldName, theComparator, theFinalizer ); }
@Test public void testComparatorsWithFinalizingAndComparatorNull() { String aggName = "billy"; AggregatorFactory aggFactory = EasyMock.createMock(AggregatorFactory.class); EasyMock.expect(aggFactory.getComparator()) .andReturn(null) .times(1); EasyMock.replay(aggFactory); FinalizingFieldAccessPostAggregator postAgg = buildDecorated( "final_billy", "joe", ImmutableMap.of(aggName, aggFactory)); List<Object> computedValues = new ArrayList<>(); Map<String, Object> forNull = new HashMap<>(); forNull.put("joe", null); // guava does not allow the value to be null. computedValues.add(postAgg.compute(ImmutableMap.of("joe", "test_val1"))); computedValues.add(postAgg.compute(ImmutableMap.of("joe", "test_val2"))); computedValues.add(postAgg.compute(forNull)); computedValues.add(postAgg.compute(ImmutableMap.of("joe", "test_val4"))); Collections.sort(computedValues, postAgg.getComparator()); Assert.assertArrayEquals( new Object[]{null, "test_val1", "test_val2", "test_val4"}, computedValues.toArray(new Object[0]) ); EasyMock.verify(); }
@Test public void testComputedWithFinalizing() { String aggName = "biily"; AggregatorFactory aggFactory = EasyMock.createMock(AggregatorFactory.class); EasyMock.expect(aggFactory.getComparator()).andReturn(Comparators.naturalNullsFirst()).once(); EasyMock.expect(aggFactory.finalizeComputation("test")).andReturn(3L).once(); EasyMock.replay(aggFactory); FinalizingFieldAccessPostAggregator postAgg = buildDecorated( "final_billy", aggName, ImmutableMap.of(aggName, aggFactory) ); // Check that the class matches exactly; see https://github.com/apache/incubator-druid/issues/6063 Assert.assertEquals(FinalizingFieldAccessPostAggregator.class, postAgg.getClass()); Map<String, Object> metricValues = new HashMap<>(); metricValues.put(aggName, "test"); Assert.assertEquals(new Long(3L), postAgg.compute(metricValues)); EasyMock.verify(aggFactory); }
@Override public FinalizingFieldAccessPostAggregator decorate(final Map<String, AggregatorFactory> aggregators) { final Comparator<Object> theComparator; final Function<Object, Object> theFinalizer; if (aggregators != null && aggregators.containsKey(fieldName)) { //noinspection unchecked theComparator = aggregators.get(fieldName).getComparator(); } else { //noinspection unchecked theComparator = (Comparator) Comparators.naturalNullsFirst(); } if (aggregators != null && aggregators.containsKey(fieldName)) { theFinalizer = aggregators.get(fieldName)::finalizeComputation; } else { theFinalizer = Function.identity(); } return new FinalizingFieldAccessPostAggregator( name, fieldName, theComparator, theFinalizer ); }
computedValues.add(postAgg.compute(ImmutableMap.of(aggName, "test_val1"))); computedValues.add(postAgg.compute(ImmutableMap.of(aggName, "test_val2"))); computedValues.add(postAgg.compute(ImmutableMap.of(aggName, "test_val3"))); computedValues.add(postAgg.compute(ImmutableMap.of(aggName, "test_val4"))); Collections.sort(computedValues, postAgg.getComparator()); Assert.assertArrayEquals(new Object[]{3L, 10L, 21L, null}, computedValues.toArray(new Object[0])); EasyMock.verify();
@Test public void testSerde() throws IOException { final FinalizingFieldAccessPostAggregator original = new FinalizingFieldAccessPostAggregator("foo", "bar"); final FinalizingFieldAccessPostAggregator decorated = original.decorate( ImmutableMap.of("bar", new CountAggregatorFactory("bar")) ); final ObjectMapper objectMapper = TestHelper.makeJsonMapper(); Assert.assertEquals( original, objectMapper.readValue(objectMapper.writeValueAsString(decorated), PostAggregator.class) ); }