chore: fix errors

hantsy · Nov 25, 2024 · 4a09159 · 4a09159
1 parent 6a200a6
commit 4a09159
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dgs-codegen/src/main/java/com/example/demo/gql/directives/UppercaseDirectiveWiring.java b/dgs-codegen/src/main/java/com/example/demo/gql/directives/UppercaseDirectiveWiring.java
@@ -2,11 +2,11 @@
 
 import com.netflix.graphql.dgs.DgsDirective;
 import graphql.schema.DataFetcherFactories;
+import graphql.schema.FieldCoordinates;
 import graphql.schema.GraphQLFieldDefinition;
 import graphql.schema.idl.SchemaDirectiveWiring;
 import graphql.schema.idl.SchemaDirectiveWiringEnvironment;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.stereotype.Component;
 
 @DgsDirective(name = "uppercase")
 @Slf4j
@@ -16,8 +16,8 @@ public GraphQLFieldDefinition onField(SchemaDirectiveWiringEnvironment<GraphQLFi
 
         var field = env.getElement();
         var parentType = env.getFieldsContainer();
-
-        var originalDataFetcher = env.getCodeRegistry().getDataFetcher(parentType, field);
+        var fieldCoordinates = FieldCoordinates.coordinates(parentType, field.getName());
+        var originalDataFetcher = env.getCodeRegistry().getDataFetcher(fieldCoordinates, field);
         var dataFetcher = DataFetcherFactories.wrapDataFetcher(
                 originalDataFetcher,
                 (dataFetchingEnvironment, value) -> {
@@ -29,7 +29,7 @@ public GraphQLFieldDefinition onField(SchemaDirectiveWiringEnvironment<GraphQLFi
                 }
         );
 
-        env.getCodeRegistry().dataFetcher(parentType, field, dataFetcher);
+        env.getCodeRegistry().dataFetcher(fieldCoordinates, dataFetcher);
         return field;
     }
 }