深入理解async-book项目中的Future Trait
2025-06-20 11:03:53作者:廉彬冶Miranda
什么是Future Trait
在Rust的异步编程模型中,Future
trait扮演着核心角色。它代表了一个异步计算过程,这个计算最终可能会产生一个值(也可能是空值()
)。理解Future
的工作原理对于掌握Rust异步编程至关重要。
Future的基本模型
让我们先看一个简化版的Future
trait定义:
trait SimpleFuture {
type Output;
fn poll(&mut self, wake: fn()) -> Poll<Self::Output>;
}
enum Poll<T> {
Ready(T),
Pending,
}
这个简化版本展示了Future
的核心机制:
- poll方法:驱动Future向前执行
- Poll枚举:表示当前Future的状态
Ready(T)
:计算完成,返回结果Pending
:计算尚未完成
Future的执行机制
poll
方法是Future
的核心,它的工作流程如下:
- 当Future可以继续执行时,返回
Poll::Ready(result)
- 当Future需要等待时,返回
Poll::Pending
并安排wake()
函数在Future可以继续时被调用 - 当
wake()
被调用时,执行器会再次调用poll
方法
这种机制的关键在于wake()
函数,它避免了执行器不断轮询所有Future的低效行为,而是精确知道哪些Future已经准备好继续执行。
实际应用示例
考虑一个从socket读取数据的场景:
struct SocketRead<'a> {
socket: &'a Socket,
}
impl SimpleFuture for SocketRead<'_> {
type Output = Vec<u8>;
fn poll(&mut self, wake: fn()) -> Poll<Self::Output> {
if self.socket.has_data_to_read() {
Poll::Ready(self.socket.read_buf())
} else {
self.socket.set_readable_callback(wake);
Poll::Pending
}
}
}
这个例子展示了:
- 有数据时立即返回
- 无数据时注册回调并返回Pending
- 数据到达时通过wake通知执行器
Future的组合
Future
的强大之处在于可以高效组合多个异步操作,无需额外分配内存:
并行执行多个Future
struct Join<FutureA, FutureB> {
a: Option<FutureA>,
b: Option<FutureB>,
}
impl<FutureA, FutureB> SimpleFuture for Join<FutureA, FutureB>
where
FutureA: SimpleFuture<Output = ()>,
FutureB: SimpleFuture<Output = ()>,
{
type Output = ();
fn poll(&mut self, wake: fn()) -> Poll<Self::Output> {
if let Some(a) = &mut self.a {
if let Poll::Ready(()) = a.poll(wake) {
self.a.take();
}
}
if let Some(b) = &mut self.b {
if let Poll::Ready(()) = b.poll(wake) {
self.b.take();
}
}
if self.a.is_none() && self.b.is_none() {
Poll::Ready(())
} else {
Poll::Pending
}
}
}
顺序执行多个Future
struct AndThenFut<FutureA, FutureB> {
first: Option<FutureA>,
second: FutureB,
}
impl<FutureA, FutureB> SimpleFuture for AndThenFut<FutureA, FutureB>
where
FutureA: SimpleFuture<Output = ()>,
FutureB: SimpleFuture<Output = ()>,
{
type Output = ();
fn poll(&mut self, wake: fn()) -> Poll<Self::Output> {
if let Some(first) = &mut self.first {
match first.poll(wake) {
Poll::Ready(()) => self.first.take(),
Poll::Pending => return Poll::Pending,
};
}
self.second.poll(wake)
}
}
这些组合方式展示了如何在不使用回调地狱的情况下构建复杂的异步控制流。
标准库中的Future Trait
实际的标准库Future
trait与我们的简化版有些重要区别:
trait Future {
type Output;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output>;
}
主要变化:
-
Pin<&mut Self>:使用固定指针确保Future在内存中不会移动,这对于实现自引用结构体(如async/await生成的Future)至关重要。
-
Context<'_>替代wake: fn():提供了更丰富的唤醒机制,特别是通过
Waker
类型可以精确唤醒特定任务,这对于管理大量并发连接的系统非常重要。
理解Pinning和Context
虽然本文不深入讨论pinning和Context的细节,但需要知道:
- Pinning:确保对象不会被移动,使得自引用结构体成为可能
- Context/Waker:提供了更精确的任务唤醒机制,替代简单的函数指针
总结
Future
trait是Rust异步编程的基石,它通过poll/wake机制实现了高效的异步操作。理解其工作原理有助于:
- 编写更高效的异步代码
- 理解async/await语法糖背后的机制
- 构建复杂的异步控制流
- 理解Rust异步生态系统的底层原理
通过本文的讲解,希望读者能够对async-book项目中Future
trait的设计和实现有更深入的理解,为掌握Rust异步编程打下坚实基础。
登录后查看全文
热门项目推荐
GLM-4.6
GLM-4.6在GLM-4.5基础上全面升级:200K超长上下文窗口支持复杂任务,代码性能大幅提升,前端页面生成更优。推理能力增强且支持工具调用,智能体表现更出色,写作风格更贴合人类偏好。八项公开基准测试显示其全面超越GLM-4.5,比肩DeepSeek-V3.1-Terminus等国内外领先模型。【此简介由AI生成】Jinja00- DDeepSeek-V3.2-ExpDeepSeek-V3.2-Exp是DeepSeek推出的实验性模型,基于V3.1-Terminus架构,创新引入DeepSeek Sparse Attention稀疏注意力机制,在保持模型输出质量的同时,大幅提升长文本场景下的训练与推理效率。该模型在MMLU-Pro、GPQA-Diamond等多领域公开基准测试中表现与V3.1-Terminus相当,支持HuggingFace、SGLang、vLLM等多种本地运行方式,开源内核设计便于研究,采用MIT许可证。【此简介由AI生成】Python00
openPangu-Ultra-MoE-718B-V1.1
昇腾原生的开源盘古 Ultra-MoE-718B-V1.1 语言模型Python00ops-transformer
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。C++0108AI内容魔方
AI内容专区,汇集全球AI开源项目,集结模块、可组合的内容,致力于分享、交流。02Spark-Chemistry-X1-13B
科大讯飞星火化学-X1-13B (iFLYTEK Spark Chemistry-X1-13B) 是一款专为化学领域优化的大语言模型。它由星火-X1 (Spark-X1) 基础模型微调而来,在化学知识问答、分子性质预测、化学名称转换和科学推理方面展现出强大的能力,同时保持了强大的通用语言理解与生成能力。Python00GOT-OCR-2.0-hf
阶跃星辰StepFun推出的GOT-OCR-2.0-hf是一款强大的多语言OCR开源模型,支持从普通文档到复杂场景的文字识别。它能精准处理表格、图表、数学公式、几何图形甚至乐谱等特殊内容,输出结果可通过第三方工具渲染成多种格式。模型支持1024×1024高分辨率输入,具备多页批量处理、动态分块识别和交互式区域选择等创新功能,用户可通过坐标或颜色指定识别区域。基于Apache 2.0协议开源,提供Hugging Face演示和完整代码,适用于学术研究到工业应用的广泛场景,为OCR领域带来突破性解决方案。00- HHowToCook程序员在家做饭方法指南。Programmer's guide about how to cook at home (Chinese only).Dockerfile010
- PpathwayPathway is an open framework for high-throughput and low-latency real-time data processing.Python00
项目优选
收起

deepin linux kernel
C
22
6

OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
213
2.21 K

暂无简介
Dart
521
115

Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
9
1

🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
978
578

本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
552
86

Ascend Extension for PyTorch
Python
65
94

React Native鸿蒙化仓库
JavaScript
209
285

openGauss kernel ~ openGauss is an open source relational database management system
C++
147
194

旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
1.02 K
399