Support non null types in batch resolver

smallrye · Oct 27, 2023 · 75cde71 · 75cde71
1 parent b9e489d
commit 75cde71
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/server/implementation/src/main/java/io/smallrye/graphql/bootstrap/FederationDataFetcher.java b/server/implementation/src/main/java/io/smallrye/graphql/bootstrap/FederationDataFetcher.java
@@ -27,8 +27,10 @@
 import graphql.schema.GraphQLFieldDefinition;
 import graphql.schema.GraphQLList;
 import graphql.schema.GraphQLNamedSchemaElement;
+import graphql.schema.GraphQLNonNull;
 import graphql.schema.GraphQLObjectType;
 import graphql.schema.GraphQLOutputType;
+import graphql.schema.GraphQLType;
 import io.smallrye.graphql.spi.config.Config;
 
 class FederationDataFetcher implements DataFetcher<CompletableFuture<List<Object>>> {
@@ -116,7 +118,10 @@ private boolean matchesReturnType(GraphQLFieldDefinition field, String typename)
     }
 
     private boolean matchesReturnTypeList(GraphQLFieldDefinition field, String typename) {
-        GraphQLOutputType listType = field.getType();
+        GraphQLType listType = field.getType();
+        if (listType instanceof GraphQLNonNull) {
+            listType = ((GraphQLNonNull) listType).getOriginalWrappedType();
+        }
         if (listType instanceof GraphQLList) {
             var returnType = ((GraphQLList) listType).getOriginalWrappedType();
             return returnType instanceof GraphQLNamedSchemaElement