Дорога обучения FastAPI (44) WebSockets

Дорога обучения FastAPI (44)

2024-07-12

Наш предыдущий анализ был полностью основан на http-запросах, поэтому, если веб-сокеты могут его поддерживать, ответ — да, давайте посмотрим, как это реализовать.


from fastapi import WebSocket, FastAPI
from fastapi.responses import HTMLResponse
 
app = FastAPI()
 
html = """
 
<!DOCTYPE html>
 
<html>
 
    <head>
 
        <title>Chat</title>
 
    </head>
 
    <body>
 
        <h1>WebSocket 聊天</h1>
 
        <form action="" onsubmit="sendMessage(event)">
 
            <input type="text" id="messageText" autocomplete="off"/>
 
            <button>Send</button>
 
        </form>
 
        <ul id='messages'>
 
        </ul>
 
        <script>
 
            var ws = new WebSocket("ws://localhost:8000/ws");
 
            ws.onmessage = function(event) {
 
                var messages = document.getElementById('messages')
 
                var message = document.createElement('li')
 
                var content = document.createTextNode(event.data)
 
                message.appendChild(content)
 
                messages.appendChild(message)
 
            };
 
            function sendMessage(event) {
 
                var input = document.getElementById("messageText")
 
                ws.send(input.value)
 
                input.value = ''
 
                event.preventDefault()
 
            }
</script>
 
    </body>
 
</html>
 
"""
 
 
@app.get("/")
async def get_page():
    return HTMLResponse(html)
 
 
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        await websocket.send_text(f"接收到的消息是：{data}")

На самом деле это очень просто: мы пишем интерфейс для получения сообщений на сервере. Когда приходит сообщение, мы его обрабатываем по http, а теперь это вебсокет. Мы вернули полученное сообщение непосредственно во фронтенд.

Давайте посмотрим на эффект:

Таким образом мы просто реализуем функцию вебсокета. Итак, у нас есть конкретное сообщение, и мы закрываем чат, как с этим бороться?


# 接收特定的消息后，终止该聊天
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        if data == "bye":
            await websocket.send_text(f"请注意：窗口即将关闭")
            await websocket.send_text("窗口已关闭")
            await websocket.close()
            break
        else:
            await websocket.send_text(f"接收到的消息是：{data}")

Код внешнего интерфейса не изменился, был изменен только внутренний код.

На самом деле, здесь очень просто реализовать связь через вебсокет.

Конечно, мы также можем использовать его в запросах.

Зависит от
Безопасность
Печенье
Заголовок
Путь
Запрос

Мы можем посмотреть на следующий пример простого написания бэкэнда.


# 使用多种参数，比如获取cookie等
from typing import Optional
from fastapi import Cookie, Query, status, Depends
 
 
async def get_cookie_or_token(
    websocket: WebSocket,
    session: Optional[str] = Cookie(None),
    token: Optional[str] = Query(None),
):
    if session is None and token is None:
        await websocket.close(code=status.WS_1008_POLICY_VIOLATION)
    return session or token
 
 
@app.websocket("item/ws")
async def websocket_endpoint(
    websocket: WebSocket,
    q: Optional[int] = None,
    cookie_or_token: str = Depends(get_cookie_or_token),
):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        await websocket.send_text(
            f"Session cookie or query token value is: {cookie_or_token}"
        )
        if q is not None:
            await websocket.send_text(f"parameter q is: {q}")
        await websocket.send_text(f"Message text was: {data}, for : {q}")

Хотя мы просто реализовали серверную часть, мы видим, что у fastapi есть хорошая поддержка веб-сокетов.

Обмен технологиями

Дорога обучения FastAPI (44)

Личный профиль

моя контактная информация