buz 2.15.10rc5__py3-none-any.whl → 2.15.10rc6__py3-none-any.whl

buz/event/infrastructure/buz_kafka/base_buz_aiokafka_async_consumer.py

@@ -114,8 +114,8 @@ class BaseBuzAIOKafkaAsyncConsumer(AsyncConsumer):
     async def configure_http_check_server(self, health_check_port: int) -> web.TCPSite:
         self._logger.info(f"Starting health check server on port {health_check_port}")
         app = web.Application()
-        app.router.add_get("/health", lambda request: self.__health_check())
-        app.router.add_get("/ready", lambda request: self.__is_ready())
+        app.router.add_get("/health", self.__health_check)
+        app.router.add_get("/ready", self.__is_ready)
 
         runner = web.AppRunner(app)
         await runner.setup()
@@ -351,6 +351,8 @@ class BaseBuzAIOKafkaAsyncConsumer(AsyncConsumer):
         self._logger.info(
             f"rebalancing in progress, assigning partitions {topics_partitions} to consumer {consumer.get_consumer_group()}"
         )
+        for topic_partition in topics_partitions:
+            self.__queue_per_consumer_mapper[consumer].create_key(topic_partition)
 
     async def __on_partition_revoked(self, consumer: AIOKafkaConsumer, topics_partitions: set[TopicPartition]) -> None:
         self._logger.info(
@@ -362,7 +364,7 @@ class BaseBuzAIOKafkaAsyncConsumer(AsyncConsumer):
             consumer_fqn=consumer.get_consumer_group(),
         ):
             for topic_partition in topics_partitions:
-                self.__queue_per_consumer_mapper[consumer].clear(topic_partition)
+                self.__queue_per_consumer_mapper[consumer].remove_key(topic_partition)
 
         self.decrease_number_of_rebalancing_in_progress()
 
@@ -382,7 +384,7 @@ class BaseBuzAIOKafkaAsyncConsumer(AsyncConsumer):
     async def __manage_kafka_consumers_stopping(self) -> None:
         await gather(*[kafka_consumer.stop() for kafka_consumer in self.__queue_per_consumer_mapper.keys()])
 
