Populate callInfo.peer object for streaming RPCs (#1356)

grpc · Jul 11, 2017 · 86ec6ba · 86ec6ba
1 parent d69dedd
commit 86ec6ba
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/stream.go b/stream.go
@@ -29,6 +29,7 @@ import (
 	"golang.org/x/net/trace"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
+	"google.golang.org/grpc/peer"
 	"google.golang.org/grpc/stats"
 	"google.golang.org/grpc/status"
 	"google.golang.org/grpc/transport"
@@ -221,6 +222,10 @@ func newClientStream(ctx context.Context, desc *StreamDesc, cc *ClientConn, meth
 		}
 		break
 	}
+	// Set callInfo.peer object from stream's context.
+	if peer, ok := peer.FromContext(s.Context()); ok {
+		c.peer = peer
+	}
 	cs := &clientStream{
 		opts:   opts,
 		c:      c,