From 0c7e068c133db15e53e547ae3b8c1bdded8654fe Mon Sep 17 00:00:00 2001
From: Dmytro Milinevskyi <dmytro.milinevskyi@datadoghq.com>
Date: Thu, 31 Mar 2022 04:00:47 +0200
Subject: [PATCH] Revert "consumer-group: do not cancel sessions context during
 rebalance"

---
 consumer_group.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/consumer_group.go b/consumer_group.go
index 37576ad4d..18f587880 100644
--- a/consumer_group.go
+++ b/consumer_group.go
@@ -869,6 +869,7 @@ 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: