From 68cd2e10a7c42150cb9f5acdef9ca66d7c98b959 Mon Sep 17 00:00:00 2001
From: xvrl <xavier@metamarkets.com>
Date: Tue, 12 Feb 2013 09:45:26 -0800
Subject: [PATCH] remove unnecessary segment grouping

---
 .../TimeBoundaryQueryQueryToolChest.java             | 12 +-----------
 1 file changed, 1 insertion(+), 11 deletions(-)

diff --git a/client/src/main/java/com/metamx/druid/query/timeboundary/TimeBoundaryQueryQueryToolChest.java b/client/src/main/java/com/metamx/druid/query/timeboundary/TimeBoundaryQueryQueryToolChest.java
index b27a6fd5b80..8e41f1f6eb8 100644
--- a/client/src/main/java/com/metamx/druid/query/timeboundary/TimeBoundaryQueryQueryToolChest.java
+++ b/client/src/main/java/com/metamx/druid/query/timeboundary/TimeBoundaryQueryQueryToolChest.java
@@ -74,19 +74,9 @@ public class TimeBoundaryQueryQueryToolChest
     TimelineObjectHolder<String, ServerSelector> min = null;
     TimelineObjectHolder<String, ServerSelector> max = null;
 
-    // keep track of all segments in a given shard
-    Map<String, Set<TimelineObjectHolder<String, ServerSelector>>> segmentGroups = Maps.newHashMap();
-
     for(TimelineObjectHolder<String, ServerSelector> e : input) {
       final long start = e.getInterval().getStartMillis();
       final long end = e.getInterval().getEndMillis();
-      final String version = e.getVersion();
-
-      if(segmentGroups.containsKey(version)) {
-        segmentGroups.get(version).add(e);
-      } else {
-        segmentGroups.put(version, Sets.newHashSet(e));
-      }
 
       if(min == null || start < minMillis) {
         min = e;
@@ -98,7 +88,7 @@ public class TimeBoundaryQueryQueryToolChest
       }
     }
 
-    return Lists.newArrayList(Sets.union(segmentGroups.get(min.getVersion()), segmentGroups.get(max.getVersion())));
+    return min == max ? Lists.newArrayList(min) : Lists.newArrayList(min , max);
   }
 
   @Override