Make the forwarder function customizable

grpc-ecosystem · Jul 13, 2015 · 5a2c9c3 · 5a2c9c3
1 parent d430203
commit 5a2c9c3
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 15 deletions.
diff --git a/examples/a_bit_of_everything.pb.gw.go b/examples/a_bit_of_everything.pb.gw.go
diff --git a/examples/echo_service.pb.gw.go b/examples/echo_service.pb.gw.go
diff --git a/protoc-gen-grpc-gateway/gengateway/template.go b/protoc-gen-grpc-gateway/gengateway/template.go
@@ -228,9 +228,9 @@ func Register{{$svc.GetName}}Handler(ctx context.Context, mux *runtime.ServeMux,
 			return
 		}
 		{{if $m.GetServerStreaming}}
-		runtime.ForwardResponseStream(w, func() (proto.Message, error) { return resp.Recv() })
+		forward_{{$svc.GetName}}_{{$m.GetName}}_{{$b.Index}}(w, func() (proto.Message, error) { return resp.Recv() })
 		{{else}}
-		runtime.ForwardResponseMessage(ctx, w, resp)
+		forward_{{$svc.GetName}}_{{$m.GetName}}_{{$b.Index}}(ctx, w, resp)
 		{{end}}
 	})
 	{{end}}
@@ -245,5 +245,13 @@ var (
 	{{end}}
 	{{end}}
 )
+
+var (
+	{{range $m := $svc.Methods}}
+	{{range $b := $m.Bindings}}
+	forward_{{$svc.GetName}}_{{$m.GetName}}_{{$b.Index}} = {{if $m.GetServerStreaming}}runtime.ForwardResponseStream{{else}}runtime.ForwardResponseMessage{{end}}
+	{{end}}
+	{{end}}
+)
 {{end}}`))
 )