poll_* methods to support custom futures implementations

Cargo.toml

@@ -59,6 +59,8 @@ bytes = "1.5.0"
 axum-core = { version = "0.4.3", optional = true }
 http = { version = "1", optional = true }
 async-trait = { version = "0.1", optional = true }
+tokio-util = { version = "0.7", features = ["codec", "io"] }
+futures = { version = "0.3", default-features = false, features = ["std"] }
 
 [features]
 default = ["simd"]

src/fragment.rs

@@ -105,7 +105,7 @@ impl<'f, S> FragmentCollector<S> {
   {
     loop {
       let (res, obligated_send) =
-        self.read_half.read_frame_inner(&mut self.stream).await;
+        self.read_half.read_frame(&mut self.stream).await;
       let is_closed = self.write_half.closed;
       if let Some(obligated_send) = obligated_send {
         if !is_closed {
@@ -179,7 +179,7 @@ impl<'f, S> FragmentCollectorRead<S> {
   {
     loop {
       let (res, obligated_send) =
-        self.read_half.read_frame_inner(&mut self.stream).await;
+        self.read_half.read_frame(&mut self.stream).await;
       if let Some(frame) = obligated_send {
         let res = send_fn(frame).await;
         res.map_err(|e| WebSocketError::SendError(e.into()))?;

src/frame.rs

@@ -138,7 +138,7 @@ pub struct Frame<'f> {
   pub payload: Payload<'f>,
 }
 
-const MAX_HEAD_SIZE: usize = 16;
+pub(crate) const MAX_HEAD_SIZE: usize = 16;
 
 impl<'f> Frame<'f> {
   /// Creates a new WebSocket `Frame`.
@@ -321,6 +321,9 @@ impl<'f> Frame<'f> {
   }
 
   /// Writes the frame to the buffer and returns a slice of the buffer containing the frame.
+  ///
+  /// This function will NOT append the frame to the Vec, but rather replace the current bytes
+  /// with the frame's serialized bytes.
   pub fn write<'a>(&mut self, buf: &'a mut Vec<u8>) -> &'a [u8] {
     fn reserve_enough(buf: &mut Vec<u8>, len: usize) {
       if buf.len() < len {
@@ -330,7 +333,7 @@ impl<'f> Frame<'f> {
     let len = self.payload.len();
     reserve_enough(buf, len + MAX_HEAD_SIZE);
 
-    let size = self.fmt_head(buf);
+    let size = self.fmt_head(&mut *buf);
     buf[size..size + len].copy_from_slice(&self.payload);
     &buf[..size + len]
   }