Use arg function for generated Complexity method

99designs · Sep 5, 2018 · 31505ff · 31505ff
1 parent ebdbeba
commit 31505ff
Show file tree

Hide file tree

Showing 10 changed files with 96 additions and 691 deletions.
diff --git a/codegen/templates/data.go b/codegen/templates/data.go
diff --git a/codegen/templates/generated.gotpl b/codegen/templates/generated.gotpl
@@ -99,17 +99,11 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
 						if e.complexity.{{$object.GQLType|toCamel}}.{{$field.GQLName|toCamel}} == nil {
 							break
 						}
-						{{ if $field.Args }}args := map[string]interface{}{} {{end}}
-						{{ range $i, $arg := $field.Args }}
-							var arg{{$i}} {{$arg.Signature }}
-							if tmp, ok := rawArgs[{{$arg.GQLName|quote}}]; ok {
-								var err error
-								{{$arg.Unmarshal (print "arg" $i) "tmp" }}
-								if err != nil {
-									return 0, false
-								}
+						{{ if $field.Args }}
+							args, err := {{ $field.ArgsFunc }}(rawArgs)
+							if err != nil {
+								return 0, false
 							}
-							args[{{$arg.GQLName|quote}}] = arg{{$i}}
 						{{ end }}
 						return e.complexity.{{$object.GQLType|toCamel}}.{{$field.GQLName|toCamel}}(childComplexity{{if $field.Args}}, {{$field.ComplexityArgs}} {{end}}), true
 				{{ end }}