minor cleanup

zio · Feb 13, 2025 · 3db59ca · 3db59ca
1 parent fea1236
commit 3db59ca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/zio-kafka-test/src/test/scala/zio/kafka/consumer/ConsumerSpec.scala b/zio-kafka-test/src/test/scala/zio/kafka/consumer/ConsumerSpec.scala
@@ -1549,7 +1549,7 @@ object ConsumerSpec extends ZIOSpecDefaultSlf4j with KafkaRandom {
                                       .plainStream(Subscription.topics(topicB), Serde.string, Serde.string)
                                       .mapChunks(_.map(_.value))
                                       .take(messageCount.toLong)
-                                      .timeout(20.seconds)
+                                      .timeout(10.seconds)
                                       .runCollect
                                       .tapError(e => ZIO.logError(s"Error: $e")) <* ZIO.logDebug("Done")
                                   }

diff --git a/zio-kafka/src/main/scala/zio/kafka/consumer/internal/Runloop.scala b/zio-kafka/src/main/scala/zio/kafka/consumer/internal/Runloop.scala
@@ -214,7 +214,8 @@ private[consumer] final class Runloop private (
                s" ${pendingCommitCount} pending commits," +
                s" resuming $partitionsToFetch partitions"
            )
-      _ <- currentStateRef.set(state)
+      _                            <- currentStateRef.set(state)
+      initialConsumerGroupMetadata <- consumer.withConsumerZIO(c => getConsumerGroupMetadataIfAny(c))
       pollResult <-
         consumer.runloopAccess { c =>
           for {
@@ -245,7 +246,7 @@ private[consumer] final class Runloop private (
                                   ignoreRecordsForTps = Set.empty,
                                   pendingRequests = state.pendingRequests,
                                   assignedStreams = state.assignedStreams,
-                                  consumerGroupMetadata = None
+                                  consumerGroupMetadata = initialConsumerGroupMetadata
                                 )
                               )