add poll_send_data() methods

Author: SergioBenitez

h3/src/connection.rs

@@ -21,7 +21,7 @@ use crate::{
         varint::VarInt,
     },
     qpack,
-    quic::{self, SendStream as _},
+    quic::{self, SendStream},
     stream::{self, AcceptRecvStream, AcceptedRecvStream, BufRecvStream, UniStreamHeader},
     webtransport::SessionId,
 };
@@ -765,6 +765,13 @@ where
         Ok(())
     }
 
+    /// Send some data on the response body.
+    pub fn poll_send_data(&mut self, cx: &mut Context<'_>, buf: B) -> Poll<Result<(), Error>> {
+        ready!(self.stream.poll_ready(cx))?;
+        let frame = Frame::Data(buf);
+        Poll::Ready(Ok(self.stream.send_data(frame)?))
+    }
+
     /// Send a set of trailers to end the request.
     pub async fn send_trailers(&mut self, trailers: HeaderMap) -> Result<(), Error> {
         //= https://www.rfc-editor.org/rfc/rfc9114#section-4.2

h3/src/server.rs

@@ -745,6 +745,11 @@ where
         self.inner.send_data(buf).await
     }
 
+    /// Poll for data send from the client
+    pub fn poll_send_data(&mut self, cx: &mut Context<'_>, buf: B) -> Poll<Result<(), Error>> {
+        self.inner.poll_send_data(cx, buf)
+    }
+
     /// Stop a stream with an error code
     ///
     /// The code can be [`Code::H3_NO_ERROR`].