讲师中心 微信公众号
AI工具推荐 视频效率加速

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

酷雪大大_1354

酷雪大大_1354

发布时间:2026-10-01 06:34:37

|

106人浏览过

|

来源于php中文网

原创

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自动清理已断开的连接,避免内存泄漏:

立即学习“Python免费学习笔记(深入)”;

from weakref import WeakSet
<p>active_connections = WeakSet()</p><div class="aritcle_card flexRow">
                                                        <div class="artcardd flexRow">
                                                                <a class="aritcle_card_img" href="/xiazai/skill6081" title="python-code-analyz"><img
                                                                                src="https://img.php.cn/upload/skill/000/000/081/179077148379011.jpg" alt="python-code-analyz"  onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
                                                                <div class="aritcle_card_info flexColumn">
                                                                        <a href="/xiazai/skill6081" title="python-code-analyz">python-code-analyz</a>
                                                                        <p>专业Python代码分析与优化,支持语法检查、安全扫描、性能评估、复杂度分析及重构后优化代码生成。</p>
                                                                </div>
                                                                <a href="/xiazai/skill6081" title="python-code-analyz" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span> </a>
                                                        </div>
                                                </div><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</code></pre>
  • 用list或<code>dict手动管理 → 需要额外逻辑清理断连,易出错
  • 没做conn != websocket判断 → 自己发的消息又回传给自己,形成回环
  • 广播时未加try/except → 某个连接已断但尚未被WeakSet清理,send_text()抛WebSocketDisconnect中断整个广播

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

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

稳妥做法是先用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后面,就容易在凌晨三点被报警叫醒。

热门AI工具

更多
PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

相关专题

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

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

1631

2023.07.20

python能做什么
python能做什么

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

3944

2023.07.25

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

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

1629

2023.07.31

python教程
python教程

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

22697

2023.08.03

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

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

2787

2023.08.04

python eval
python eval

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

2827

2023.08.04

scratch和python区别
scratch和python区别

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

1123

2023.08.11

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

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

596

2023.08.10

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

0

2026.09.30

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn