From 82f0244f96b9523533d14959997f0721c39f6b1e Mon Sep 17 00:00:00 2001
From: Dmytro Milinevskyi <dmytro.milinevskyi@datadoghq.com>
Date: Wed, 23 Mar 2022 12:29:18 +0100
Subject: [PATCH] consumer-group: do not cancel sessions context during
 rebalance

---
 consumer_group.go | 1 -
 1 file changed, 1 deletion(-)

diff --git a/consumer_group.go b/consumer_group.go
index 18f587880..37576ad4d 100644
--- a/consumer_group.go
+++ b/consumer_group.go
@@ -869,7 +869,6 @@ func (s *consumerGroupSession) heartbeatLoop() {
 			retries = s.parent.config.Metadata.Retry.Max
 		case ErrRebalanceInProgress:
 			retries = s.parent.config.Metadata.Retry.Max
-			s.cancel()
 		case ErrUnknownMemberId, ErrIllegalGeneration:
 			return
 		default: