Implement into_inner to get the underlying stream (#367)

Co-authored-by: Abhishek Kumar <[email protected]>

Author: hi-abhi

src/compat.rs

@@ -96,6 +96,11 @@ impl<S> AllowStd<S> {
             }
         }
     }
+
+    /// Returns the underlying stream.
+    pub(crate) fn into_inner(self) -> S {
+        self.inner
+    }
 }
 
 // Proxy Waker that we pass to the internal AsyncRead/Write of the

src/lib.rs

@@ -247,6 +247,11 @@ impl<S> WebSocketStream<S> {
         f(&mut self.inner)
     }
 
+    /// Consumes the `WebSocketStream` and returns the underlying stream.
+    pub fn into_inner(self) -> S {
+        self.inner.into_inner().into_inner()
+    }
+
     /// Returns a shared reference to the inner stream.
     pub fn get_ref(&self) -> &S
     where