34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
"""中间件模块"""
|
|
import asyncio
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
from starlette.requests import Request
|
|
from starlette.responses import Response
|
|
from ..Convention.Runtime.GlobalConfig import ProjectConfig
|
|
|
|
config = ProjectConfig()
|
|
MAX_CONCURRENT_REQUESTS = config.FindItem("max_concurrent_requests", 100)
|
|
|
|
class ConcurrencyLimitMiddleware(BaseHTTPMiddleware):
|
|
"""并发限制中间件 - 防止内存爆炸"""
|
|
|
|
def __init__(self, app, max_concurrent: int = MAX_CONCURRENT_REQUESTS):
|
|
super().__init__(app)
|
|
self.semaphore = asyncio.Semaphore(max_concurrent)
|
|
self.max_concurrent = max_concurrent
|
|
config.Log("Info", f"并发限制中间件已启用，最大并发数：{max_concurrent}")
|
|
|
|
async def dispatch(self, request: Request, call_next) -> Response:
|
|
"""处理请求"""
|
|
async with self.semaphore:
|
|
try:
|
|
response = await call_next(request)
|
|
return response
|
|
except Exception as e:
|
|
config.Log("Error", f"请求处理错误: {e}", exc_info=True)
|
|
return Response(
|
|
content='{"error": "Internal Server Error"}',
|
|
status_code=500,
|
|
media_type="application/json"
|
|
)
|
|
|
|
__all__ = ["ConcurrencyLimitMiddleware"] |