Изменения вебсокета

This commit is contained in:
urec56 2024-06-06 22:23:42 +05:00
parent 9e39125a99
commit 10d85fe612

View file

@ -1,6 +1,7 @@
from fastapi import FastAPI
from fastapi import FastAPI, Request
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from starlette.middleware.base import BaseHTTPMiddleware
from app.users.router import router as user_router
from app.chat.websocket import router as websocket_router
@ -31,3 +32,14 @@ app.add_middleware(
)
app.mount("/static", StaticFiles(directory="app/static"), name="static")
class AddHeaderMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request: Request, call_next):
response = await call_next(request)
response.headers["Sec-Websocket-Protocol"] = request.headers["Sec-Websocket-Protocol"]
return response
app.add_middleware(AddHeaderMiddleware)