Merge pull request #31 from wolfapple/fix/rebalancing

Patch sarama_kafka rebalance fix
ThreeDotsLabs · Jun 27, 2024 · bfef80e · bfef80e
2 parents 2664bae + 05d796c
commit bfef80e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/kafka/subscriber.go b/pkg/kafka/subscriber.go
@@ -518,6 +518,10 @@ func (h consumerGroupHandler) ConsumeClaim(sess sarama.ConsumerGroupSession, cla
 		case <-h.ctx.Done():
 			h.logger.Debug("Ctx was cancelled, stopping consumerGroupHandler", logFields)
 			return nil
+
+		case <-sess.Context().Done():
+			h.logger.Debug("Session ctx was cancelled, stopping consumerGroupHandler", logFields)
+			return nil
 		default:
 			continue
 		}