Record generation_id after fetch (faust-streaming#235)

Co-authored-by: Eric Kerstens <ekerstens@expediagroup.com>
ekerstens · Dec 10, 2021 · 499e338 · 499e338
1 parent 40e477f
commit 499e338
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/faust/transport/consumer.py b/faust/transport/consumer.py
@@ -701,8 +701,8 @@ async def getmany(self, timeout: float) -> AsyncIterator[Tuple[TP, Message]]:
         # has 1 partition, then t2 will end up being starved most of the time.
         #
         # We solve this by going round-robin through each topic.
-        generation_id = self.app.consumer_generation_id
         records, active_partitions = await self._wait_next_records(timeout)
+        generation_id = self.app.consumer_generation_id
         if records is None or self.should_stop:
             return