feat: added graphql.UnmarshalInputFromContext (#2131)

* feat: added graphql.UnmarshalInputFromContext * chore: run go generate for _examples * fix: apply suggestions from code review Co-authored-by: Ariel Mashraki <7413593+a8m@users.noreply.github.com> * fix: update error cases * fix: fixed unit-test by update root_.gotpl * fix: apply suggestions from code review * fix: update graphql/input.go Co-authored-by: Ariel Mashraki <7413593+a8m@users.noreply.github.com> Co-authored-by: Ariel Mashraki <7413593+a8m@users.noreply.github.com>
99designs · Apr 28, 2022 · fce3a11 · fce3a11
1 parent 6a24e88
commit fce3a11
Show file tree

Hide file tree

Showing 19 changed files with 162 additions and 0 deletions.
diff --git a/_examples/chat/generated.go b/_examples/chat/generated.go
diff --git a/_examples/config/generated.go b/_examples/config/generated.go
diff --git a/_examples/dataloader/generated.go b/_examples/dataloader/generated.go
diff --git a/_examples/federation/accounts/graph/generated/generated.go b/_examples/federation/accounts/graph/generated/generated.go
diff --git a/_examples/federation/products/graph/generated/generated.go b/_examples/federation/products/graph/generated/generated.go
diff --git a/_examples/federation/reviews/graph/generated/generated.go b/_examples/federation/reviews/graph/generated/generated.go
diff --git a/_examples/fileupload/generated.go b/_examples/fileupload/generated.go
diff --git a/_examples/scalars/generated.go b/_examples/scalars/generated.go
diff --git a/_examples/selection/generated.go b/_examples/selection/generated.go
diff --git a/_examples/starwars/generated/exec.go b/_examples/starwars/generated/exec.go
diff --git a/_examples/todo/generated.go b/_examples/todo/generated.go
diff --git a/_examples/type-system-extension/generated.go b/_examples/type-system-extension/generated.go
diff --git a/codegen/generated!.gotpl b/codegen/generated!.gotpl
@@ -135,13 +135,21 @@
 	func (e *executableSchema) Exec(ctx context.Context) graphql.ResponseHandler {
 		rc := graphql.GetOperationContext(ctx)
 		ec := executionContext{rc, e}
+		inputUnmarshalMap := graphql.BuildUnmarshalerMap(
+			{{- range $input := .Inputs -}}
+				{{ if not $input.HasUnmarshal }}
+					ec.unmarshalInput{{ $input.Name }},
+				{{- end }}
+			{{- end }}
+		)
 		first := true
 
 		switch rc.Operation.Operation {
 		{{- if .QueryRoot }} case ast.Query:
 			return func(ctx context.Context) *graphql.Response {
 				if !first { return nil }
 				first = false
+				ctx = graphql.WithUnmarshalerMap(ctx, inputUnmarshalMap)
 				{{ if .Directives.LocationDirectives "QUERY" -}}
 					data := ec._queryMiddleware(ctx, rc.Operation, func(ctx context.Context) (interface{}, error){
 						return ec._{{.QueryRoot.Name}}(ctx, rc.Operation.SelectionSet), nil
@@ -162,6 +170,7 @@
 			return func(ctx context.Context) *graphql.Response {
 				if !first { return nil }
 				first = false
+				ctx = graphql.WithUnmarshalerMap(ctx, inputUnmarshalMap)
 				{{ if .Directives.LocationDirectives "MUTATION" -}}
 					data := ec._mutationMiddleware(ctx, rc.Operation, func(ctx context.Context) (interface{}, error){
 						return ec._{{.MutationRoot.Name}}(ctx, rc.Operation.SelectionSet), nil

diff --git a/codegen/root_.gotpl b/codegen/root_.gotpl
@@ -102,13 +102,21 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
 func (e *executableSchema) Exec(ctx context.Context) graphql.ResponseHandler {
 	rc := graphql.GetOperationContext(ctx)
 	ec := executionContext{rc, e}
+	inputUnmarshalMap := graphql.BuildUnmarshalerMap(
+		{{- range $input := .Inputs -}}
+			{{ if not $input.HasUnmarshal }}
+				ec.unmarshalInput{{ $input.Name }},
+			{{- end }}
+		{{- end }}
+	)
 	first := true
 
 	switch rc.Operation.Operation {
 	{{- if .QueryRoot }} case ast.Query:
 		return func(ctx context.Context) *graphql.Response {
 			if !first { return nil }
 			first = false
+			ctx = graphql.WithUnmarshalerMap(ctx, inputUnmarshalMap)
 			{{ if .Directives.LocationDirectives "QUERY" -}}
 				data := ec._queryMiddleware(ctx, rc.Operation, func(ctx context.Context) (interface{}, error){
 					return ec._{{.QueryRoot.Name}}(ctx, rc.Operation.SelectionSet), nil
@@ -129,6 +137,7 @@ func (e *executableSchema) Exec(ctx context.Context) graphql.ResponseHandler {
 		return func(ctx context.Context) *graphql.Response {
 			if !first { return nil }
 			first = false
+			ctx = graphql.WithUnmarshalerMap(ctx, inputUnmarshalMap)
 			{{ if .Directives.LocationDirectives "MUTATION" -}}
 				data := ec._mutationMiddleware(ctx, rc.Operation, func(ctx context.Context) (interface{}, error){
 					return ec._{{.MutationRoot.Name}}(ctx, rc.Operation.SelectionSet), nil

diff --git a/codegen/testserver/followschema/root_.generated.go b/codegen/testserver/followschema/root_.generated.go
diff --git a/codegen/testserver/singlefile/generated.go b/codegen/testserver/singlefile/generated.go