TEMP: things that might change in #3

Author: Jean-Christophe BEGUE

src/quic/mod.rs

@@ -3,7 +3,7 @@
 //! This module includes traits and types meant to allow being generic over any
 //! QUIC implementation.
 
-use std::task::{self, Poll};
+use std::{pin::Pin, task::{self, Poll}};
 
 use bytes::Buf;
 
@@ -68,7 +68,7 @@ pub trait SendStream<B: Buf> {
     fn send_data(&mut self, data: B) -> Result<(), Self::Error>;
 
     /// Poll to finish the sending side of the stream.
-    fn poll_finish(&mut self) -> Poll<Result<(), Self::Error>>;
+    fn poll_finish(&mut self, cx: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>>;
 
     /// Send a QUIC reset code.
     fn reset(&mut self, reset_code: u64);
@@ -86,7 +86,7 @@ pub trait RecvStream {
     /// When the receive side will no longer receive more data (such as because
     /// the peer closed their sending side), this should return `None`.
     fn poll_data(
-        &mut self,
+        self: Pin<&mut Self>,
         cx: &mut task::Context<'_>,
     ) -> Poll<Result<Option<Self::Buf>, Self::Error>>;