add PollDataChannel::poll_write_vectored (#7)

and PollDataChannel::set_read_buf_capacity

Author: melekes

crates/data/src/data_channel/data_channel_test.rs

@@ -627,8 +627,8 @@ async fn test_poll_data_channel() -> Result<()> {
     let dc1 = Arc::new(DataChannel::accept(&a1, Config::default()).await?);
     bridge_process_at_least_one(&br).await;
 
-    let mut poll_dc0 = PollDataChannel::new(dc0.clone());
-    let mut poll_dc1 = PollDataChannel::new(dc1.clone());
+    let mut poll_dc0 = PollDataChannel::new(dc0);
+    let mut poll_dc1 = PollDataChannel::new(dc1);
 
     sbuf[0..4].copy_from_slice(&1u32.to_be_bytes());
     let n = poll_dc0
@@ -650,8 +650,8 @@ async fn test_poll_data_channel() -> Result<()> {
         "data should match"
     );
 
-    dc0.close().await?;
-    dc1.close().await?;
+    poll_dc0.into_inner().close().await?;
+    poll_dc1.into_inner().close().await?;
     bridge_process_at_least_one(&br).await;
 
     close_association_pair(&br, a0, a1).await;

crates/data/src/data_channel/mod.rs

@@ -406,6 +406,11 @@ impl PollDataChannel {
     pub fn buffered_amount_low_threshold(&self) -> usize {
         self.poll_stream.buffered_amount_low_threshold()
     }
+
+    /// Set the capacity of the temporary read buffer (default: 8192).
+    pub fn set_read_buf_capacity(&mut self, capacity: usize) {
+        self.poll_stream.set_read_buf_capacity(capacity)
+    }
 }
 
 impl AsyncRead for PollDataChannel {
@@ -434,6 +439,14 @@ impl AsyncWrite for PollDataChannel {
     fn poll_shutdown(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<()>> {
         Pin::new(&mut self.poll_stream).poll_shutdown(cx)
     }
+
+    fn poll_write_vectored(
+        mut self: Pin<&mut Self>,
+        cx: &mut Context<'_>,
+        bufs: &[io::IoSlice<'_>],
+    ) -> Poll<io::Result<usize>> {
+        Pin::new(&mut self.poll_stream).poll_write_vectored(cx, bufs)
+    }
 }
 
 impl Clone for PollDataChannel {