implement union type

tribunadigital · Feb 9, 2021 · 0290bcb · 0290bcb
1 parent b20b6e0
commit 0290bcb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/internal/exec/exec.go b/internal/exec/exec.go
@@ -264,6 +264,12 @@ func (r *Request) execSelectionSet(ctx context.Context, sels []selected.Selectio
 			return
 		}
 
+		if s.ExtResolver != nil {
+			if v, ok := s.ExtResolver[t.String()]; ok {
+				resolver = reflect.NewAt(v.Elem().Type(), unsafe.Pointer(resolver.Elem().UnsafeAddr()))
+			}
+		}
+
 		r.execSelections(ctx, sels, path, s, resolver, out, false)
 		return
 	}

diff --git a/internal/exec/resolvable/resolvable.go b/internal/exec/resolvable/resolvable.go
@@ -178,6 +178,9 @@ func (b *execBuilder) makeExec(t common.Type, resolverType reflect.Type, ext map
 		return b.makeObjectExec(t.Name, t.Fields, t.PossibleTypes, nonNull, resolverType, ext)
 
 	case *schema.Union:
+		if v, ok := ext[t.Name]; ok {
+			resolverType = reflect.TypeOf(v)
+		}
 		return b.makeObjectExec(t.Name, nil, t.PossibleTypes, nonNull, resolverType, ext)
 	}