capture args in map

99designs · Apr 5, 2018 · 28d0c81 · 28d0c81
1 parent e266fab
commit 28d0c81
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/codegen/object.go b/codegen/object.go
@@ -89,8 +89,8 @@ func (f *Field) CallArgs() string {
 		}
 	}
 
-	for i := range f.Args {
-		args = append(args, "arg"+strconv.Itoa(i))
+	for _, arg := range f.Args {
+		args = append(args, "args["+strconv.Quote(arg.GQLName)+"].("+arg.Signature()+")")
 	}
 
 	return strings.Join(args, ", ")

diff --git a/codegen/templates/args.gotpl b/codegen/templates/args.gotpl
@@ -1,3 +1,4 @@
+	{{- if . }}args := map[string]interface{}{} {{end}}
 	{{- range $i, $arg := . }}
 		var arg{{$i}} {{$arg.Signature }}
 		if tmp, ok := field.Args[{{$arg.GQLName|quote}}]; ok {
@@ -25,4 +26,5 @@
 			}
 		}
 		{{end }}
+		args[{{$arg.GQLName|quote}}] = arg{{$i}}
 	{{- end -}}