Ver Fonte

Update gRPC codegen to match API update in grpc/grpc-go#370.

David Symonds há 10 anos atrás
pai
commit
3d2510a4dd
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      protoc-gen-go/internal/grpc/grpc.go

+ 2 - 2
protoc-gen-go/internal/grpc/grpc.go

@@ -368,9 +368,9 @@ func (g *grpc) generateServerMethod(servName string, method *pb.MethodDescriptor
 	outType := g.typeName(method.GetOutputType())
 
 	if !method.GetServerStreaming() && !method.GetClientStreaming() {
-		g.P("func ", hname, "(srv interface{}, ctx ", contextPkg, ".Context, codec ", grpcPkg, ".Codec, buf []byte) (interface{}, error) {")
+		g.P("func ", hname, "(srv interface{}, ctx ", contextPkg, ".Context, dec func(interface{}) error) (interface{}, error) {")
 		g.P("in := new(", inType, ")")
-		g.P("if err := codec.Unmarshal(buf, in); err != nil { return nil, err }")
+		g.P("if err := dec(in); err != nil { return nil, err }")
 		g.P("out, err := srv.(", servName, "Server).", methName, "(ctx, in)")
 		g.P("if err != nil { return nil, err }")
 		g.P("return out, nil")