fix graceful shutdown

Trendyol · Dec 10, 2023 · f2ad547 · f2ad547
1 parent f3b28f1
commit f2ad547
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/batch_consumer.go b/batch_consumer.go
@@ -76,6 +76,8 @@ func (b *batchConsumer) startBatch() {
  b.consume(&messages, &commitMessages)
  case msg, ok := <-b.incomingMessageStream:
  if !ok {
+ close(b.batchConsumingStream)
+ close(b.messageProcessedStream)
  return
  }
 

diff --git a/consumer.go b/consumer.go
@@ -68,6 +68,8 @@ func (c *consumer) startBatch() {
  c.consume(&messages, &commitMessages)
  case msg, ok := <-c.incomingMessageStream:
  if !ok {
+ close(c.singleConsumingStream)
+ close(c.messageProcessedStream)
  return
  }
 

diff --git a/consumer_base.go b/consumer_base.go
@@ -160,10 +160,8 @@ func (c *base) Stop() error {
  c.cancelFn()
  c.quit <- struct{}{}
  close(c.incomingMessageStream)
- close(c.singleConsumingStream)
- close(c.batchConsumingStream)
- close(c.messageProcessedStream)
  c.wg.Wait()
+
  err = c.r.Close()
  })