Use getInt to access IntArrayList

trinodb · Jul 30, 2020 · deb9c13 · deb9c13
1 parent 15ffd9c
commit deb9c13
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/presto-main/src/main/java/io/prestosql/operator/aggregation/TypedSet.java b/presto-main/src/main/java/io/prestosql/operator/aggregation/TypedSet.java
@@ -122,7 +122,7 @@ public boolean contains(Block block, int position)
             return containsNullElement;
         }
         else {
-            return blockPositionByHash.get(getHashPositionOfElement(block, position)) != EMPTY_SLOT;
+            return blockPositionByHash.getInt(getHashPositionOfElement(block, position)) != EMPTY_SLOT;
         }
     }
 
@@ -137,7 +137,7 @@ public void add(Block block, int position)
         }
 
         int hashPosition = getHashPositionOfElement(block, position);
-        if (blockPositionByHash.get(hashPosition) == EMPTY_SLOT) {
+        if (blockPositionByHash.getInt(hashPosition) == EMPTY_SLOT) {
             addNewElement(hashPosition, block, position);
         }
     }
@@ -149,7 +149,7 @@ public int size()
 
     public int positionOf(Block block, int position)
     {
-        return blockPositionByHash.get(getHashPositionOfElement(block, position));
+        return blockPositionByHash.getInt(getHashPositionOfElement(block, position));
     }
 
     /**
@@ -159,7 +159,7 @@ private int getHashPositionOfElement(Block block, int position)
     {
         int hashPosition = getMaskedHash(hashPosition(elementType, block, position));
         while (true) {
-            int blockPosition = blockPositionByHash.get(hashPosition);
+            int blockPosition = blockPositionByHash.getInt(hashPosition);
             if (blockPosition == EMPTY_SLOT) {
                 // Doesn't have this element
                 return hashPosition;