use int builder

Author: dnhatn

x-pack/plugin/esql/compute/src/main/generated-src/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregator.java

@@ -107,10 +107,13 @@ $if(BytesRef)$
             return delegate;
         }
         BytesRefVector dict = valuesOrdinal.getDictionaryVector();
-        final int[] hashIds = new int[dict.getPositionCount()];
+        final IntVector hashIds;
         BytesRef spare = new BytesRef();
-        for (int p = 0; p < dict.getPositionCount(); p++) {
-            hashIds[p] = Math.toIntExact(BlockHash.hashOrdToGroup(state.bytes.add(dict.getBytesRef(p, spare))));
+        try (var hashIdsBuilder = values.blockFactory().newIntVectorFixedBuilder(dict.getPositionCount())) {
+            for (int p = 0; p < dict.getPositionCount(); p++) {
+                hashIdsBuilder.appendInt(Math.toIntExact(BlockHash.hashOrdToGroup(state.bytes.add(dict.getBytesRef(p, spare)))));
+            }
+            hashIds = hashIdsBuilder.build();
         }
         IntBlock ordinalIds = valuesOrdinal.getOrdinalsBlock();
         return new GroupingAggregatorFunction.AddInput() {
@@ -130,7 +133,7 @@ $if(BytesRef)$
                         int valuesStart = ordinalIds.getFirstValueIndex(groupPosition + positionOffset);
                         int valuesEnd = valuesStart + ordinalIds.getValueCount(groupPosition + positionOffset);
                         for (int v = valuesStart; v < valuesEnd; v++) {
-                            state.values.add(groupId, ordinalIds.getInt(v));
+                            state.values.add(groupId, hashIds.getInt(ordinalIds.getInt(v)));
                         }
                     }
                 }
@@ -152,7 +155,7 @@ $if(BytesRef)$
                         int valuesStart = ordinalIds.getFirstValueIndex(groupPosition + positionOffset);
                         int valuesEnd = valuesStart + ordinalIds.getValueCount(groupPosition + positionOffset);
                         for (int v = valuesStart; v < valuesEnd; v++) {
-                            state.values.add(groupId, ordinalIds.getInt(v));
+                            state.values.add(groupId, hashIds.getInt(ordinalIds.getInt(v)));
                         }
                     }
                 }
@@ -168,14 +171,14 @@ $if(BytesRef)$
                     int valuesStart = ordinalIds.getFirstValueIndex(groupPosition + positionOffset);
                     int valuesEnd = valuesStart + ordinalIds.getValueCount(groupPosition + positionOffset);
                     for (int v = valuesStart; v < valuesEnd; v++) {
-                        state.values.add(groupId, ordinalIds.getInt(v));
+                        state.values.add(groupId, hashIds.getInt(ordinalIds.getInt(v)));
                     }
                 }
             }
 
             @Override
             public void close() {
-                delegate.close();
+                Releasables.close(hashIds, delegate);
             }
         };
     }
@@ -190,10 +193,13 @@ $if(BytesRef)$
             return delegate;
         }
         BytesRefVector dict = valuesOrdinal.getDictionaryVector();
-        final int[] hashIds = new int[dict.getPositionCount()];
+        final IntVector hashIds;
         BytesRef spare = new BytesRef();
-        for (int p = 0; p < dict.getPositionCount(); p++) {
-            hashIds[p] = Math.toIntExact(BlockHash.hashOrdToGroup(state.bytes.add(dict.getBytesRef(p, spare))));
+        try (var hashIdsBuilder = values.blockFactory().newIntVectorFixedBuilder(dict.getPositionCount())) {
+            for (int p = 0; p < dict.getPositionCount(); p++) {
+                hashIdsBuilder.appendInt(Math.toIntExact(BlockHash.hashOrdToGroup(state.bytes.add(dict.getBytesRef(p, spare)))));
+            }
+            hashIds = hashIdsBuilder.build();
         }
         var ordinalIds = valuesOrdinal.getOrdinalsVector();
         return new GroupingAggregatorFunction.AddInput() {
@@ -207,7 +213,7 @@ $if(BytesRef)$
                     int groupEnd = groupStart + groupIds.getValueCount(groupPosition);
                     for (int g = groupStart; g < groupEnd; g++) {
                         int groupId = groupIds.getInt(g);
-                        state.values.add(groupId, ordinalIds.getInt(groupPosition + positionOffset));
+                        state.values.add(groupId, hashIds.getInt(ordinalIds.getInt(groupPosition + positionOffset)));
                     }
                 }
             }
@@ -222,7 +228,7 @@ $if(BytesRef)$
                     int groupEnd = groupStart + groupIds.getValueCount(groupPosition);
                     for (int g = groupStart; g < groupEnd; g++) {
                         int groupId = groupIds.getInt(g);
-                        state.values.add(groupId, ordinalIds.getInt(groupPosition + positionOffset));
+                        state.values.add(groupId, hashIds.getInt(ordinalIds.getInt(groupPosition + positionOffset)));
                     }
                 }
             }
@@ -231,13 +237,13 @@ $if(BytesRef)$
             public void add(int positionOffset, IntVector groupIds) {
                 for (int groupPosition = 0; groupPosition < groupIds.getPositionCount(); groupPosition++) {
                     int groupId = groupIds.getInt(groupPosition);
-                    state.values.add(groupId, ordinalIds.getInt(groupPosition + positionOffset));
+                    state.values.add(groupId, hashIds.getInt(ordinalIds.getInt(groupPosition + positionOffset)));
                 }
             }
 
             @Override
             public void close() {
-                delegate.close();
+                Releasables.close(hashIds, delegate);
             }
         };
     }