Merge pull request #951 from bsara/issue-950-fix-metrics-instrumentation

graphql-java-kickstart · Jun 15, 2023 · 159d5b7 · 159d5b7
2 parents cb82525 + 4c5a3e0
commit 159d5b7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...main/java/graphql/kickstart/autoconfigure/web/servlet/metrics/MetricsInstrumentation.java b/...main/java/graphql/kickstart/autoconfigure/web/servlet/metrics/MetricsInstrumentation.java
@@ -1,6 +1,7 @@
 package graphql.kickstart.autoconfigure.web.servlet.metrics;
 
 import graphql.ExecutionResult;
+import graphql.execution.instrumentation.InstrumentationState;
 import graphql.execution.instrumentation.parameters.InstrumentationExecutionParameters;
 import graphql.execution.instrumentation.tracing.TracingInstrumentation;
 import io.micrometer.core.instrument.MeterRegistry;
@@ -39,6 +40,12 @@ public MetricsInstrumentation(MeterRegistry meterRegistry, boolean tracingEnable
   @Override
   public CompletableFuture<ExecutionResult> instrumentExecutionResult(
       ExecutionResult executionResult, InstrumentationExecutionParameters parameters) {
+    return this.instrumentExecutionResult(executionResult, parameters, null);
+  }
+
+  @Override
+  public CompletableFuture<ExecutionResult> instrumentExecutionResult(
+      ExecutionResult executionResult, InstrumentationExecutionParameters parameters, InstrumentationState rawState) {
 
     if (executionResult.getExtensions() != null
         && executionResult.getExtensions().containsKey(TRACING)) {