Fix union queries not working in @requires

plugin/federation/federation.go

@@ -4,6 +4,7 @@ import (
 	_ "embed"
 	"errors"
 	"fmt"
+	"github.com/99designs/gqlgen/internal/rewrite"
 	"sort"
 	"strings"
 
@@ -12,7 +13,6 @@ import (
 	"github.com/99designs/gqlgen/codegen"
 	"github.com/99designs/gqlgen/codegen/config"
 	"github.com/99designs/gqlgen/codegen/templates"
-	"github.com/99designs/gqlgen/internal/rewrite"
 	"github.com/99designs/gqlgen/plugin/federation/fieldset"
 )
 
@@ -279,10 +279,13 @@ func (f *Federation) GenerateCode(data *codegen.Data) error {
 				typeString := strings.Split(obj.Type.String(), ".")
 				requiresImports[strings.Join(typeString[:len(typeString)-1], ".")] = true
 
+				if containsUnionField(reqField) {
+					continue
+				}
+
 				cgField := reqField.Field.TypeReference(obj, data.Objects)
 				reqField.Type = cgField.TypeReference
 			}
-
 			// add type info to entity
 			e.Type = obj.Type
 		}
@@ -329,6 +332,15 @@ func (f *Federation) GenerateCode(data *codegen.Data) error {
 	})
 }
 
+func containsUnionField(reqField *Requires) bool {
+	for _, requireFields := range reqField.Field {
+		if strings.HasPrefix(requireFields, "... on") {
+			return true
+		}
+	}
+	return false
+}
+
 // Fill in types for key fields
 func populateKeyFieldTypes(
 	resolver *EntityResolver,

plugin/federation/fieldset/fieldset.go

@@ -133,8 +133,22 @@
 // parseUnnestedKeyFieldSet // handles simple case where none of the fields are nested.
 func parseUnnestedKeyFieldSet(raw string, prefix []string) Set {
 	ret := Set{}
+	unionField := false
 
 	for _, s := range strings.Fields(raw) {
+		if s == "..." {
+			continue
+		}
+		if s == "on" {
+			unionField = true
+			continue
+		}
+
+		if unionField == true {
+			s = "... on " + s
+			unionField = false
+		}
+
 		next := append(prefix[0:len(prefix):len(prefix)], s) //nolint:gocritic // set cap=len in order to force slice reallocation
 		ret = append(ret, next)
 	}

plugin/federation/testdata/explicitrequires/schema.graphql

@@ -11,6 +11,22 @@ type World @key(fields: "hello { name } foo   ") {
     hello: Hello
 }
 
+type Person @key(fields: "name"){
+    name: String!
+    gender: Gender!
+    welcomeMessage: String @requires(fields:"gender { ... on Male {description} ... on Female {description}}")
+}
+
+union Gender = Male | Female
+
+type Male {
+    description: String!
+}
+
+type Female {
+    description: String!
+}
+
 type WorldWithMultipleKeys @key(fields: "hello { name } foo   ") @key(fields: "bar") {
     foo: String!
     bar: Int!