FastAPI Backend Development Bootcamp · Урок

Реализация WebSockets в FastAPI

Научитесь добавлять конечные точки WebSocket в приложение FastAPI и обрабатывать подключения.

«Реализация WebSockets в FastAPI» — бесплатный урок FastAPI Backend Development Bootcamp на CoddyKit. Это урок 2 из 4. Ты можешь прочитать весь урок бесплатно ниже — а потом практиковать его прямо в браузере с встроенным редактором кода и ИИ-репетитором 24/7. Это часть пути обучения FastAPI Backend Development Bootcamp, и твой прогресс синхронизируется между веб-версией и приложением CoddyKit. Курс FastAPI Backend Development Bootcamp содержит 4 уроков всего.

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

Real-time with FastAPI WebSockets

Welcome to implementing WebSockets with FastAPI! WebSockets provide a persistent, full-duplex communication channel between a client and a server.

Full-duplex: Both client and server can send and receive messages simultaneously.
Real-time: Ideal for applications needing instant updates, like chat, live dashboards, or gaming.

FastAPI has excellent built-in support for WebSockets, leveraging Python's async/await features.

Defining a WebSocket Endpoint

Just like HTTP endpoints, you define WebSocket endpoints using a decorator. Instead of @app.get() or @app.post(), you use @app.websocket().

Your endpoint function must be async def and accept a websocket: WebSocket parameter. This WebSocket object is your primary tool for interaction.

The first step inside your function is always to await websocket.accept() to establish the connection.

Your First WebSocket Connection

Let's create our first WebSocket endpoint. This simple example accepts a connection, sends a welcome message, and then closes the connection.

Notice the @app.websocket("/ws") decorator and the async def function, which are key for WebSocket handling.

Try running this example and observe the connection:

from fastapi import FastAPI, WebSocket

app = FastAPI()

@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    await websocket.send_text("Welcome! Connection established.")
    await websocket.close()

Handling WebSocket Disconnections

WebSocket connections can be interrupted for various reasons (client closes tab, network error). It's crucial to handle these disconnections gracefully.

FastAPI raises a WebSocketDisconnect exception when a client disconnects. You can catch this exception to perform cleanup tasks, like removing the client from a list of active connections.

Wrap your WebSocket communication logic in a try...except WebSocketDisconnect block.

Receiving Messages from Clients

Once connected, your server can receive messages from the client. The WebSocket object provides methods for this:

await websocket.receive_text(): For receiving string data.
await websocket.receive_bytes(): For receiving binary data.
await websocket.receive_json(): For receiving JSON data (requires python-multipart).

These operations are asynchronous, so always use await.

Echoing Client Messages Back

This example demonstrates a simple 'echo' server. It accepts a connection, then continuously receives messages from the client and sends them back.

The while True loop keeps the connection open, and the try...except WebSocketDisconnect handles when the client leaves.

from fastapi import FastAPI, WebSocket, WebSocketDisconnect

app = FastAPI()

@app.websocket("/ws/echo")
async def websocket_echo(websocket: WebSocket):
    await websocket.accept()
    try:
        while True:
            data = await websocket.receive_text()
            print(f"Received from client: {data}")
            await websocket.send_text(f"Server echoed: {data}")
    except WebSocketDisconnect:
        print("Client disconnected.")

Sending Messages to Clients

Similarly, your server can send messages to the client. The WebSocket object offers:

await websocket.send_text("your message"): To send string data.
await websocket.send_bytes(b"your bytes"): To send binary data.
await websocket.send_json({"key": "value"}): To send JSON data.

Remember to await these calls. It's common to send a response right after receiving a message.

Working with JSON Data

For structured data, sending and receiving JSON is preferred. FastAPI's WebSocket object handles serialization/deserialization for you.

Sending: Pass a Python dict to send_json().
Receiving: receive_json() returns a Python dict.

This simplifies data exchange compared to manually parsing strings.

Simple Connection Manager

For applications with multiple clients, you'll need to manage active connections. A common pattern is to use a class to keep track of all connected WebSocket objects.

This manager can then be used to add/remove connections and send messages to specific clients or broadcast to all.

from fastapi import WebSocket

class ConnectionManager:
    def __init__(self):
        self.active_connections: list[WebSocket] = []

    async def connect(self, websocket: WebSocket):
        await websocket.accept()
        self.active_connections.append(websocket)

    def disconnect(self, websocket: WebSocket):
        self.active_connections.remove(websocket)

    async def send_personal_message(self, message: str, websocket: WebSocket):
        await websocket.send_text(message)

# In a real app, you'd integrate 'manager' into your FastAPI endpoint
manager = ConnectionManager()

WebSocket Interaction Check

You've learned the basics of setting up and interacting with WebSocket connections in FastAPI.

Let's check your understanding of accepting new connections.

Recap: Implementing WebSockets

You've successfully learned how to implement WebSockets in FastAPI!

Use @app.websocket() to define endpoints.
Always await websocket.accept() to establish the connection.
Handle disconnections with try...except WebSocketDisconnect.
Use await websocket.receive_text() and await websocket.send_text() for communication.
For structured data, send_json() and receive_json() are your friends.

Next, you'll build a real-time chat application to put these concepts into practice!

Часто задаваемые вопросы

Урок «Реализация WebSockets в FastAPI» бесплатный?

Да — полный текст урока «Реализация WebSockets в FastAPI» бесплатно доступен здесь в веб-версии. Чтобы практиковать его интерактивно (встроенный редактор кода и ИИ-репетитор 24/7) и разблокировать остальной курс FastAPI Backend Development Bootcamp, подпишись на CoddyKit PRO. Курс FastAPI Backend Development Bootcamp содержит 4 уроков всего.

Чему я научусь в уроке «Реализация WebSockets в FastAPI»?

Научитесь добавлять конечные точки WebSocket в приложение FastAPI и обрабатывать подключения. Ты практикуешь FastAPI Backend Development Bootcamp с помощью реального кода, который запускаешь прямо в браузере, и ИИ-репетитор 24/7 отвечает на твои вопросы во время урока.

Нужен ли мне опыт, чтобы начать FastAPI Backend Development Bootcamp?

Предыдущий опыт не требуется. FastAPI Backend Development Bootcamp на CoddyKit структурирован для всех уровней — от новичков до продвинутых, поэтому ты можешь начать отсюда или с самого начала и учиться в своем темпе. Это урок 2 из 4.

Сколько времени занимает урок «Реализация WebSockets в FastAPI»?

Большинство уроков CoddyKit занимают около 5–10 минут. Каждый из них компактный и интерактивный, поэтому ты постоянно делаешь прогресс и продолжаешь с того же места в веб-версии и приложении.

Можно ли писать и запускать код в этом уроке FastAPI Backend Development Bootcamp?

Да. Каждый урок FastAPI Backend Development Bootcamp включает встроенный редактор кода, поэтому ты пишешь и запускаешь реальный код прямо в браузере и получаешь моментальную обратную связь от AI — локальная установка не требуется.

Все уроки этого курса

Основы протокола WebSocket
Реализация WebSockets в FastAPI
Создание чата в реальном времени
Масштабирование WebSockets с помощью шины Pub/Sub

← Назад к FastAPI Backend Development Bootcamp