From 61569f62be5cbe798c2411bbc35e45f08e962e4b Mon Sep 17 00:00:00 2001
From: Gian Merlino <gianmerlino@gmail.com>
Date: Tue, 29 Jan 2013 12:53:52 -0800
Subject: [PATCH 1/2] TimeseriesQueryRunner tests with small incremental
 indexes

---
 .../TimeseriesQueryRunnerBonusTest.java       | 105 ++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerBonusTest.java

diff --git a/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerBonusTest.java b/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerBonusTest.java
new file mode 100644
index 000000000000..91865f47c7dc
--- /dev/null
+++ b/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerBonusTest.java
@@ -0,0 +1,105 @@
+package com.metamx.druid.query.timeseries;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Lists;
+import com.metamx.common.guava.Sequences;
+import com.metamx.druid.Druids;
+import com.metamx.druid.Query;
+import com.metamx.druid.QueryGranularity;
+import com.metamx.druid.aggregation.AggregatorFactory;
+import com.metamx.druid.aggregation.CountAggregatorFactory;
+import com.metamx.druid.index.IncrementalIndexSegment;
+import com.metamx.druid.index.Segment;
+import com.metamx.druid.index.v1.IncrementalIndex;
+import com.metamx.druid.input.MapBasedInputRow;
+import com.metamx.druid.query.FinalizeResultsQueryRunner;
+import com.metamx.druid.query.QueryRunner;
+import com.metamx.druid.query.QueryRunnerFactory;
+import com.metamx.druid.result.Result;
+import com.metamx.druid.result.TimeseriesResultValue;
+import junit.framework.Assert;
+import org.joda.time.DateTime;
+import org.joda.time.Interval;
+import org.junit.Test;
+
+import java.util.List;
+
+public class TimeseriesQueryRunnerBonusTest
+{
+  @Test
+  public void testOneRowAtATime() throws Exception
+  {
+    final IncrementalIndex oneRowIndex = new IncrementalIndex(
+        new DateTime("2012-01-01T00:00:00Z").getMillis(), QueryGranularity.NONE, new AggregatorFactory[]{}
+    );
+
+    List<Result<TimeseriesResultValue>> results;
+
+    oneRowIndex.add(
+        new MapBasedInputRow(
+            new DateTime("2012-01-01T00:00:00Z").getMillis(),
+            ImmutableList.of("dim1"),
+            ImmutableMap.<String, Object>of("dim1", "x")
+        )
+    );
+
+    results = runTimeseriesCount(oneRowIndex);
+
+    Assert.assertEquals("index size", 1, oneRowIndex.size());
+    Assert.assertEquals("result size", 1, results.size());
+    Assert.assertEquals("result timestamp", new DateTime("2012-01-01T00:00:00Z"), results.get(0).getTimestamp());
+    Assert.assertEquals("result count metric", 1, (long) results.get(0).getValue().getLongMetric("rows"));
+
+    oneRowIndex.add(
+        new MapBasedInputRow(
+            new DateTime("2012-01-01T00:00:00Z").getMillis(),
+            ImmutableList.of("dim1"),
+            ImmutableMap.<String, Object>of("dim1", "y")
+        )
+    );
+
+    results = runTimeseriesCount(oneRowIndex);
+
+    Assert.assertEquals("index size", 2, oneRowIndex.size());
+    Assert.assertEquals("result size", 1, results.size());
+    Assert.assertEquals("result timestamp", new DateTime("2012-01-01T00:00:00Z"), results.get(0).getTimestamp());
+    Assert.assertEquals("result count metric", 2, (long) results.get(0).getValue().getLongMetric("rows"));
+  }
+
+  private static List<Result<TimeseriesResultValue>> runTimeseriesCount(IncrementalIndex index)
+  {
+    final QueryRunnerFactory factory = new TimeseriesQueryRunnerFactory();
+    final QueryRunner<Result<TimeseriesResultValue>> runner = makeQueryRunner(
+        factory,
+        new IncrementalIndexSegment(index)
+    );
+
+    TimeseriesQuery query = Druids.newTimeseriesQueryBuilder()
+                                  .dataSource("xxx")
+                                  .granularity(QueryGranularity.ALL)
+                                  .intervals(ImmutableList.of(new Interval("2012-01-01T00:00:00Z/P1D")))
+                                  .aggregators(
+                                      ImmutableList.<AggregatorFactory>of(
+                                          new CountAggregatorFactory("rows")
+                                      )
+                                  )
+                                  .build();
+
+    return Sequences.toList(
+        runner.run(query),
+        Lists.<Result<TimeseriesResultValue>>newArrayList()
+    );
+  }
+
+  private static <T> QueryRunner<T> makeQueryRunner(
+      QueryRunnerFactory<T, Query<T>> factory,
+      Segment adapter
+  )
+  {
+    return new FinalizeResultsQueryRunner<T>(
+        factory.createRunner(adapter),
+        factory.getToolchest()
+    );
+  }
+}

From 8fc10a61a0bde0d00e2251b330f078dee514638f Mon Sep 17 00:00:00 2001
From: Fangjin Yang <fangjin@metamarketsgroup.com>
Date: Tue, 29 Jan 2013 13:23:57 -0800
Subject: [PATCH 2/2] bug fix for incremental index and 1 row

---
 .../metamx/druid/index/v1/IncrementalIndexStorageAdapter.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/src/main/java/com/metamx/druid/index/v1/IncrementalIndexStorageAdapter.java b/server/src/main/java/com/metamx/druid/index/v1/IncrementalIndexStorageAdapter.java
index ef5ecef5d56b..bf8e0ac96532 100644
--- a/server/src/main/java/com/metamx/druid/index/v1/IncrementalIndexStorageAdapter.java
+++ b/server/src/main/java/com/metamx/druid/index/v1/IncrementalIndexStorageAdapter.java
@@ -215,7 +215,7 @@ public void reset()
                           while (baseIter.hasNext()) {
                             currEntry.set(baseIter.next());
                             if (filterMatcher.matches()) {
-                              break;
+                              return;
                             }
 
                             numAdvanced++;