From f18419b0fb92bf523668d1272c1bd9764c5015ed Mon Sep 17 00:00:00 2001
From: Nodar Ambroladze <nodar.ambroladze@gmail.com>
Date: Thu, 16 May 2024 22:48:37 +0200
Subject: [PATCH] Delete heartbeat before stopAndWait

---
 pubsub/consumer.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pubsub/consumer.go b/pubsub/consumer.go
index af1345f059..d74d4ef1b2 100644
--- a/pubsub/consumer.go
+++ b/pubsub/consumer.go
@@ -108,8 +108,8 @@ func (c *Consumer[Request, Response]) listenForInterrupt() {
 }
 
 func (c *Consumer[Request, Response]) StopAndWait() {
-	c.StopWaiter.StopAndWait()
 	c.deleteHeartBeat(c.GetContext())
+	c.StopWaiter.StopAndWait()
 }
 
 func heartBeatKey(id string) string {