使用FastAPI搭建WebSocket服务,前端通过JavaScript连接,实现双向实时通信。后端用@app.websocket定义接口,管理连接列表并广播消息;前端创建WebSocket实例,监听onmessage事件接收数据,调用send发送消息。配合HTML页面和输入交互,完成简单聊天功能。注意处理异常、清理连接及心跳保活,确保稳定性。

要在Python网页版中实现WebSocket通信,核心是使用支持异步处理的后端框架配合前端JavaScript建立长连接。WebSocket能实现客户端与服务器之间的双向实时通信,非常适合聊天应用、实时数据推送等场景。下面从后端搭建、前端连接到完整示例,一步步说明如何开发。
选择合适的Python后端框架
Python中常用的WebSocket支持框架有以下几种:
- FastAPI + Uvicorn:基于ASGI的现代框架,原生支持WebSocket,开发效率高
- Sanic:异步Web框架,性能优秀,语法类似Flask
- Tornado:老牌异步框架,自带WebSocket模块,适合独立部署
推荐使用FastAPI,因其文档清晰、集成简单,且可直接通过@app.websocket装饰器定义接口。
使用FastAPI搭建WebSocket服务端
安装依赖:
立即学习“Python免费学习笔记(深入)”;
pip install fastapi uvicorn websockets
编写后端代码(main.py):
ws.onmessage = function(event) {
const output = document.getElementById("output");
const msg = document.createElement("p");
msg.textContent = event.data;
output.appendChild(msg);
};
function sendMessage() {
const input = document.getElementById("message");
ws.send(input.value);
input.value = "";
}- 在try...finally中确保断开时清理连接列表
- 设置心跳机制防止连接超时关闭
- 对敏感数据做校验和过滤
- 生产环境使用Nginx反向代理时,确保配置了正确的Upgrade头
例如增强异常处理:
finally:
if websocket in active_connections:
active_connections.remove(websocket)基本上就这些。只要后端用FastAPI这类支持异步的框架开启WebSocket路径,前端用new WebSocket(url)连接并监听消息,就能实现双向实时通信。不复杂但容易忽略的是连接维护和错误恢复逻辑,上线前务必测试断网重连等情况。










