Add getRawHash method to GroupByHash

soscpd · Nov 7, 2016 · 892b073 · 892b073
1 parent 4f607ad
commit 892b073
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/operator/BigintGroupByHash.java b/presto-main/src/main/java/com/facebook/presto/operator/BigintGroupByHash.java
@@ -20,6 +20,7 @@
 import com.facebook.presto.spi.PrestoException;
 import com.facebook.presto.spi.block.Block;
 import com.facebook.presto.spi.block.BlockBuilder;
+import com.facebook.presto.spi.type.BigintType;
 import com.facebook.presto.spi.type.Type;
 import com.facebook.presto.type.BigintOperators;
 import com.google.common.collect.ImmutableList;
@@ -189,6 +190,12 @@ public int putIfAbsent(int position, Page page)
         return putIfAbsent(position, block);
     }
 
+    @Override
+    public long getRawHash(int groupId)
+    {
+        return BigintType.hash(valuesByGroupId.get(groupId));
+    }
+
     private int putIfAbsent(int position, Block block)
     {
         if (block.isNull(position)) {

diff --git a/presto-main/src/main/java/com/facebook/presto/operator/GroupByHash.java b/presto-main/src/main/java/com/facebook/presto/operator/GroupByHash.java
@@ -64,4 +64,6 @@ static GroupByHash createGroupByHash(
     boolean contains(int position, Page page, int[] hashChannels);
 
     int putIfAbsent(int position, Page page);
+
+    long getRawHash(int groupyId);
 }
diff --git a/presto-main/src/main/java/com/facebook/presto/operator/MultiChannelGroupByHash.java b/presto-main/src/main/java/com/facebook/presto/operator/MultiChannelGroupByHash.java
@@ -132,6 +132,15 @@ public MultiChannelGroupByHash(
         groupAddressByGroupId.ensureCapacity(maxFill);
     }
 
+    @Override
+    public long getRawHash(int groupId)
+    {
+        long address = groupAddressByGroupId.get(groupId);
+        int blockIndex = decodeSliceIndex(address);
+        int position = decodePosition(address);
+        return hashStrategy.hashPosition(blockIndex, position);
+    }
+
     @Override
     public long getEstimatedSize()
     {