Merge pull request #223 from blinklabs-io/fix/protocol-state-transiti…

…on-timer-shutdown fix: stop protocol state transition timer on shutdown
blinklabs-io · Apr 5, 2023 · 695c11e · 695c11e
2 parents e6a93ec + 8b1e784
commit 695c11e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/protocol/protocol.go b/protocol/protocol.go
@@ -107,6 +107,13 @@ func (p *Protocol) Start() {
 		close(p.sendStateQueueChan)
 		close(p.recvReadyChan)
 		close(p.sendReadyChan)
+		if p.stateTransitionTimer != nil {
+			// Stop timer and drain channel
+			if !p.stateTransitionTimer.Stop() {
+				<-p.stateTransitionTimer.C
+			}
+			p.stateTransitionTimer = nil
+		}
 	}()
 	// Set initial state
 	p.setState(p.config.InitialState)