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

如何使用Python asyncio实现一个简易的异步RPC框架?

秋婷大大_4136

秋婷大大_4136

发布时间:2026-06-13 09:10:40

|

870人浏览过

|

来源于php中文网

原创

在 asyncio 中,client 通过 request_id 关联 Future 实现非阻塞等待:发送前创建 Future 并存入字典,server 响应时调用 set_result();使用 StreamReader/StreamWriter 配合 JSON 行协议,连接级唯一 ID 防重,禁用 pickle 改用 JSON 序列化。

如何使用python asyncio实现一个简易的异步rpc框架?

asyncio 里怎么让 client 等 server 回应而不阻塞?

核心是用 asyncio.Queue 或 asyncio.Future 做请求-响应配对。不能靠全局 dict + while True 轮询,那会吃光 CPU;也不能直接用 await 一个未绑定事件的变量——得让每个请求生成唯一 request_id,server 返回时按 id 找到对应 Future 并 set_result()。

常见错误:client 发完请求就 await queue.get(),但没做超时或取消处理,一旦 server 挂了或丢包,client 就永远卡住。

  • client 发请求前,先创建 loop.create_future(),存进本地字典:futures[req_id] = future
  • 用 asyncio.wait_for(future, timeout=5.0) 包一层,避免无限等待
  • server 收到请求后,处理完必须调用 futures[req_id].set_result(response)(注意:id 必须和 client 一致)
  • client 的接收协程要持续监听 socket,收到响应后立即查表并 set_result,否则 future 永远不会完成

用 asyncio.StreamReader/StreamWriter 还是 asyncio.Protocol?

选 asyncio.StreamReader/asyncio.StreamWriter。它更轻、更可控,适合 RPC 这种「一问一答」模型;而 asyncio.Protocol 更适合长连接、多路复用或自定义状态机的场景(比如 HTTP/2),反而容易把简单问题搞复杂。

性能影响明显:Protocol 默认启用 buffer_size 和 flow control,对小包 RPC 反而增加延迟;StreamReader 则允许你精确控制每次 readexactly(n) 或 readuntil(b'\n'),方便解析定长头或分隔符协议。

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

Forge — Python Developer
Forge — Python Developer

Python开发助手 — 自动化脚本、API封装、命令行工具、数据管道、网页爬虫。经过测试的代码,包含错误处理和文档。当你需要...

下载
  • server 端用 async def handle_client(reader, writer),每个连接一个协程
  • client 端用 asyncio.open_connection(host, port),拿到 reader/writer 后直接读写
  • 协议建议用 JSON 行协议:每条消息 json.dumps({...}) + '\n',server 用 await reader.readuntil(b'\n') 解包,避免粘包
  • 别在 writer.write() 后立刻 close(),要等 await writer.drain(),否则可能丢最后几字节

如何保证 request_id 不重复且可跨连接追踪?

request_id 不能只靠 uuid.uuid4().hex,因为 client 多个并发请求时没问题,但 server 如果重启、或 client 重连,旧 id 可能还在 pending 队列里——导致响应错配。真正安全的做法是:client 每次新连接从 1 开始计数,加上时间戳或连接 ID 做前缀,例如 f"{conn_id}_{seq}"。

容易踩的坑:用 time.time_ns() 当 id,看似唯一,但高并发下纳秒级也可能撞;或者用 threading.local() 存 seq,但在 asyncio 中协程切换不触发 local 变量隔离,会导致混乱。

  • client 每建立一个新连接,生成唯一 conn_id = uuid.uuid4().hex[:8]
  • 每个连接内维护一个原子整数 self._seq = itertools.count(1)(注意不是全局 count)
  • 发请求时组合 id:req_id = f"{conn_id}_{next(self._seq)}"
  • server 不需要保存 conn_id,只需原样透传响应,client 靠完整 id 查自己的 futures 字典

为什么不能直接 pickle 对象传给 asyncio socket?

因为 pickle 不安全:反序列化任意数据可能执行任意代码;而且不同 Python 版本间不兼容,server 用 3.11、client 用 3.9 就可能解不开。RPC 框架必须面向异构环境,JSON 是底线。

有人试过 msgpack 或 protobuf,可以,但前提是 client/server 协议版本严格对齐。简易框架优先选 JSON——调试友好、浏览器能直接看、出错时内容可读。

  • 所有请求/响应体必须是 JSON-serializable:dict/list/str/int/float/None/bool
  • 函数名、参数、错误信息都走 JSON 字段,例如 {"method": "add", "params": [1,2], "id": "a1b2_3"}
  • 异常不能直接抛 Exception,要转成 {"error": {"code": -32603, "message": "division by zero"}}
  • 别在 JSON 里塞 bytes 或 datetime,先用 isoformat() 或 base64 编码
实际跑起来最常卡住的地方,是 client 收响应的协程没启动,或者 server 处理完忘记调 future.set_result()——这两处没有报错,只有超时,排查时容易绕远。建议第一步先加日志:每发一个请求打一行 f"sending {req_id}",每收到响应打 f"got {req_id}",再看哪边断了。

热门AI工具

更多
DeepSeek

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

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

蛙蛙写作

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

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

WorkBuddy

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

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

豆包大模型

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

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

相关专题

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

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

1771

2023.07.20

python能做什么
python能做什么

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

4624

2023.07.25

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

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

1749

2023.07.31

python教程
python教程

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

27317

2023.08.03

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

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

3307

2023.08.04

python eval
python eval

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

3307

2023.08.04

scratch和python区别
scratch和python区别

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

1203

2023.08.11

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

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

596

2023.08.10

Playwright 测试失败怎么办
Playwright 测试失败怎么办

Playwright测试失败专题汇集超时、定位器匹配、浏览器提前关闭、依赖与插件配置、文件上传、SSL及Docker启动等故障文章,并介绍通过截图、报告和Trace定位错误。

0

2026.10.21

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

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

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