grpc
/
grpc-go
mirror of https://github.com/grpc/grpc-go.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remote end stats in header() and SendMsg()

This commit is contained in:
Menghan Li 2016-11-07 18:06:11 -08:00
parent badd5540a4
commit 3de821b781
1 changed files with 2 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
stream.go
View File

@ -278,18 +278,7 @@ func (cs *clientStream) Context() context.Context {
return cs.s.Context()
}
func (cs *clientStream) Header() (_ metadata.MD, err error) {
defer func() {
if err != nil && stats.On() {
// Only handle end stats if err != nil.
end := &stats.End{
Client: true,
EndTime: time.Now(),
Error: err,
}
stats.Handle(cs.statsCtx, end)
}
}()
func (cs *clientStream) Header() (metadata.MD, error) {
m, err := cs.s.Header()
if err != nil {
if _, ok := err.(transport.ConnectionError); !ok {
@ -311,17 +300,7 @@ func (cs *clientStream) SendMsg(m interface{}) (err error) {
}
cs.mu.Unlock()
}
defer func() {
if err != nil && err != io.EOF && stats.On() {
// Only handle end stats if err != nil.
// If err == nil, stats.End will be handled when user calls RecvMsg.
end := &stats.End{
Client: true,
Error: err,
}
stats.Handle(cs.statsCtx, end)
}
}()
// TODO generate stats.End if err != nil && err != io.EOF.
defer func() {
if err != nil {
cs.finish(err)