主页 / user-23005998

narumi's questions

Martin Hope
narumi
Asked: 2024-11-05 02:51:24 +0800 CST
  • 9

我一直在 Rust 中研究 Future 实现，但遇到了一些我无法完全理解的行为。具体来说，我在方法std::sync::mpsc::Receiver内部使用poll，并尝试在while let和之间做出选择if let以接收消息。

这是我的轮询方法的简化版本：

fn poll(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Self::Output> {
    let this = self.get_mut();

    // Option 1: Using `while let`
    while let Ok(new_attributes) = this.receiver.try_recv() {
        // Process the message
    }

    // Option 2: Using `if let`
    if let Ok(new_attributes) = this.receiver.try_recv() {
        // Process the message
    }

    Poll::Pending
}

我观察到的情况

当我使用 while let 时，一切似乎都按预期工作：当新消息到达时，Future 会重新轮询。但是，当我使用 if let 时，Future 似乎挂起了，并且永远不会再次唤醒，即使有新消息可用。

我的理解

我知道 Waker 应该用于通知执行者应该重新轮询 Future。鉴于我的轮询函数始终返回 Poll::Pending，我预计 Future 将继续被重复轮询，除非进程本身停止。但是，我不明白为什么 while let 确保 Waker 被正确触发，而 if let 似乎没有这样做。

其他背景信息

未来是这样产生的：

ctx.task_executor()
    .spawn_critical_blocking("transaction execution service", Box::pin(fut_struct));

spawn_critical_blocking 函数的工作原理如下：

pub fn spawn_critical_blocking<F>(&self, name: &'static str, fut: F) -> JoinHandle<()>
where
    F: Future<Output = ()> + Send + 'static,
{
    self.spawn_critical_as(name, fut, TaskKind::Blocking)
}

在内部：

fn spawn_critical_as<F>(
    &self,
    name: &'static str,
    fut: F,
    task_kind: TaskKind,
) -> JoinHandle<()>
where
    F: Future<Output = ()> + Send + 'static,
{
    // Wrapping the future and handling task errors
    let task = std::panic::AssertUnwindSafe(fut)
        .catch_unwind()
        .map_err(...);

    let task = async move {
        // Handling task shutdown and execution
        let task = pin!(task);
        let _ = select(on_shutdown, task).await;
    };

    self.spawn_on_rt(task, task_kind)
}

然后该spawn_on_rt方法使用spawn_blocking。

我的问题

  • 为什么使用 while let 可以确保 Waker 被触发，但使用 if let
  • 做

不是？

  • 导致这种行为差异的根本机制是什么？

如有任何见解或解释我将不胜感激！

asynchronous