hyper_async_io/tokio/
asyncwrite.rs1use 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}