jararaca 0.2.36__py3-none-any.whl → 0.2.37a0__py3-none-any.whl

jararaca/__init__.py

@@ -2,6 +2,14 @@ from importlib import import_module
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from jararaca.messagebus.bus_message_controller import (
+        ack,
+        nack,
+        reject,
+        retry,
+        retry_later,
+        use_bus_message_controller,
+    )
     from jararaca.microservice import AppContext, AppInterceptor
     from jararaca.observability.interceptor import ObservabilityInterceptor
     from jararaca.observability.providers.otel import OtelObservabilityProvider
@@ -47,7 +55,7 @@ if TYPE_CHECKING:
     from .core.providers import ProviderSpec, Token
     from .di import Container
     from .messagebus.decorators import MessageBusController, MessageHandler
-    from .messagebus.interceptors.publisher_interceptor import (
+    from .messagebus.interceptors.aiopika_publisher_interceptor import (
         AIOPikaConnectionFactory,
         MessageBusPublisherInterceptor,
     )
@@ -104,6 +112,12 @@ if TYPE_CHECKING:
     from .tools.app_config.interceptor import AppConfigurationInterceptor
 
     __all__ = [
+        "use_bus_message_controller",
+        "ack",
+        "nack",
+        "reject",
+        "retry",
+        "retry_later",
         "RPCRequestNetworkError",
         "FILTER_SORT_ENTITY_ATTR_MAP",
         "FilterModel",
@@ -196,6 +210,16 @@ if TYPE_CHECKING:
 __SPEC_PARENT__: str = __spec__.parent  # type: ignore
 # A mapping of {<member name>: (package, <module name>)} defining dynamic imports
 _dynamic_imports: "dict[str, tuple[str, str, str | None]]" = {
+    "use_bus_message_controller": (
+        __SPEC_PARENT__,
+        "messagebus.bus_message_controller",
+        None,
+    ),
+    "ack": (__SPEC_PARENT__, "messagebus.bus_message_controller", None),
+    "nack": (__SPEC_PARENT__, "messagebus.bus_message_controller", None),
+    "reject": (__SPEC_PARENT__, "messagebus.bus_message_controller", None),
+    "retry": (__SPEC_PARENT__, "messagebus.bus_message_controller", None),
+    "retry_later": (__SPEC_PARENT__, "messagebus.bus_message_controller", None),
     "RPCRequestNetworkError": (__SPEC_PARENT__, "rpc.http.decorators", None),
     "FILTER_SORT_ENTITY_ATTR_MAP": (__SPEC_PARENT__, "persistence.sort_filter", None),
     "FilterModel": (__SPEC_PARENT__, "persistence.sort_filter", None),

jararaca/messagebus/bus_message_controller.py

@@ -0,0 +1,67 @@
+from contextlib import contextmanager, suppress
+from contextvars import ContextVar
+from typing import Any, Generator, Protocol
+
+
+class BusMessageController(Protocol):
+
+    async def ack(self) -> None:
+        pass
+
+    async def nack(self) -> None:
+        pass
+
+    async def reject(self) -> None:
+        pass
+
+    async def retry(self) -> None:
+        pass
+
+    async def retry_later(self, delay: int) -> None:
+        pass
+
+
+bus_message_controller_ctxvar = ContextVar[BusMessageController](
+    "bus_message_controller"
+)
+
+
+@contextmanager
+def provide_bus_message_controller(
+    controller: BusMessageController,
+) -> Generator[None, Any, None]:
+    token = bus_message_controller_ctxvar.set(controller)
+    try:
+        yield
+    finally:
+        with suppress(LookupError):
+            bus_message_controller_ctxvar.reset(token)
+
+
+def use_bus_message_controller() -> BusMessageController:
+    return bus_message_controller_ctxvar.get()
+
+
+async def ack() -> None:
+    controller = use_bus_message_controller()
+    await controller.ack()
+
+
+async def nack() -> None:
+    controller = use_bus_message_controller()
+    await controller.nack()
+
+
+async def reject() -> None:
+    controller = use_bus_message_controller()
+    await controller.reject()
+
+
+async def retry() -> None:
+    controller = use_bus_message_controller()
+    await controller.retry()
+
+
+async def retry_later(delay: int) -> None:
+    controller = use_bus_message_controller()
+    await controller.retry_later(delay)

jararaca/messagebus/worker.py

@@ -14,6 +14,10 @@ from pydantic import BaseModel
 from jararaca.core.uow import UnitOfWorkContextProvider
 from jararaca.di import Container
 from jararaca.lifecycle import AppLifecycle
+from jararaca.messagebus.bus_message_controller import (
+    BusMessageController,
+    provide_bus_message_controller,
+)
 from jararaca.messagebus.decorators import (
     MESSAGE_HANDLER_DATA_SET,
     MessageBusController,
@@ -170,7 +174,6 @@ class MessageHandlerCallback:
         async with self.consumer.lock:
             task = asyncio.create_task(self.handle_message(aio_pika_message))
             self.consumer.tasks.add(task)
-            # task.add_done_callback(self.tasks.discard)
             task.add_done_callback(self.handle_message_consume_done)
 
     def handle_message_consume_done(self, task: asyncio.Task[Any]) -> None:
@@ -273,7 +276,10 @@ class MessageHandlerCallback:
                 ctx = none_context()
             async with ctx:
                 try:
-                    await handler(builded_message)
+                    with provide_bus_message_controller(
+                        AioPikaMessageBusController(aio_pika_message)
+                    ):
+                        await handler(builded_message)
                     if not incoming_message_spec.auto_ack:
                         await aio_pika_message.ack()
                 except BaseException as base_exc:
@@ -374,3 +380,23 @@ class MessageBusWorker:
                 signal.SIGINT, on_shutdown, runner.get_loop()
             )
             runner.run(self.start_async())
+
+
+class AioPikaMessageBusController(BusMessageController):
+    def __init__(self, aio_pika_message: aio_pika.abc.AbstractIncomingMessage):
+        self.aio_pika_message = aio_pika_message
+
+    async def ack(self) -> None:
+        await self.aio_pika_message.ack()
+
+    async def nack(self) -> None:
+        await self.aio_pika_message.nack()
+
+    async def reject(self) -> None:
+        await self.aio_pika_message.reject()
+
+    async def retry(self) -> None:
+        await self.aio_pika_message.reject(requeue=True)
+
+    async def retry_later(self, delay: int) -> None:
+        raise NotImplementedError("Not implemented")

{jararaca-0.2.36.dist-info → jararaca-0.2.37a0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jararaca
-Version: 0.2.36
+Version: 0.2.37a0
 Summary: A simple and fast API framework for Python
 Home-page: https://github.com/LuscasLeo/jararaca
 Author: Lucas S

{jararaca-0.2.36.dist-info → jararaca-0.2.37a0.dist-info}/RECORD

@@ -1,7 +1,7 @@
 LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
 README.md,sha256=mte30I-ZEJJp-Oax-OganNgl6G9GaCZPL6JVFAvZGz4,7034
-pyproject.toml,sha256=J8jl9rhTU5wrfQYdGUqML72igUL3qR8in2aLGmaSRwY,1837
-jararaca/__init__.py,sha256=MbSXT_kddJ4hUNI43u3stzFzjG3ctkJewriZ6_NvaII,14019
+pyproject.toml,sha256=eCYmaQ1KvrC-AjVxBZKuaneOlX0K4k8xixQojXEJt2Q,1839
+jararaca/__init__.py,sha256=1Tv-cJ5C_4VL73ceISjoqZQe7DBAPN2G14VWnzlDxuY,14842
 jararaca/__main__.py,sha256=-O3vsB5lHdqNFjUtoELDF81IYFtR-DSiiFMzRaiSsv4,67
 jararaca/cli.py,sha256=JKk4xrRbtX2fM8yYw794lbxvJFH73bWw3GGIvrpAkeE,5706
 jararaca/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -12,12 +12,13 @@ jararaca/di.py,sha256=h3IsXdYZjJj8PJfnEDn0ZAwdd4EBfh8jU-wWO8ko_t4,76
 jararaca/files/entity.py.mako,sha256=tjQ-1udAMvVqgRokhsrR4uy3P_OVnbk3XZ8X69ixWhE,3098
 jararaca/lifecycle.py,sha256=qKlzLQQioS8QkxNJ_FC_5WbmT77cNbc_S7OcQeOoHkI,1895
 jararaca/messagebus/__init__.py,sha256=Zdl74HcS9K0FW6XUt7bVvaHEyxL8pWsqqakeRENIn0w,54
+jararaca/messagebus/bus_message_controller.py,sha256=Xd_qwnX5jUvgBTCarHR36fvtol9lPTsYp2IIGKyQQaE,1487
 jararaca/messagebus/decorators.py,sha256=GHlaXRuHtrz6R0HgcG2gJybpGYtdts9meDVSRPwN74I,4245
 jararaca/messagebus/interceptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jararaca/messagebus/interceptors/publisher_interceptor.py,sha256=rqAX644-VHarg6kPPsXGTmxYd79PpJXiJ1N1gIooJTo,2470
+jararaca/messagebus/interceptors/aiopika_publisher_interceptor.py,sha256=rqAX644-VHarg6kPPsXGTmxYd79PpJXiJ1N1gIooJTo,2470
 jararaca/messagebus/publisher.py,sha256=5ay9Znwybqt981OOykdWkFisSvGiTeTpPXDFLMnaiqg,1109
 jararaca/messagebus/types.py,sha256=iYLyLxWqOHkDadxyMqQPWy3itLNQfvD6oQe8jcq9nzo,887
-jararaca/messagebus/worker.py,sha256=aNp9tO9DinwcpdYPAA-e9lm3Y9dzH-dxx_4QYZUt9H8,12688
+jararaca/messagebus/worker.py,sha256=_K71_95XYv5FKiL84Us4uVTYfWNTk4mqVS6JnxxumbE,13530
 jararaca/microservice.py,sha256=1TvDKVMMREH27Ly8eTEheMmSfro4_Az_JKM_NdDvrgc,6636
 jararaca/observability/decorators.py,sha256=XffBinFXdiNkY6eo8_1nkr_GapM0RUGBg0aicBIelag,2220
 jararaca/observability/interceptor.py,sha256=GHkuGKFWftN7MDjvYeGFGEPnuJETNhtxRK6yuPrCrpU,1462
@@ -58,8 +59,8 @@ jararaca/tools/app_config/decorators.py,sha256=-ckkMZ1dswOmECdo1rFrZ15UAku--txaN
 jararaca/tools/app_config/interceptor.py,sha256=nfFZiS80hrbnL7-XEYrwmp2rwaVYBqxvqu3Y-6o_ov4,2575
 jararaca/tools/metadata.py,sha256=7nlCDYgItNybentPSSCc2MLqN7IpBd0VyQzfjfQycVI,1402
 jararaca/tools/typescript/interface_parser.py,sha256=l-QyPVntATcbL4JYm48xq2gNWfV1y2iArvOuIueFi8w,28829
-jararaca-0.2.36.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-jararaca-0.2.36.dist-info/METADATA,sha256=9vNEuIG1GmXeLV95m5_7VSLDcSlZBoXjy-uHNQtGb6A,8552
-jararaca-0.2.36.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-jararaca-0.2.36.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
-jararaca-0.2.36.dist-info/RECORD,,
+jararaca-0.2.37a0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+jararaca-0.2.37a0.dist-info/METADATA,sha256=LOL1VDz7acD53uTKUnVmodyCSKxiaZs_IBvCaUFJhDA,8554
+jararaca-0.2.37a0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+jararaca-0.2.37a0.dist-info/entry_points.txt,sha256=WIh3aIvz8LwUJZIDfs4EeH3VoFyCGEk7cWJurW38q0I,45
+jararaca-0.2.37a0.dist-info/RECORD,,

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "jararaca"
-version = "0.2.36"
+version = "0.2.37a0"
 description = "A simple and fast API framework for Python"
 authors = ["Lucas S <me@luscasleo.dev>"]
 readme = "README.md"