relocate response log
This commit is contained in:
16
stream.go
16
stream.go
@ -191,6 +191,14 @@ func (cs *clientStream) SendMsg(m interface{}) (err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (cs *clientStream) RecvMsg(m interface{}) (err error) {
|
func (cs *clientStream) RecvMsg(m interface{}) (err error) {
|
||||||
|
err = recv(cs.p, cs.codec, m)
|
||||||
|
defer func() {
|
||||||
|
// err != nil indicates the termination of the stream.
|
||||||
|
if err != nil {
|
||||||
|
cs.finish(err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
if err == nil {
|
||||||
if cs.tracing {
|
if cs.tracing {
|
||||||
cs.mu.Lock()
|
cs.mu.Lock()
|
||||||
if cs.traceInfo.tr != nil {
|
if cs.traceInfo.tr != nil {
|
||||||
@ -204,14 +212,6 @@ func (cs *clientStream) RecvMsg(m interface{}) (err error) {
|
|||||||
}
|
}
|
||||||
cs.mu.Unlock()
|
cs.mu.Unlock()
|
||||||
}
|
}
|
||||||
err = recv(cs.p, cs.codec, m)
|
|
||||||
defer func() {
|
|
||||||
// err != nil indicates the termination of the stream.
|
|
||||||
if err != nil {
|
|
||||||
cs.finish(err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
if err == nil {
|
|
||||||
if !cs.desc.ClientStreams || cs.desc.ServerStreams {
|
if !cs.desc.ClientStreams || cs.desc.ServerStreams {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user