feat(register): register the middleware

This commit is contained in:
YuehuCao 2025-09-11 22:27:05 +08:00
parent 6256b3377d
commit da75ba746c
2 changed files with 15 additions and 0 deletions

View File

@ -11,6 +11,7 @@ from webapi.providers import message_queue
from webapi.providers import exception_handler from webapi.providers import exception_handler
from webapi.providers import probes from webapi.providers import probes
from webapi.providers import metrics from webapi.providers import metrics
from webapi.providers import middleware
from .freeleaps_app import FreeleapsApp from .freeleaps_app import FreeleapsApp
from common.config.app_settings import app_settings from common.config.app_settings import app_settings
@ -22,6 +23,11 @@ def create_app() -> FastAPI:
app = FreeleapsApp() app = FreeleapsApp()
register_logger() register_logger()
# 1. Register middleware firstly
register(app, middleware)
# 2. Register other providers
register(app, exception_handler) register(app, exception_handler)
# Register probe APIs if enabled # Register probe APIs if enabled
if app_settings.PROBES_ENABLED: if app_settings.PROBES_ENABLED:

View File

@ -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)