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

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.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from starlette.middleware.base import BaseHTTPMiddleware
from app.users.router import router as user_router from app.users.router import router as user_router
from app.chat.websocket import router as websocket_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") 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)