buz 2.15.10rc3__tar.gz → 2.15.10rc4__tar.gz

{buz-2.15.10rc3 → buz-2.15.10rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.15.10rc3
+Version: 2.15.10rc4
 Summary: Buz is a set of light, simple and extensible implementations of event, command and query buses.
 License: MIT
 Author: Luis Pintado Lozano

{buz-2.15.10rc3 → buz-2.15.10rc4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "buz"
-version = "2.15.10-rc3"
+version = "2.15.10-rc4"
 description = "Buz is a set of light, simple and extensible implementations of event, command and query buses."
 readme = "README.md"
 authors = ["Luis Pintado Lozano <luis.pintado.lozano@gmail.com>", "Gerardo Parra <gprauxiliar@gmail.com>"]

{buz-2.15.10rc3 → buz-2.15.10rc4}/src/buz/kafka/infrastructure/aiokafka/rebalance/kafka_callback_rebalancer.py

@@ -1,5 +1,6 @@
+from datetime import datetime
 from logging import Logger
-from typing import Awaitable, Callable
+from typing import Awaitable, Callable, Optional
 
 from aiokafka import ConsumerRebalanceListener, TopicPartition
 
@@ -16,11 +17,14 @@ class KafkaCallbackRebalancer(ConsumerRebalanceListener):
         self.__on_partition_revoked = on_partitions_revoked
         self.__on_partition_assigned = on_partitions_assigned
         self.__assigned_partitions: set[TopicPartition] = set()
+        self.__rebalancing_start_time: Optional[datetime] = None
 
     async def on_partitions_revoked(self, revoked: set[TopicPartition]) -> None:
         if len(revoked) == 0:
             return None
 
+        self.__rebalancing_start_time = datetime.now()
+
         self.__logger.info(f"Partitions revoked by the rebalancing process: '{revoked}'")
 
         await self.__on_partition_revoked(revoked)
@@ -40,3 +44,7 @@ class KafkaCallbackRebalancer(ConsumerRebalanceListener):
         self.__assigned_partitions.update(new_partitions_assigned)
 
         self.__logger.info(f"Partitions after assigning process: '{self.__assigned_partitions}'")
+
+        if self.__rebalancing_start_time is not None:
+            elapsed_time_ms = int((datetime.now() - self.__rebalancing_start_time).total_seconds() * 1000)
+            self.__logger.info(f"Rebalancing process ended, elapsed time: {elapsed_time_ms} ms")