return arg in middleware

99designs · Dec 8, 2018 · f9ee6ce · f9ee6ce
1 parent 5e0456f
commit f9ee6ce
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 13 deletions.
diff --git a/codegen/templates/args.gotpl b/codegen/templates/args.gotpl
@@ -15,12 +15,13 @@
 							return e.directives.{{$directive.Name|ucFirst}}({{$directive.ResolveArgs "tmp" "n" }})
 						},
 					{{- end }}
-					}...)(ctx, func(ctx2 context.Context)(args{{$i}} interface{},err error){
-					{{$arg.Unmarshal (print "args" $i) "tmp" }}
+					}...)(ctx, func(ctx2 context.Context)(interface{}, error){
+					var err error
+					{{$arg.Unmarshal (print "arg" $i) "tmp" }}
 					if err != nil {
 						return nil, err
 					}
-					return
+					return arg{{ $i }}, nil
 				})
 				if err != nil {
 					return nil, err