opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

relocate response log

Browse Source
This commit is contained in:
yangzhouhan
2015-07-24 15:39:25 -07:00
parent 3b4ecfaa42
commit be4cb2a81a
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
stream.go
View File

@ -191,19 +191,6 @@ func (cs *clientStream) SendMsg(m interface{}) (err error) {
} }
func (cs *clientStream) RecvMsg(m interface{}) (err error) { func (cs *clientStream) RecvMsg(m interface{}) (err error) {
if cs.tracing {
cs.mu.Lock()
if cs.traceInfo.tr != nil {
if cs.traceInfo.tr != nil {
p := &payload{
sent: false,
msg: m,
}
cs.traceInfo.tr.LazyLog(p, true)
}
}
cs.mu.Unlock()
}
err = recv(cs.p, cs.codec, m) err = recv(cs.p, cs.codec, m)
defer func() { defer func() {
// err != nil indicates the termination of the stream. // err != nil indicates the termination of the stream.
@ -212,6 +199,19 @@ func (cs *clientStream) RecvMsg(m interface{}) (err error) {
} }
}() }()
if err == nil { if err == nil {
if cs.tracing {
cs.mu.Lock()
if cs.traceInfo.tr != nil {
if cs.traceInfo.tr != nil {
p := &payload{
sent: false,
msg: m,
}
cs.traceInfo.tr.LazyLog(p, true)
}
}
cs.mu.Unlock()
}
if !cs.desc.ClientStreams || cs.desc.ServerStreams { if !cs.desc.ClientStreams || cs.desc.ServerStreams {
return return
} }