Rust异步：如何让Future确保仅调用最新的Waker？

核心问题拆解 #

你遇到的本质是多线程场景下，Future如何安全更新后台线程持有的Waker，同时遵守「仅调用最新Waker」的契约，以及poll()相关内存模型细节的模糊点。

无竞态实现方案 #

要解决这个问题，核心是借助原子同步原语来管理Waker的存储与访问，避免竞态：

• 使用线程安全的原子容器存储Waker，比如标准库的AtomicWaker（或基于Arc<Mutex<Option<Waker>>>的自定义实现），确保poll()线程和后台工作线程能安全地更新、读取Waker。
• 在poll()流程中：
  1. 先检查后台工作是否完成，若已完成则直接返回Ready。
  2. 若未完成，将当前传入的Waker更新到原子容器中（覆盖旧的Waker），然后返回Pending。
• 后台工作线程的逻辑：
  1. 完成部分/全部工作后，从原子容器中获取最新的Waker并调用wake()。
  2. 若工作仍未完成（比如需要等待外部事件），则后续唤醒时会使用原子容器中最新的Waker。

关于竞态场景的契约合规性 #

当poll()正在更新Waker时，后台线程调用旧Waker的情况，并不违反契约：

• Future的契约要求是「仅调用最新的Waker」，这里的「最新」是指调用Waker时，当前存在的最新Waker。旧Waker的调用发生在新Waker完成存储之前，此时旧Waker仍是当时的有效Waker，调用它只会触发调度器再次poll该Future，而再次poll会更新Waker为当前最新的，后续后台线程的唤醒操作就会使用新Waker。

内存模型的细节补充 #

Rust Future的poll()契约依赖Rust官方的内存模型（即happens-after关系）：

• 当poll()线程将新Waker存入原子容器时，这个写操作会与后台线程后续的读操作建立happens-after关系，确保后台线程能看到最新的Waker值。
• 虽然官方Future trait文档没有详细展开内存模型细节，但《Rustonomicon》中关于并发内存模型的章节，以及AtomicWaker的官方文档，都补充了这部分的实现细节——AtomicWaker正是通过原子操作和内存屏障，保证了Waker更新与唤醒之间的线程安全可见性。

总结 #

履行「仅调用最新Waker」契约的关键，是用原子同步原语管理Waker的生命周期，借助Rust内存模型的happens-after关系保证线程间的可见性。标准库的AtomicWaker已经封装了这些细节，直接使用即可避免手动实现的竞态问题。

内容的提问来源于stack exchange，提问作者Martin Geisse