Python中如何用FastAPI实现WebSocket即时通讯

酷雪大大_1354

酷雪大大_1354

2026-10-01

106人浏览

原创

fastapi websocket连接立即断开主因是未显式调用await websocket.accept()或重复调用:前者致客户端报403或1006错误,后者抛“already accepted”异常;必须且仅在接收/发送前调用一次accept()。

python中如何用fastapi实现websocket即时通讯

WebSocket连接为什么总在建立后立即断开

FastAPI的WebSocket对象必须显式调用accept(),否则客户端会收到1006错误并断连。常见写法是把accept()写在while True循环里——这会导致第二次accept()失败,直接断开连接。

正确做法是只调用一次,在接收/发送逻辑之前:

@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()  # ← 必须且仅此一次
    try:
        while True:
            data = await websocket.receive_text()
            await websocket.send_text(f"echo: {data}")
    except WebSocketDisconnect:
        pass
  • 漏掉await websocket.accept() → 客户端报错403 Forbidden或直接关闭
  • 在循环内重复调用accept() → 触发RuntimeError: Already accepted
  • 没包try/except WebSocketDisconnect → 断连时抛未捕获异常,日志刷屏

如何安全地广播消息给所有在线客户端

FastAPI本身不维护WebSocket连接池,需要自己用线程安全的数据结构(如set配合asyncio.Lock)管理活动连接。直接用全局list或dict在并发场景下会丢连接或引发RuntimeError: Set changed size during iteration。

推荐方案:用weakref.WeakSet自动清理已断开的连接,避免内存泄漏:

from weakref import WeakSet
<p>active_connections = WeakSet()</p><p>@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
active_connections.add(websocket)
try:
while True:
data = await websocket.receive_text()</p><h1>广播给其他人(不含自己)</h1><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">        for conn in active_connections:
            if conn != websocket:
                await conn.send_text(data)
except WebSocketDisconnect:
    pass
  • 用list或<code>dict手动管理 → 需要额外逻辑清理断连,易出错
  • 没做conn != websocket判断 → 自己发的消息又回传给自己,形成回环
  • 广播时未加try/except → 某个连接已断但尚未被WeakSet清理,send_text()抛WebSocketDisconnect中断整个广播

为什么websocket.receive_json()总是报JSONDecodeError

客户端发来的数据未必是合法JSON字符串,比如空格、换行、BOM头或二进制帧混入。FastAPI默认不校验帧类型,receive_json()遇到非文本帧(如Binary)会直接崩溃。

python-code-analyz
python-code-analyz

专业Python代码分析与优化,支持语法检查、安全扫描、性能评估、复杂度分析及重构后优化代码生成。

下载

稳妥做法是先用receive()拿到原始dict,再根据type字段分支处理:

msg = await websocket.receive()
if msg["type"] == "text":
    try:
        data = json.loads(msg["text"])
    except json.JSONDecodeError:
        await websocket.close(code=4000)  # 自定义错误码
        return
elif msg["type"] == "binary":
    # 处理二进制数据,例如protobuf或图片
    pass
  • 盲目用receive_json() → 任意非JSON文本(如" "或"\n")都触发异常
  • 忽略msg["type"] → 二进制帧传进来直接炸,无法恢复
  • 没设close(code=...) → 异常后连接挂起,占用资源

生产环境必须关掉的调试配置

开发时用uvicorn.run(..., reload=True)很方便,但上线后必须禁用reload和debug,否则:

  • reload=True → 文件监视器在多进程下失效,可能触发多次热重载,导致连接状态错乱
  • debug=True → 暴露完整堆栈、路径、环境变量,WebSocket握手阶段就可能泄露敏感信息
  • 没设timeout_keep_alive → 默认5秒,高延迟网络下频繁断连重连

上线启动命令应类似:

uvicorn main:app --host 0.0.0.0 --port 8000 \
  --workers 4 \
  --timeout-keep-alive 30 \
  --log-level warning

WebSocket长连接对超时和日志敏感,30秒保活比默认值更稳妥;warning级别可过滤大量无意义的INFO连接日志,避免IO瓶颈。

真正难的不是写通一条连接,而是让几百个并发连接稳定跑三天不掉、不出错、不拖垮内存——这些细节没压到每个await后面,就容易在凌晨三点被报警叫醒。

大量免费API接口:立即使用
涵盖生活服务API、金融科技API、企业工商API、等相关的API接口服务。免费API接口可安全、合规地连接上下游,为数据API应用能力赋能!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python websocket fastapi

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1631

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

3944

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1629

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

22697

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2787

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2827

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1123

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2183

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程