v3rpc: report eventReceived correctly
parent
3cf90a4dff
commit
b868f4b1b1
|
@ -66,10 +66,19 @@ func sendLoop(stream pb.Watch_WatchServer, watcher storage.Watcher, closec chan
|
|||
return
|
||||
}
|
||||
err := stream.Send(&pb.WatchResponse{Event: &e})
|
||||
storage.ReportEventReceived()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
case <-closec:
|
||||
// drain the chan to clean up pending events
|
||||
for {
|
||||
_, ok := <-watcher.Chan()
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
storage.ReportEventReceived()
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue