buz 2.14.3__py3-none-any.whl → 2.15.0__py3-none-any.whl

buz/event/exceptions/subscribers_not_found_exception.py

@@ -1,11 +1,15 @@
 from typing import Optional, Sequence
 
-from buz.event import Event, Subscriber
+from buz.event import Event
+from buz.event.meta_subscriber import MetaSubscriber
 
 
 class SubscribersNotFoundException(Exception):
     def __init__(
-        self, event: Event, allowed_subscriber_fqns: set[str], event_subscribers: Optional[Sequence[Subscriber]] = None
+        self,
+        event: Event,
+        allowed_subscriber_fqns: set[str],
+        event_subscribers: Optional[Sequence[MetaSubscriber]] = None,
     ) -> None:
         self.event = event
         self.allowed_subscriber_fqns = allowed_subscriber_fqns

buz/event/infrastructure/kombu/kombu_consumer.py

@@ -1,3 +1,4 @@
+import asyncio
 from logging import Logger
 from typing import Optional, Callable, cast
 
@@ -5,11 +6,14 @@ from kombu import Connection, Queue, Consumer as MessageConsumer, Message
 from kombu.mixins import ConsumerMixin
 from kombu.transport.pyamqp import Channel
 
-from buz.event import Event, Subscriber
+from buz.event import Event
+from buz.event.async_subscriber import AsyncSubscriber
 from buz.event.infrastructure.kombu.allowed_kombu_serializer import AllowedKombuSerializer
 from buz.event.exceptions.term_signal_interruption_exception import TermSignalInterruptionException
+from buz.event.meta_subscriber import MetaSubscriber
 from buz.event.middleware import ConsumeMiddleware, ConsumeMiddlewareChainResolver
 from buz.event.middleware.exceptions.event_already_in_progress_exception import EventAlreadyInProgressException
+from buz.event.subscriber import Subscriber
 from buz.locator import Locator
 from buz.event.consumer import Consumer
 from buz.event.exceptions.event_restore_exception import EventRestoreException
@@ -27,7 +31,7 @@ class KombuConsumer(ConsumerMixin, Consumer):
         queues_mapping: QueueToSubscriberFqnMapping,
         serializer: AllowedKombuSerializer,
         prefetch_count: int,
-        locator: Locator[Event, Subscriber],
+        locator: Locator[Event, MetaSubscriber],
         logger: Logger,
         consume_retrier: Optional[ConsumeRetrier] = None,
         reject_callback: Optional[RejectCallback] = None,
@@ -83,7 +87,7 @@ class KombuConsumer(ConsumerMixin, Consumer):
         except Exception as exc:
             raise EventRestoreException(body, str(message)) from exc
 
-    def __subscribers(self, event: Event, allowed_subscriber_fqns: set[str]) -> list[Subscriber]:
+    def __subscribers(self, event: Event, allowed_subscriber_fqns: set[str]) -> list[MetaSubscriber]:
         event_subscribers = None
         try:
             event_subscribers = self.__locator.get(event)
@@ -97,22 +101,41 @@ class KombuConsumer(ConsumerMixin, Consumer):
         except Exception as exc:
             raise SubscribersNotFoundException(event, allowed_subscriber_fqns, event_subscribers) from exc
 
-    def __consume_event(self, message: Message, event: Event, subscribers: list[Subscriber]) -> None:
+    def __consume_event(self, message: Message, event: Event, meta_subscribers: list[MetaSubscriber]) -> None:
         try:
             if self.should_stop is True:
                 raise TermSignalInterruptionException()
 
-            for subscriber in subscribers:
+            for meta_subscriber in meta_subscribers:
+                # The problem here is that the chain resolver works with syncsubscribers, an asyncsubscriber would require of a async function,
+                # but we are using run-until-complete to run the async function, so we are not really using the async function, we are just running it as a sync function, so we can cast the asyncsubscriber to a subscriber
+                subscriber = cast(Subscriber, meta_subscriber)
                 self.__consume_middleware_chain_resolver.resolve(event, subscriber, self.__perform_consume)
             message.ack()
         except Exception as exc:
-            self.__on_consume_exception(message, event, subscribers, exc)
+            self.__on_consume_exception(message, event, meta_subscribers, exc)
 
