Kako upravljaš WebSocketima u FastAPI-ju?

Question

Accepted Answer

FastAPI podržava **WebSockets** za sveobuhvatnu, dvosmjernu komunikaciju (chat, žive ažuriranja, obavijesti) preko `@app.websocket` dekoratora. Za razliku od HTTP-ovog model zahtjev-odgovor, WebSocket veza ostaje otvorena, što omogućava serveru i klijentu slobodno slati poruke.

## Osnovni WebSocket krajnji točka

```python
from fastapi import WebSocket, WebSocketDisconnect

@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()                  # accept the connection
    try:
        while True:
            data = await websocket.receive_text()    # receive a message from the client
            await websocket.send_text(f"Echo: {data}")  # send a message back
    except WebSocketDisconnect:
        print("client disconnected")           # handle disconnection (cleanup)
```

Uzorak: `accept()` prihvaća vezu, zatim petlja — `receive_text()`/`receive_json()` čita poruke, `send_text()`/`send_json()` ih šalje. Iznimka `WebSocketDisconnect` signalizira da je klijent otišao (gdje čistiš).

## Emitiranje na više klijenata (npr. chat)

```python
class ConnectionManager:
    def __init__(self): self.active: list[WebSocket] = []
    async def connect(self, ws): await ws.accept(); self.active.append(ws)
    def disconnect(self, ws): self.active.remove(ws)
    async def broadcast(self, msg):
        for conn in self.active:
            await conn.send_text(msg)          # send to ALL connected clients

manager = ConnectionManager()

@app.websocket("/chat")
async def chat(ws: WebSocket):
    await manager.connect(ws)
    try:
        while True:
            msg = await ws.receive_text()
            await manager.broadcast(msg)        # everyone gets the message
    except WebSocketDisconnect:
        manager.disconnect(ws)                  # remove on disconnect
```

Upravljanje popisom aktivnih veza omogućava ti da **emitiraš** svim klijentima — osnova chat-a i značajki živih ažuriranja.

## Napomena o skaliranju

```text
⚠️ WebSocket connections are STATEFUL and live on ONE server instance.
To scale across multiple instances, use a Redis Pub/Sub (or similar) backplane so
messages/broadcasts reach clients connected to OTHER instances.
(Like sessions, don't keep connection state only in one process when scaling horizontally.)
```

## Zašto je važno

WebSocketi omogućavaju **sveobuhvatne, dvosmierne** značajke — chat, žive obavijesti, prisutnost, suradničko uređivanje, live nadzorne ploče — što HTTP-ov model zahtjev-odgovor ne može pružiti (server može slati klijentima, ne samo odgovarati).

Razumijevanje WebSocket podrške FastAPI-ja je vrijedno za izgradnju modernih interaktivnih aplikacija: dekorator `@app.websocket` s accept/receive/send petljom pokriva osnovnu komunikaciju u stvarnom vremenu, a upravljanje aktivnim vezama omogućava **emitiranje** (jezgra značajki za više korisnika poput chat-a).

Asinkrona osnova FastAPI-ja je odličan odabor za WebSockete.

Kritično važna je **napomena o skaliranju**: WebSocket veze su s potporom za stanje i vezane na jednu instancu servera, pa skaliranje na više instanci zahtijeva **Redis Pub/Sub backplane** za preusmjeravanje poruka između instanci — česta greška jer (kao i kod sesija) stanje veze ne može biti u memoriji jednog procesa pri pokretanju više instanci.

Znanje osnovnog WebSocket uzorka, emitiranja, rukovanju prekidom veze i razmatranja skaliranja je važno znanje razine seniora za izgradnju sveobuhvatnih, interaktivnih iskustava u stvarnom vremenu koja se sve više očekuju u modernim aplikacijama.