dnt_net: always increment singleMes in send request retries

libp2p · Sep 17, 2017 · c0e98e4 · c0e98e4
1 parent c4f4864
commit c0e98e4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dht_net.go b/dht_net.go
@@ -231,6 +231,9 @@ func (ms *messageSender) SendRequest(ctx context.Context, pmes *pb.Message) (*pb
 		}
 
 		if err := ms.w.WriteMsg(pmes); err != nil {
+			if retry {
+				ms.singleMes++
+			}
 			ms.resetHard()
 			ms.lk.Unlock()
 
@@ -289,6 +292,9 @@ func (ms *messageSender) SendRequest(ctx context.Context, pmes *pb.Message) (*pb
 
 		if res.err != nil {
 			ms.lk.Lock()
+			if retry {
+				ms.singleMes++
+			}
 			ms.resetSoft(rcount)
 			ms.lk.Unlock()