@Override public int compare(EventHolder o1, EventHolder o2) { int retVal = Longs.compare(o1.getTimestamp().getMillis(), o2.getTimestamp().getMillis()); if (retVal == 0) { retVal = o1.getSegmentId().compareTo(o2.getSegmentId()); } if (retVal == 0) { retVal = Integer.compare(o1.getOffset(), o2.getOffset()); } return retVal; } };
@Override protected List<EventHolder> getEventHolders() { final List<EventHolder> values = Lists.newArrayListWithCapacity(pQueue.size()); while (!pQueue.isEmpty()) { EventHolder event = pQueue.remove(); pagingIdentifiers.put(event.getSegmentId(), event.getOffset()); values.add(event); } return values; } }
private Iterable<Result<SelectResultValue>> makeSelectResults( Set<String> dimensions, Set<String> metrics, Object... objects ) { List<Result<SelectResultValue>> retVal = new ArrayList<>(); int index = 0; while (index < objects.length) { DateTime timestamp = (DateTime) objects[index++]; List<EventHolder> values = new ArrayList<>(); while (index < objects.length && !(objects[index] instanceof DateTime)) { values.add(new EventHolder(null, 0, (Map) objects[index++])); } Result<SelectResultValue> result = new Result<>( timestamp, new SelectResultValue(ImmutableMap.of(timestamp.toString(), 0), dimensions, metrics, values) ); retVal.add(result); } return retVal; }
EventHolder acHolder = actualEvts.next(); Assert.assertEquals(exHolder.getTimestamp(), acHolder.getTimestamp()); Assert.assertEquals(exHolder.getOffset(), acHolder.getOffset()); for (Map.Entry<String, Object> ex : exHolder.getEvent().entrySet()) { Object actVal = acHolder.getEvent().get(ex.getKey()); if (acHolder.getEvent().get(ex.getKey()) instanceof Double) { actVal = ((Double) actVal).floatValue();
int end = Math.max(-1, start - threshold); for (int i = start; i > end; i--) { holders.add(new EventHolder(segmentId, newOffset--, group.get(i))); holders.add(new EventHolder(segmentId, newOffset++, group.get(i))); int lastOffset = holders.isEmpty() ? offset : holders.get(holders.size() - 1).getOffset(); expected.add( new Result(
@Test public void testSimpleDataIngestAndSelectQuery() throws Exception { SketchModule sm = new SketchModule(); sm.configure(null); AggregationTestHelper selectQueryAggregationTestHelper = AggregationTestHelper.createSelectQueryAggregationTestHelper( sm.getJacksonModules(), tempFolder ); Sequence seq = selectQueryAggregationTestHelper.runQueryOnSegments( ImmutableList.of(s1, s2), readFileFromClasspathAsString("select_query.json") ); Result<SelectResultValue> result = (Result<SelectResultValue>) Iterables.getOnlyElement(seq.toList()); Assert.assertEquals(DateTimes.of("2014-10-20T00:00:00.000Z"), result.getTimestamp()); Assert.assertEquals(100, result.getValue().getEvents().size()); Assert.assertEquals("AgMDAAAazJMCAAAAAACAPzz9j7pWTMdROWGf15uY1nI=", result.getValue().getEvents().get(0).getEvent().get("pty_country")); }
.build() ), new EventHolder( segmentIdInThisQuery, 1, .build() ), new EventHolder( segmentIdInThisQuery, 2, .build() ), new EventHolder( segmentIdInThisQuery, -2, .build() ), new EventHolder( segmentIdInThisQuery, -3,
@Override protected List<EventHolder> getEventHolders() { final List<EventHolder> values = Lists.newArrayListWithCapacity(pQueue.size()); while (!pQueue.isEmpty()) { EventHolder event = pQueue.remove(); pagingIdentifiers.put(event.getSegmentId(), event.getOffset()); values.add(event); } return values; } }
.build() ), new EventHolder( segmentIdString, 1, .build() ), new EventHolder( segmentIdString, 2, .build() ), new EventHolder( segmentIdString, -2, .build() ), new EventHolder( segmentIdString, -3,
@Override public int compare(EventHolder o1, EventHolder o2) { int retVal = Longs.compare(o1.getTimestamp().getMillis(), o2.getTimestamp().getMillis()); if (retVal == 0) { retVal = o1.getSegmentId().compareTo(o2.getSegmentId()); } if (retVal == 0) { retVal = Integer.compare(o1.getOffset(), o2.getOffset()); } return retVal; } };
.build() ), new EventHolder( segmentIdString, 1, .build() ), new EventHolder( segmentIdString, 2, .build() ), new EventHolder( segmentIdString, -2, .build() ), new EventHolder( segmentIdString, -3,
Sets.newHashSet("sixth"), Arrays.asList( new EventHolder( segmentId1, 0, new EventHolder( segmentId1, 1, new EventHolder( segmentId1, 2, Sets.newHashSet("fifth"), Arrays.asList( new EventHolder( segmentId2, 0, new EventHolder( segmentId2, 1, new EventHolder( segmentId2, 2, new EventHolder(
Sets.newHashSet("eight", "nineth"), Collections.singletonList( new EventHolder( segmentId1, 0, Sets.newHashSet("seventh"), Collections.singletonList( new EventHolder( segmentId2, 0,
new EventHolder( segmentId, lastOffset = offset.current(),
new EventHolder( segmentId, lastOffset = offset.current(),