Merge pull request #6 from Ifechukwu001/feat/rabbitmq

Implemented the RabbitMQ service broker

Author: CaptainAril

.env.example

@@ -30,4 +30,6 @@ JWT_ISSUER=xxx
 OTP_LIFETIME=10
 
 API_GATEWAY_PUBLIC_KEY=xxxxxxxxxxxxxxxxxxxxxxxx
-API_KEY_EXPIRES_AT=2
+API_KEY_EXPIRES_AT=2
+
+RABBITMQ_HOST=localhost

README.md

@@ -44,5 +44,5 @@ The Swagger documentation for the application is hosted on [Render](https://df-u
 - Run the server locally on port 8000
 
   ```bash
-  uv run manage.py runserver 8000
+  uv run uvicorn src.config.asgi:application --reload --port 8000
   ```

pyproject.toml

@@ -19,7 +19,9 @@ dependencies = [
   "pydantic[email]>=2.10.6",
   "django-ninja>=1.3.0",
   "pyjwt>=2.10.1",
-  "typing>=3.10.0.0",
+  "faststream[rabbit]>=0.5.39",
+  "starlette>=0.46.2",
+  "uvicorn>=0.34.2",
 ]
 
 [dependency-groups]

src/api/services/external/RabbitMQRoutes.py

@@ -0,0 +1 @@
+

src/api/services/external/__init__.py

@@ -10,7 +10,22 @@
 import os
 
 from django.core.asgi import get_asgi_application
+from faststream.rabbit import RabbitBroker
+from starlette.routing import Mount
+from starlette.applications import Starlette
+
+from src.env import rabbitmq_config
+
+broker = RabbitBroker(host=rabbitmq_config["host"])
+
 
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "src.config.settings")
 
-application = get_asgi_application()
+application = Starlette(
+    routes=[Mount("/", get_asgi_application())],  # type: ignore
+    on_startup=[broker.start],
+    on_shutdown=[broker.close],
+)
+
+
+from src.api.services.external import RabbitMQRoutes as RabbitMQRoutes  # noqa: E402

src/config/asgi.py

@@ -50,6 +50,10 @@ class OTP(TypedDict):
     lifetime: int
 
 
+class RabbitMQ(TypedDict):
+    host: str
+
+
 env = Env()
 
 app: App = {
@@ -105,4 +109,17 @@ class OTP(TypedDict):
 
 otp: OTP = {"lifetime": get_env_int("OTP_LIFETIME")}
 
-__all__ = ["api_gateway", "app", "cache", "db", "env", "jwt_config", "log", "otp"]
+rabbitmq_config: RabbitMQ = {"host": get_env_str("RABBITMQ_HOST")}
+
+
+__all__ = [
+    "api_gateway",
+    "app",
+    "cache",
+    "db",
+    "env",
+    "jwt_config",
+    "log",
+    "otp",
+    "rabbitmq_config",
+]