How to implement websocket authentication with fastapi?

Question

for example:

async def get_cookie_or_token(
websocket: WebSocket,
session: Optional[str] = Cookie(None),
token: Optional[str] = Query(None),