Merge pull request #91 from anthonyraymond/patch-2

Fix heart beat problem in server/client/conn.go closes #66
go-stomp · Jan 13, 2021 · 0036ca2 · 0036ca2
2 parents 9c81a64 + dd67e27
commit 0036ca2
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/server/client/conn.go b/server/client/conn.go
@@ -188,11 +188,11 @@ func (c *Conn) processLoop() {
 
 	c.writer = frame.NewWriter(c.rw)
 	c.stateFunc = connecting
-	for {
-		var timerChannel <-chan time.Time
-		var timer *time.Timer
 
-		if c.writeTimeout > 0 {
+	var timerChannel <-chan time.Time
+	var timer *time.Timer
+	for {
+		if c.writeTimeout > 0 && timer == nil {
 			timer = time.NewTimer(c.writeTimeout)
 			timerChannel = timer.C
 		}
@@ -310,6 +310,11 @@ func (c *Conn) processLoop() {
 			}
 
 		case _ = <-timerChannel:
+			// stop the heart-beat timer
+			if timer != nil {
+				timer.Stop()
+				timer = nil
+			}
 			// write a heart-beat
 			err := c.writer.Write(nil)
 			if err != nil {