buz 2.15.10rc5__tar.gz → 2.15.10rc6__tar.gz

{buz-2.15.10rc5 → buz-2.15.10rc6}/PKG-INFO

@@ -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 → buz-2.15.10rc6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "buz"
-version = "2.15.10-rc5"
+version = "2.15.10-rc6"
 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>"]
@@ -30,7 +30,7 @@ dacite = "^1.8.1"
 aiokafka = {version = "==0.12.0", extras = ["lz4"], optional = true}
 asgiref = {version = "^3.8.1", optional = true}
 cachetools = "^5.5.0"
-aiohttp = "^3.11.13"
+aiohttp = "^3.12.15"
 
 [tool.poetry.dev-dependencies]
 black = "^23.3"

{buz-2.15.10rc5 → buz-2.15.10rc6}/src/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-2.15.10rc5 → buz-2.15.10rc6}/src/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-2.15.10rc5 → buz-2.15.10rc6}/src/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-2.15.10rc5 → buz-2.15.10rc6}/src/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