From da75ba746c06b7a20255fa41aa403f76368905e9 Mon Sep 17 00:00:00 2001 From: YuehuCao Date: Thu, 11 Sep 2025 22:27:05 +0800 Subject: [PATCH] feat(register): register the middleware --- apps/notification/webapi/bootstrap/application.py | 6 ++++++ apps/notification/webapi/providers/middleware.py | 9 +++++++++ 2 files changed, 15 insertions(+) create mode 100644 apps/notification/webapi/providers/middleware.py diff --git a/apps/notification/webapi/bootstrap/application.py b/apps/notification/webapi/bootstrap/application.py index d98218d..65adca9 100644 --- a/apps/notification/webapi/bootstrap/application.py +++ b/apps/notification/webapi/bootstrap/application.py @@ -11,6 +11,7 @@ from webapi.providers import message_queue from webapi.providers import exception_handler from webapi.providers import probes from webapi.providers import metrics +from webapi.providers import middleware from .freeleaps_app import FreeleapsApp from common.config.app_settings import app_settings @@ -22,6 +23,11 @@ def create_app() -> FastAPI: app = FreeleapsApp() register_logger() + + # 1. Register middleware firstly + register(app, middleware) + + # 2. Register other providers register(app, exception_handler) # Register probe APIs if enabled if app_settings.PROBES_ENABLED: diff --git a/apps/notification/webapi/providers/middleware.py b/apps/notification/webapi/providers/middleware.py new file mode 100644 index 0000000..04ba64a --- /dev/null +++ b/apps/notification/webapi/providers/middleware.py @@ -0,0 +1,9 @@ +from webapi.middleware.api_key_middleware import NotificationServiceMiddleware + + +def register(app): + """ + Register middleware to FastAPI application + """ + # Register API Key middleware + app.add_middleware(NotificationServiceMiddleware) \ No newline at end of file