Improve (ncpio/mqttdpinger): net.Conn SetReadDeadline 2

sb-im · Nov 3, 2021 · 790ec05 · 790ec05
1 parent 3c4c7f3
commit 790ec05
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ncpio/mqttdpinger.go b/ncpio/mqttdpinger.go
@@ -45,6 +45,8 @@ func NewPingHandler(client *paho.Client, topic string) *PingHandler {
 // the required interface function()
 func (p *PingHandler) Start(c net.Conn, pt time.Duration) {
 	p.mu.Lock()
+	p.sendCount = 0
+	p.recvCount = 0
 	p.conn = c
 	p.stop = make(chan struct{})
 	p.mu.Unlock()
@@ -78,6 +80,9 @@ func (p *PingHandler) Start(c net.Conn, pt time.Duration) {
 				atomic.AddInt32(&p.pingOutstanding, 1)
 				p.lastPing = time.Now()
 				p.sendCount++
+				if p.sendCount - p.recvCount > 3 {
+					return
+				}
 				p.debug.Println("pingHandler sending ping request")
 			}
 		}