embed grpc server stream and client so they can be accessed (#1916)

2020-08-09 15:43:41 +01:00
parent 6cb3fad421
commit 1003b968cf
2 changed files with 13 additions and 7 deletions
--- a/grpc.go
+++ b/grpc.go
@@ -238,15 +238,15 @@ func (g *grpcClient) stream(ctx context.Context, node *registry.Node, req client

 	// setup the stream response
 	stream := &grpcStream{
-		context: ctx,
-		request: req,
+		ClientStream: st,
+		context:      ctx,
+		request:      req,
 		response: &response{
 			conn:   cc,
 			stream: st,
 			codec:  cf,
 			gcodec: codec,
 		},
-		stream: st,
 		conn:   cc,
 		cancel: cancel,
 	}