Fix check

zio · Oct 29, 2024 · 0df325e · 0df325e
1 parent 869e779
commit 0df325e
Showing 1 changed file with 4 additions and 5 deletions.
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
@@ -564,13 +564,12 @@ private[consumer] final class Runloop private (
                                 _ <-
                                   ZIO
                                     .logWarning(
-                                      s"Not all assigned partitions have a (single) stream or vice versa. Assigned: ${assignedTps.mkString(",")}, streams: ${updatedAssignedStreams.map(_.tp).mkString(",")}"
+                                      s"Not all assigned partitions have a (single) stream or vice versa. Assigned: ${currentAssigned.mkString(",")}, streams: ${updatedAssignedStreams.map(_.tp).mkString(",")}"
                                     )
                                     .when(
-                                      assignedTps.size != updatedAssignedStreams.size ||
-                                        assignedTps != updatedAssignedStreams
-                                          .map(_.tp)
-                                          .toSet
+                                      currentAssigned != updatedAssignedStreams
+                                        .map(_.tp)
+                                        .toSet || currentAssigned.size != updatedAssignedStreams.size
                                     )
                               } yield Runloop.PollResult(
                                 records = polledRecords,