hyper_async_io/tokio/
asyncwrite.rs

1use crate::BodyWriter;
2
3impl tokio::io::AsyncWrite for BodyWriter {
4    fn poll_write(
5        self: std::pin::Pin<&mut Self>,
6        cx: &mut std::task::Context<'_>,
7        buf: &[u8],
8    ) -> std::task::Poll<Result<usize, std::io::Error>> {
9        let sender = &mut self.get_mut().sender;
10        match sender.poll_ready(cx) {
11            std::task::Poll::Ready(_) => {
12                match sender.try_send_data(hyper::body::Bytes::copy_from_slice(buf)) {
13                    Ok(_) => std::task::Poll::Ready(Ok(buf.len())),
14                    Err(_) => std::task::Poll::Pending,
15                }
16            }
17            std::task::Poll::Pending => std::task::Poll::Pending,
18        }
19    }
20    fn poll_flush(
21        self: std::pin::Pin<&mut Self>,
22        _cx: &mut std::task::Context<'_>,
23    ) -> std::task::Poll<Result<(), std::io::Error>> {
24        std::task::Poll::Ready(Ok(()))
25    }
26    fn poll_shutdown(
27        self: std::pin::Pin<&mut Self>,
28        _cx: &mut std::task::Context<'_>,
29    ) -> std::task::Poll<Result<(), std::io::Error>> {
30        std::task::Poll::Ready(Ok(()))
31    }
32}