Add IoBoxed::take() method (#292)

ntex-io/CHANGES.md

@@ -1,5 +1,9 @@
 # Changes
 
+## [1.0.1] - 2024-02-05
+
+* Add IoBoxed::take() method
+
 ## [1.0.0] - 2024-01-09
 
 * Release

ntex-io/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex-io"
-version = "1.0.0"
+version = "1.0.1"
 authors = ["ntex contributors <team@ntex.rs>"]
 description = "Utilities for encoding and decoding frames"
 keywords = ["network", "framework", "async", "futures"]

ntex-io/src/io.rs

@@ -1017,7 +1017,7 @@ mod tests {
         let (client, server) = IoTest::create();
         let f = DropFilter { p: p.clone() };
         format!("{:?}", f);
-        let io = Io::new(server).add_filter(f);
+        let mut io = Io::new(server).add_filter(f);
 
         client.remote_buffer_cap(1024);
         client.write(TEXT);
@@ -1030,7 +1030,14 @@ mod tests {
         let buf = client.read().await.unwrap();
         assert_eq!(buf, Bytes::from_static(b"test"));
 
+        let io2 = io.take();
+        let mut io3: crate::IoBoxed = io2.into();
+        let io4 = io3.take();
+
         drop(io);
+        drop(io3);
+        drop(io4);
+
         assert_eq!(p.get(), 1);
     }
 }

ntex-io/src/seal.rs

@@ -15,6 +15,16 @@ impl fmt::Debug for Sealed {
 /// Boxed `Io` object with erased filter type
 pub struct IoBoxed(Io<Sealed>);
 
+impl IoBoxed {
+    #[inline]
+    /// Clone current io object.
+    ///
+    /// Current io object becomes closed.
+    pub fn take(&mut self) -> Self {
+        IoBoxed(self.0.take())
+    }
+}
+
 impl From<Io<Sealed>> for IoBoxed {
     fn from(io: Io<Sealed>) -> Self {
         Self(io)