buz 2.15.10rc3__py3-none-any.whl → 2.15.10rc4__py3-none-any.whl

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")

{buz-2.15.10rc3.dist-info → buz-2.15.10rc4.dist-info}/METADATA

@@ -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.dist-info → buz-2.15.10rc4.dist-info}/RECORD

@@ -167,7 +167,7 @@ buz/kafka/infrastructure/aiokafka/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 buz/kafka/infrastructure/aiokafka/aiokafka_consumer.py,sha256=3-73e2CJhSjHONsCH_kZB1x5EwhnzLG2cCKGmkU4q0s,10008
 buz/kafka/infrastructure/aiokafka/aiokafka_producer.py,sha256=LteHKIHpT6MKplwmwsPYMsd2GWNJCzus65XDHCIdoN8,3823
 buz/kafka/infrastructure/aiokafka/rebalance/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-buz/kafka/infrastructure/aiokafka/rebalance/kafka_callback_rebalancer.py,sha256=3l7NkTrCt3rBktVIS73cTmCOvv6eFguoCbGMYIUfCFc,1774
+buz/kafka/infrastructure/aiokafka/rebalance/kafka_callback_rebalancer.py,sha256=-sgv29vbDWbvtcPD4qCYmbeVqu78LwkTXgC1fWCOdyk,2194
 buz/kafka/infrastructure/aiokafka/translators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 buz/kafka/infrastructure/aiokafka/translators/consumer_initial_offset_position_translator.py,sha256=WmxkQfoXeTy9mIJtGGhM0eDKeQxhcJczeVAGCbtonVI,617
 buz/kafka/infrastructure/cdc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -250,7 +250,7 @@ buz/serializer/message_to_json_bytes_serializer.py,sha256=RGZJ64t4t4Pz2FCASZZCv-
 buz/wrapper/__init__.py,sha256=GnRdJFcncn-qp0hzDG9dBHLmTJSbHFVjE_yr-MdW_n4,77
 buz/wrapper/async_to_sync.py,sha256=OfK-vrVUhuN-LLLvekLdMbQYtH0ue5lfbvuasj6ovMI,698
 buz/wrapper/event_loop.py,sha256=pfBJ1g-8A2a3YgW8Gf9Fg0kkewoh3-wgTy2KIFDyfHk,266
-buz-2.15.10rc3.dist-info/LICENSE,sha256=Jytu2S-2SPEgsB0y6BF-_LUxIWY7402fl0JSh36TLZE,1062
-buz-2.15.10rc3.dist-info/METADATA,sha256=RH1nBTDq2HcwS_3FwsmWp8zrw_hNop7_jJyRKmAY9e4,1601
-buz-2.15.10rc3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-buz-2.15.10rc3.dist-info/RECORD,,
+buz-2.15.10rc4.dist-info/LICENSE,sha256=Jytu2S-2SPEgsB0y6BF-_LUxIWY7402fl0JSh36TLZE,1062
+buz-2.15.10rc4.dist-info/METADATA,sha256=r-rwEUoCYrmm9DqP5VZHkpFvnv2MW4Xfr5bNFttaE5U,1601
+buz-2.15.10rc4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+buz-2.15.10rc4.dist-info/RECORD,,