a bit more efficient load in case the strategy doesn't fit

andimarek · andimarek · commit b6c58bfe50f5 · 2025-06-05T12:16:39.000+10:00
diff --git a/src/main/java/graphql/schema/DataLoaderWithContext.java b/src/main/java/graphql/schema/DataLoaderWithContext.java
@@ -29,11 +29,11 @@ public CompletableFuture<V> load(@NonNull K key, @Nullable Object keyContext) {
         // later than the dispatch, which results in a hanging DL
         CompletableFuture<V> result = super.load(key, keyContext);
         DataFetchingEnvironmentImpl dfeImpl = (DataFetchingEnvironmentImpl) dfe;
-        int level = dfe.getExecutionStepInfo().getPath().getLevel();
-        String path = dfe.getExecutionStepInfo().getPath().toString();
         DataFetchingEnvironmentImpl.DFEInternalState dfeInternalState = (DataFetchingEnvironmentImpl.DFEInternalState) dfeImpl.toInternal();
         if (dfeInternalState.getDataLoaderDispatchStrategy() instanceof PerLevelDataLoaderDispatchStrategy) {
             DeferredCallContext deferredCallContext = dfeInternalState.getDeferredCallContext();
+            int level = dfe.getExecutionStepInfo().getPath().getLevel();
+            String path = dfe.getExecutionStepInfo().getPath().toString();
             ((PerLevelDataLoaderDispatchStrategy) dfeInternalState.dataLoaderDispatchStrategy).newDataLoaderLoadCall(path, level, delegate, dataLoaderName, key, deferredCallContext);
         }
         return result;
