Add performance benchmarks for DataFetchingFieldSelectionSet.getImmediateFields

timward60 · timward60 · commit 90dee30d2d3f · 2025-03-18T00:15:01.000-07:00
diff --git a/src/test/java/performance/DFSelectionSetPerformance.java b/src/test/java/performance/DFSelectionSetPerformance.java
@@ -67,11 +67,32 @@ public void benchMarkThroughput(MyState myState, Blackhole blackhole) {
         blackhole.consume(fields);
     }
 
+    @Benchmark
+    @BenchmarkMode(Mode.AverageTime)
+    @OutputTimeUnit(TimeUnit.MILLISECONDS)
+    public void benchMarkAvgTime_getImmediateFields(MyState myState, Blackhole blackhole) {
+        List<SelectedField> fields = getImmediateFields(myState);
+        blackhole.consume(fields);
+    }
+
+    @Benchmark
+    @BenchmarkMode(Mode.Throughput)
+    @OutputTimeUnit(TimeUnit.MILLISECONDS)
+    public void benchMarkThroughput_getImmediateFields(MyState myState, Blackhole blackhole) {
+        List<SelectedField> fields = getImmediateFields(myState);
+        blackhole.consume(fields);
+    }
+
     private List<SelectedField> getSelectedFields(MyState myState) {
         DataFetchingFieldSelectionSet dataFetchingFieldSelectionSet = DataFetchingFieldSelectionSetImpl.newCollector(myState.schema, myState.outputFieldType, () -> myState.normalisedField);
         return dataFetchingFieldSelectionSet.getFields("wontBeFound");
     }
 
+    private List<SelectedField> getImmediateFields(MyState myState) {
+        DataFetchingFieldSelectionSet dataFetchingFieldSelectionSet = DataFetchingFieldSelectionSetImpl.newCollector(myState.schema, myState.outputFieldType, () -> myState.normalisedField);
+        return dataFetchingFieldSelectionSet.getImmediateFields();
+    }
+
     public static void mainX(String[] args) throws InterruptedException {
         MyState myState = new MyState();
         myState.setup();
