From dd46b6014e0f8b38b32328798616f5247dfd6845 Mon Sep 17 00:00:00 2001
From: Mel <einebeere@gmail.com>
Date: Fri, 8 Apr 2022 16:17:52 +0200
Subject: Tests for heartbeat

---
 pkg/discord/gateway/heartbeat.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'pkg/discord/gateway/heartbeat.go')

diff --git a/pkg/discord/gateway/heartbeat.go b/pkg/discord/gateway/heartbeat.go
index a8f793d..1df753a 100644
--- a/pkg/discord/gateway/heartbeat.go
+++ b/pkg/discord/gateway/heartbeat.go
@@ -50,7 +50,13 @@ func (h *HeartbeatImpl) heartbeatRoutine(interval time.Duration) {
 
 	// REF: heartbeat_interval * jitter
 	jitter := rand.Intn(int(interval))
-	time.Sleep(time.Duration(jitter))
+
+	select {
+	case <-time.After(time.Duration(jitter)):
+	case <-h.ctx.Done():
+		h.logger.Debug().Msg("heartbeat routine stopped before jitter heartbeat")
+		return
+	}
 
 	ticker := time.NewTicker(interval)
 	defer ticker.Stop()
-- 
cgit 1.4.1