Adds RootFieldInterceptor to extension interfaces (#1647)

* Adds RootFieldInterceptor to extension interfaces * #1647 Regenerates the integration folder * Regenerates example folder * #1647 Adds Root Field Context separate from Field Context * Re-generate after changes Signed-off-by: Steve Coffman <steve@khanacademy.org> Co-authored-by: Steve Coffman <steve@khanacademy.org>
99designs · Oct 14, 2021 · 7435403 · 7435403
1 parent 8b25c9e
commit 7435403
Show file tree

Hide file tree

Showing 24 changed files with 5,705 additions and 1,076 deletions.
diff --git a/codegen/object.gotpl b/codegen/object.gotpl
@@ -25,56 +25,84 @@ func (ec *executionContext) _{{$object.Name}}(ctx context.Context, sel ast.Selec
 {{- else }}
 func (ec *executionContext) _{{$object.Name}}(ctx context.Context, sel ast.SelectionSet{{ if not $object.Root }},obj {{$object.Reference | ref }}{{ end }}) graphql.Marshaler {
 	fields := graphql.CollectFields(ec.OperationContext, sel, {{$object.Name|lcFirst}}Implementors)
-	{{if $object.Root}}
-		ctx = graphql.WithFieldContext(ctx, &graphql.FieldContext{
-			Object: {{$object.Name|quote}},
-		})
-	{{end}}
-
+    {{- if $object.Root }}
+        ctx = graphql.WithFieldContext(ctx, &graphql.FieldContext{
+            Object: {{$object.Name|quote}},
+        })
+    {{end}}
 	out := graphql.NewFieldSet(fields)
 	var invalids uint32
 	for i, field := range fields {
-		switch field.Name {
-		case "__typename":
-			out.Values[i] = graphql.MarshalString({{$object.Name|quote}})
-		{{- range $field := $object.Fields }}
-		case "{{$field.Name}}":
-			{{- if $field.IsConcurrent }}
-				field := field
-				out.Concurrently(i, func() (res graphql.Marshaler) {
-					defer func() {
-						if r := recover(); r != nil {
-							ec.Error(ctx, ec.Recover(ctx, r))
-						}
-					}()
-					res = ec._{{$object.Name}}_{{$field.Name}}(ctx, field{{if not $object.Root}}, obj{{end}})
-					{{- if $field.TypeReference.GQL.NonNull }}
-						if res == graphql.Null {
-							{{- if $object.IsConcurrent }}
-								atomic.AddUint32(&invalids, 1)
-							{{- else }}
-								invalids++
-							{{- end }}
-						}
-					{{- end }}
-					return res
-				})
-			{{- else }}
-				out.Values[i] = ec._{{$object.Name}}_{{$field.Name}}(ctx, field{{if not $object.Root}}, obj{{end}})
-				{{- if $field.TypeReference.GQL.NonNull }}
-					if out.Values[i] == graphql.Null {
-						{{- if $object.IsConcurrent }}
-							atomic.AddUint32(&invalids, 1)
-						{{- else }}
-							invalids++
-						{{- end }}
-					}
-				{{- end }}
-			{{- end }}
-		{{- end }}
-		default:
-			panic("unknown field " + strconv.Quote(field.Name))
-		}
+        {{- if $object.Root }}
+            innerCtx := graphql.WithRootFieldContext(ctx, &graphql.RootFieldContext{
+                Object: field.Name,
+                Field: field,
+            })
+        {{end}}
+        switch field.Name {
+        case "__typename":
+            out.Values[i] = graphql.MarshalString({{$object.Name|quote}})
+        {{- range $field := $object.Fields }}
+        case "{{$field.Name}}":
+            {{- if $field.IsConcurrent }}
+                field := field
+
+                innerFunc := func(ctx context.Context) (res graphql.Marshaler) {
+                    defer func() {
+                        if r := recover(); r != nil {
+                            ec.Error(ctx, ec.Recover(ctx, r))
+                        }
+                    }()
+                    res = ec._{{$object.Name}}_{{$field.Name}}(ctx, field{{if not $object.Root}}, obj{{end}})
+                    {{- if $field.TypeReference.GQL.NonNull }}
+                        if res == graphql.Null {
+                            {{- if $object.IsConcurrent }}
+                                atomic.AddUint32(&invalids, 1)
+                            {{- else }}
+                                invalids++
+                            {{- end }}
+                        }
+                    {{- end }}
+                    return res
+                }
+
+                {{if $object.Root}}
+                    rrm := func(ctx context.Context) graphql.Marshaler {
+                        return ec.OperationContext.RootResolverMiddleware(ctx, innerFunc)
+                    }
+                {{end}}
+
+                out.Concurrently(i, func() graphql.Marshaler {
+                    {{- if $object.Root -}}
+                        return rrm(innerCtx)
+                    {{- else -}}
+                        return innerFunc(ctx)
+                    {{end}}
+                })
+            {{- else }}
+                innerFunc := func(ctx context.Context) (res graphql.Marshaler) {
+                    return ec._{{$object.Name}}_{{$field.Name}}(ctx, field{{if not $object.Root}}, obj{{end}})
+                }
+                {{if $object.Root}}
+                    out.Values[i] = ec.OperationContext.RootResolverMiddleware(innerCtx, innerFunc)
+                {{else}}
+                    out.Values[i] = innerFunc(ctx)
+                {{end}}
+
+                {{- if $field.TypeReference.GQL.NonNull }}
+                    if out.Values[i] == graphql.Null {
+                        {{- if $object.IsConcurrent }}
+                            atomic.AddUint32(&invalids, 1)
+                        {{- else }}
+                            invalids++
+                        {{- end }}
+                    }
+                {{- end }}
+            {{- end }}
+        {{- end }}
+        default:
+            panic("unknown field " + strconv.Quote(field.Name))
+        }
 	}
 	out.Dispatch()
 	if invalids > 0 { return graphql.Null }