clear client stats

grpc · Apr 25, 2017 · 18c603c · 18c603c
1 parent 4402ebf
commit 18c603c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/grpclb.go b/grpclb.go
@@ -284,6 +284,7 @@ func (b *balancer) processServerList(l *lbpb.ServerList, seq int) {
 }
 
 func (b *balancer) sendLoadReport(s *balanceLoadClientStream, interval time.Duration, done <-chan struct{}) {
+	b.clientStats = lbpb.ClientStats{} // Clear client stats.
 	ticker := time.NewTicker(interval)
 	defer ticker.Stop()
 	for {

diff --git a/grpclb/grpclb_test.go b/grpclb/grpclb_test.go
@@ -769,7 +769,7 @@ func TestGRPCLBStatsStreaming(t *testing.T) {
 	}
 	cc.Close()
 
-	time.Sleep(1 * time.Second)
+	time.Sleep(3 * time.Second)
 	if tss.ls.stats.NumCallsStarted != int64(countNormalRPC+countFailedToSend) {
 		t.Errorf("num calls started = %v, want %v+%v", tss.ls.stats.NumCallsStarted, countNormalRPC, countFailedToSend)
 	}