post review update

This commit is contained in:
Mahak Mukhi
2017-02-08 14:15:59 -08:00
parent 09fddba2ca
commit dfb494c6e5
2 changed files with 3 additions and 3 deletions

View File

@ -87,7 +87,7 @@ func recvResponse(ctx context.Context, dopts dialOptions, t transport.ClientTran
} }
c.trailerMD = stream.Trailer() c.trailerMD = stream.Trailer()
if peer, ok := peer.FromContext(stream.Context()); ok { if peer, ok := peer.FromContext(stream.Context()); ok {
c.peer = *peer c.peer = peer
} }
return nil return nil
} }

View File

@ -141,7 +141,7 @@ type callInfo struct {
failFast bool failFast bool
headerMD metadata.MD headerMD metadata.MD
trailerMD metadata.MD trailerMD metadata.MD
peer peer.Peer peer *peer.Peer
traceInfo traceInfo // in trace.go traceInfo traceInfo // in trace.go
} }
@ -189,7 +189,7 @@ func Trailer(md *metadata.MD) CallOption {
// unary RPC. // unary RPC.
func Peer(peer *peer.Peer) CallOption { func Peer(peer *peer.Peer) CallOption {
return afterCall(func(c *callInfo) { return afterCall(func(c *callInfo) {
*peer = c.peer *peer = *c.peer
}) })
} }