如何为异步函数提供Context？Tokio运行时获取该对象方法

首先要明确：poll_flush 是 AsyncWrite trait 中的底层轮询方法，只有在你手动实现异步 trait（比如 AsyncWrite、Future）时，才需要直接处理 Context。在日常的 async/await 代码中，你根本不需要手动获取或传递 Context，因为 Tokio runtime 会自动处理这部分逻辑。

以下分场景说明：

1. 当你实现 AsyncWrite trait 时 #

在 poll_flush（以及 poll_write、poll_shutdown）方法中，Context 是作为参数直接传入的，你可以直接使用它来调用其他底层轮询方法：

use tokio::io::{AsyncWrite, Poll};
use std::task::{Context, Poll as StdPoll};

struct CustomWriter;

impl AsyncWrite for CustomWriter {
    fn poll_write(
        self: std::pin::Pin<&mut Self>,
        cx: &mut Context<'_>,
        buf: &[u8],
    ) -> Poll<Result<usize, std::io::Error>> {
        // 自定义写入逻辑
        StdPoll::Ready(Ok(buf.len()))
    }

    fn poll_flush(
        self: std::pin::Pin<&mut Self>,
        cx: &mut Context<'_>, // 这里直接拿到Context参数
    ) -> Poll<Result<(), std::io::Error>> {
        // 如果你的自定义writer内部包装了另一个AsyncWrite实现，
        // 可以直接传递这个cx给它的poll_flush
        // 示例：假设内部有个inner_writer字段
        // AsyncWrite::poll_flush(std::pin::Pin::new(&mut self.inner_writer), cx)
        
        StdPoll::Ready(Ok(()))
    }

    fn poll_shutdown(
        self: std::pin::Pin<&mut Self>,
        cx: &mut Context<'_>,
    ) -> Poll<Result<(), std::io::Error>> {
        StdPoll::Ready(Ok(()))
    }
}

2. 当你手动实现 Future trait 时 #

如果你自己编写一个自定义 Future，其 poll 方法会接收 Context 参数，你可以在这个方法里调用 poll_flush 并传入该参数：

use std::future::Future;
use std::task::{Context, Poll};
use tokio::io::AsyncWrite;
use std::pin::Pin;

struct FlushTask<W> {
    writer: W,
}

impl<W: AsyncWrite + Unpin> Future for FlushTask<W> {
    type Output = Result<(), std::io::Error>;

    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
        // 将Context传递给poll_flush
        AsyncWrite::poll_flush(Pin::new(&mut self.writer), cx)
    }
}

3. 日常 async/await 代码中（无需手动处理 Context） #

如果你只是在普通的异步函数里需要刷新写入缓冲区，直接使用 AsyncWriteExt trait 提供的 flush 异步方法即可——这是 poll_flush 的高层封装，Tokio 会自动处理 Context 的传递：

use tokio::io::AsyncWriteExt;

#[tokio::main]
async fn main() -> Result<(), std::io::Error> {
    let mut file = tokio::fs::File::create("example.txt").await?;
    file.write_all(b"hello world").await?;
    // 直接await flush，无需手动处理Context
    file.flush().await?;
    Ok(())
}

总结 #

Context 是 Tokio runtime 在调度任务时自动传递给底层 poll 方法的参数，你不需要主动创建或获取它。只有在编写底层异步组件（实现 AsyncWrite、Future 等 trait）时，才会通过方法参数拿到它；日常业务代码用 async/await + 高层异步方法就足够了。

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