-    async def __health_check(self) -> web.Response:
+    async def __health_check(self, request: web.Request) -> web.Response:
         health_information = {
             "subscribers": [subscriber.fqn() for subscriber in self.__subscribers],
             "number_of_subscribers": len(self.__subscribers),
@@ -391,7 +393,7 @@ class BaseBuzAIOKafkaAsyncConsumer(AsyncConsumer):
 
         return web.Response(text=json.dumps(health_information), content_type="application/json")
 
-    async def __is_ready(self) -> web.Response:
+    async def __is_ready(self, request: web.Request) -> web.Response:
         is_ready = self.__is_worked_initialized
         status_code = 200 if is_ready else 503
 

buz/kafka/infrastructure/aiokafka/rebalance/kafka_callback_rebalancer.py

@@ -47,4 +47,6 @@ class KafkaCallbackRebalancer(ConsumerRebalanceListener):
 
         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")
+            self.__logger.info(
+                f"Rebalancing for topic {set(v.topic for v in assigned)} ended, elapsed time: {elapsed_time_ms} ms"
+            )

buz/queue/in_memory/in_memory_multiqueue_repository.py

@@ -12,18 +12,22 @@ class InMemoryMultiqueueRepository(MultiqueueRepository[K, R]):
         self.__queues = cast(dict[K, Queue[R]], {})
         self.__last_key_index = 0
 
-    def clear(self, key: K) -> None:
-        self.__queues.pop(key, None)
+    def create_key(self, key: K) -> None:
+        self.__queues[key] = Queue[R]()
+
+    def remove_key(self, key: K) -> None:
+        if key not in self.__queues:
+            return
+
+        self.__queues.pop(key)
 
     def push(self, key: K, record: R) -> None:
         if key not in self.__queues:
-            self.__add_key(key)
+            return
+
         queue = self.__queues[key]
         queue.put(record)
 
-    def __add_key(self, key: K) -> None:
-        self.__queues[key] = Queue[R]()
-
     def pop(self) -> Optional[R]:
         if not self.__queues:
             return None

buz/queue/multiqueue_repository.py

@@ -7,7 +7,11 @@ R = TypeVar("R")
 
 class MultiqueueRepository(ABC, Generic[K, R]):
     @abstractmethod
-    def clear(self, key: K) -> None:
+    def create_key(self, key: K) -> None:
+        pass
+
+    @abstractmethod
+    def remove_key(self, key: K) -> None:
         pass
 
     @abstractmethod

{buz-2.15.10rc5.dist-info → buz-2.15.10rc6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.15.10rc5
+Version: 2.15.10rc6
 Summary: Buz is a set of light, simple and extensible implementations of event, command and query buses.
 License: MIT
 Author: Luis Pintado Lozano
@@ -22,7 +22,7 @@ Classifier: Typing :: Typed
 Provides-Extra: aiokafka
 Provides-Extra: kombu
 Provides-Extra: pypendency
-Requires-Dist: aiohttp (>=3.11.13,<4.0.0)
+Requires-Dist: aiohttp (>=3.12.15,<4.0.0)
 Requires-Dist: aiokafka[lz4] (==0.12.0) ; extra == "aiokafka"
 Requires-Dist: asgiref (>=3.8.1,<4.0.0) ; extra == "aiokafka"
 Requires-Dist: cachetools (>=5.5.0,<6.0.0)

{buz-2.15.10rc5.dist-info → buz-2.15.10rc6.dist-info}/RECORD

@@ -47,7 +47,7 @@ buz/event/exceptions/worker_execution_exception.py,sha256=6mgztvXOCG_9VZ_Jptkk72
 buz/event/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 buz/event/infrastructure/buz_kafka/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 buz/event/infrastructure/buz_kafka/async_buz_kafka_event_bus.py,sha256=SyLblUVlwWOaNfZzK7vL6Ee4m-85vZVCH0rjOgqVAww,4913
-buz/event/infrastructure/buz_kafka/base_buz_aiokafka_async_consumer.py,sha256=M5mdbht294CE6UYtH7wfV9BoZTJIMLdyO9Wvb94PG_Q,20867
+buz/event/infrastructure/buz_kafka/base_buz_aiokafka_async_consumer.py,sha256=guVuZUQjxR-1ECuHjJ7Z7DTtIs2E7BSmazuP73dxgcY,21013
 buz/event/infrastructure/buz_kafka/buz_aiokafka_async_consumer.py,sha256=DRe3u69LD7Yt9WjA_hK_PRznM08_Mz4hxC_4poppjck,6446
 buz/event/infrastructure/buz_kafka/buz_aiokafka_multi_threaded_consumer.py,sha256=-pJVJq3b2SFmPT7SNmdPhqN2o64Hsjwds-shQ-Y7ytg,6389
 buz/event/infrastructure/buz_kafka/buz_kafka_event_bus.py,sha256=ymRSvcYVgbVCPgHN6rMBVBHQ5heCSwCDl6EffyqGVX8,4601
@@ -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=E0BB5mNU7x9LOpPF_77YuFl7sIaIwDL4bnOG6o5h8uw,2194
+buz/kafka/infrastructure/aiokafka/rebalance/kafka_callback_rebalancer.py,sha256=0IMc_TH4cHjIQiBWOFei_yMx6GbhO58CS-daKzffly4,2259
 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
@@ -241,16 +241,16 @@ buz/query/synchronous/self_process/self_process_query_bus.py,sha256=pKGJxXBWtqU4
 buz/query/synchronous/synced_async/__init__.py,sha256=TdFmIBeFIpl3Tvmh_FJpJMXJdPdfRxOstVqnPUi23mo,125
 buz/query/synchronous/synced_async/synced_async_query_bus.py,sha256=WxXHeEl1Pnh5Yiui8oMJKIOdzhQBGq1yAnAEk_gniRg,470
 buz/queue/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-buz/queue/in_memory/in_memory_multiqueue_repository.py,sha256=2jFSeVVbliOV-6m7AhjSkTT2stgFuHmfvNfrPTyyTzs,1523
+buz/queue/in_memory/in_memory_multiqueue_repository.py,sha256=F-5Cw6Ny8OmcmYSHOBCiSdZZfcCNBK99otq0qIPQ2Wk,1568
 buz/queue/in_memory/in_memory_queue_repository.py,sha256=hUPuQRmbrzQtR5gv4XYCM9aFQHX0U3Q3lt0yi0MIPYk,505
-buz/queue/multiqueue_repository.py,sha256=Wc4OE_jDB4mLyyhWKnizXskvgYm2glxvGW2ez1oa9Zs,531
+buz/queue/multiqueue_repository.py,sha256=Lg5QiMOsT_ZfzuOokjKlNsTJGoicqA7dt5gT4C9ppL8,612
 buz/queue/queue_repository.py,sha256=W3_lkosFu71XoiqRTT7An4kVjJQ3w0fZmFlaAfHeraM,385
 buz/serializer/message_to_bytes_serializer.py,sha256=uQ6JSTn24mLQvN48alwVXa6lfgfgCJN5vSzo829VuBQ,332
 buz/serializer/message_to_json_bytes_serializer.py,sha256=RGZJ64t4t4Pz2FCASZZCv-2LiWnertC8scE9ZzQkBsU,764
 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.10rc5.dist-info/LICENSE,sha256=Jytu2S-2SPEgsB0y6BF-_LUxIWY7402fl0JSh36TLZE,1062
-buz-2.15.10rc5.dist-info/METADATA,sha256=_A4RF9Cj31ACjYG6USaqvQun5VK74cl1scb9wQkc7S8,1601
-buz-2.15.10rc5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-buz-2.15.10rc5.dist-info/RECORD,,
+buz-2.15.10rc6.dist-info/LICENSE,sha256=Jytu2S-2SPEgsB0y6BF-_LUxIWY7402fl0JSh36TLZE,1062
+buz-2.15.10rc6.dist-info/METADATA,sha256=hIQPwtZ0Kwc54AxWOljmfsUTbiAGGZt9OtWukGK1m1E,1601
+buz-2.15.10rc6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+buz-2.15.10rc6.dist-info/RECORD,,