-    def __perform_consume(self, event: Event, subscriber: Subscriber) -> None:
-        subscriber.consume(event)
+    def __perform_consume(self, event: Event, subscriber: MetaSubscriber) -> None:
+        if isinstance(subscriber, AsyncSubscriber):
+            self.__get_or_create_event_loop().run_until_complete(subscriber.consume(event))
+            return
+
+        if isinstance(subscriber, Subscriber):
+            return subscriber.consume(event)
+
+        raise TypeError(f"Subscriber {subscriber} is not an instance of Subscriber or AsyncSubscriber")
+
+    def __get_or_create_event_loop(self) -> asyncio.AbstractEventLoop:
+        try:
+            return asyncio.get_event_loop()
+        except RuntimeError:
+            self.__logger.warning("The current thread does not have an event loop attached. Creating new event loop.")
+            event_loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(event_loop)
+            return event_loop
 
     def __on_consume_exception(
-        self, message: Message, event: Event, subscribers: list[Subscriber], exception: Exception
+        self, message: Message, event: Event, subscribers: list[MetaSubscriber], exception: Exception
     ) -> None:
         if isinstance(exception, EventAlreadyInProgressException) or isinstance(
             exception, TermSignalInterruptionException
@@ -139,7 +162,7 @@ class KombuConsumer(ConsumerMixin, Consumer):
         self.__reject_message(message, event, subscribers, exception)
 
     def __reject_message(
-        self, message: Message, event: Event, subscribers: list[Subscriber], exception: Exception
+        self, message: Message, event: Event, subscribers: list[MetaSubscriber], exception: Exception
     ) -> None:
         message.reject()
         if self.__reject_callback is not None:

buz/event/middleware/__init__.py

@@ -1,3 +1,4 @@
+from buz.event.middleware.base_async_consume_middleware import BaseAsyncConsumeMiddleware
 from buz.event.middleware.publish_middleware import PublishMiddleware, PublishCallable
 from buz.event.middleware.base_publish_middleware import BasePublishMiddleware
 from buz.event.middleware.publish_middleware_chain_resolver import PublishMiddlewareChainResolver
@@ -14,4 +15,5 @@ __all__ = [
     "ConsumeCallable",
     "BaseConsumeMiddleware",
     "ConsumeMiddlewareChainResolver",
+    "BaseAsyncConsumeMiddleware",
 ]

buz/event/middleware/base_async_consume_middleware.py

@@ -0,0 +1,20 @@
+from abc import abstractmethod
+
+from buz.event import Event
+from buz.event.async_subscriber import AsyncSubscriber
+from buz.event.middleware.async_consume_middleware import AsyncConsumeMiddleware, AsyncConsumeCallable
+
+
+class BaseAsyncConsumeMiddleware(AsyncConsumeMiddleware):
+    async def on_consume(self, event: Event, subscriber: AsyncSubscriber, consume: AsyncConsumeCallable) -> None:
+        await self._before_on_consume(event, subscriber)
+        await consume(event, subscriber)
+        await self._after_on_consume(event, subscriber)
+
+    @abstractmethod
+    async def _before_on_consume(self, event: Event, subscriber: AsyncSubscriber) -> None:
+        pass
+
+    @abstractmethod
+    async def _after_on_consume(self, event: Event, subscriber: AsyncSubscriber) -> None:
+        pass

{buz-2.14.3.dist-info → buz-2.15.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.14.3
+Version: 2.15.0
 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.14.3.dist-info → buz-2.15.0.dist-info}/RECORD

@@ -41,7 +41,7 @@ buz/event/event_bus.py,sha256=QnvZD_bKsv628TZ58s5uLntLJCA9Ngir_YHOwb4_UaU,289
 buz/event/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 buz/event/exceptions/event_not_published_exception.py,sha256=gGEiRFGdKIS-VTKg8SN54vSS10WeSkgBhlO2Gpcll_0,215
 buz/event/exceptions/event_restore_exception.py,sha256=dYHp5i1E-VCUYYhOAVYR-eJfZ3CqPpR9gm1bZ1EFXfE,245
-buz/event/exceptions/subscribers_not_found_exception.py,sha256=eD1DGzLB22bEPnxZGZd5U0geTfBooB0LiIosEg5HzTo,690
+buz/event/exceptions/subscribers_not_found_exception.py,sha256=tTaAJaVZXJ-8CwDq_Jru0FYhFDkdprFp3itauv9XJ4k,760
 buz/event/exceptions/term_signal_interruption_exception.py,sha256=RkRRF0v_K9Hg48bMeCHlyvAsqCpSYTA5Wcdbz3BpjOQ,265
 buz/event/exceptions/worker_execution_exception.py,sha256=6mgztvXOCG_9VZ_Jptkk72kZtNWQ2CPuQ3TjXEWFE14,123
 buz/event/infrastructure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -69,7 +69,7 @@ buz/event/infrastructure/kombu/allowed_kombu_serializer.py,sha256=LQ6futYsInawTC
 buz/event/infrastructure/kombu/consume_strategy/__init__.py,sha256=6dnAv-bOxoDL31gQD1dErRocdJvkLHTgdqeb4S33eWc,302
 buz/event/infrastructure/kombu/consume_strategy/consume_strategy.py,sha256=Zsv7QVpZXRLYvlV2nRbSdSwT_FgEELLyzUxdT6DyX8Q,179
 buz/event/infrastructure/kombu/consume_strategy/queue_per_subscriber_consume_strategy.py,sha256=Vsa1uC7dwS3jJ-dp_lvrE-hVWnN91-ma8oVqdLuXHMo,786
-buz/event/infrastructure/kombu/kombu_consumer.py,sha256=7EhNo_YZdXNpoVIuSm7Thk-Kv-wh3LtFP256g-IzljA,6483
+buz/event/infrastructure/kombu/kombu_consumer.py,sha256=cCTlbGM3LomTsNu6YmWvXrA3Gsn8_LkM4USbmQ8uYDk,7842
 buz/event/infrastructure/kombu/kombu_event_bus.py,sha256=VSLBtamp-YOta4KyqmfXvDurvPiHZSL9QPCozMK3Qyw,4017
 buz/event/infrastructure/kombu/publish_strategy/__init__.py,sha256=96ssn7ydJwLXYoVyrhfGcwCpXr4_5Sl0DbN6UCoeNc8,315
 buz/event/infrastructure/kombu/publish_strategy/fanout_exchange_per_event_publish_strategy.py,sha256=Pw85A1oI-cPtzHCQTr0XHQjb7-u9LVmKR3eBIonHsUU,397
@@ -81,11 +81,12 @@ buz/event/infrastructure/models/consuming_task.py,sha256=GJvn6fGTN5ZQJaOuQCX17JP
 buz/event/infrastructure/queue/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 buz/event/meta_base_subscriber.py,sha256=IP2Siol98OmoeCoOISiaCJxgxJG2SCrhmbAN4t01aWg,698
 buz/event/meta_subscriber.py,sha256=JPhhRqHkDOBWhuqtPmseUtAoYde1OmTBViqVbLBhvME,359
-buz/event/middleware/__init__.py,sha256=1_33sdvRejCF4mHuKVkbldeJde6Y2jYtSrB5vMs0Rfo,773
+buz/event/middleware/__init__.py,sha256=vbmskMeXurTSgwXqPsRQBydHhNAYnbEoqFc1pqemI7Y,897
 buz/event/middleware/async_consume_middleware.py,sha256=314z7ZyhvQIvi90kEO0t-FlnHSyRjArk3RqKOdDE6bM,459
 buz/event/middleware/async_consume_middleware_chain_resolver.py,sha256=Hw75JAs5pyZVDi7-nD4I1nbUXjwYpHQW9PctafGS4ks,1193
 buz/event/middleware/async_publish_middleware.py,sha256=JIxbRx7HVf_Q1iEziN_5RKGVJ-Oen_f1c3OL9QLmoxE,358
 buz/event/middleware/async_publish_middleware_chain_resolver.py,sha256=Hqj8CRZXJD6h9KuJaKl88iToOFN7BijoatoDo66En8w,1016
+buz/event/middleware/base_async_consume_middleware.py,sha256=7yncBDiAJYgAhKvXVqkY0jf-dH4Ug-qwqzd_DQESmjg,795
 buz/event/middleware/base_consume_middleware.py,sha256=9G1jsr_Wm-avsTfWAndi5_tf4WW6dSLwQ3rN0-jc7AE,651
 buz/event/middleware/base_publish_middleware.py,sha256=vtM8oA4LZjbZn4omPy-cIAUxQQwL-_Xb4ScU85DwjMU,531
 buz/event/middleware/consume_middleware.py,sha256=BCcs1LgIc5YZx5nf7nE_bMuiiXxscqBE4LqD-nd4JS0,363
@@ -245,7 +246,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.14.3.dist-info/LICENSE,sha256=Jytu2S-2SPEgsB0y6BF-_LUxIWY7402fl0JSh36TLZE,1062
-buz-2.14.3.dist-info/METADATA,sha256=mIxvjnK8YV9oyOgKXUnEoF_1OfXv1DuzTguIlFX6z_g,1659
-buz-2.14.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-buz-2.14.3.dist-info/RECORD,,
+buz-2.15.0.dist-info/LICENSE,sha256=Jytu2S-2SPEgsB0y6BF-_LUxIWY7402fl0JSh36TLZE,1062
+buz-2.15.0.dist-info/METADATA,sha256=QGRmI-gxSAfhDPtg8oeGCNdgbEdiXS5SXCX3i3oEVzQ,1659
+buz-2.15.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+buz-2.15.0.dist-info/RECORD,,