From 46c3f2640e580f0b4c88cec8c8ed1f60c8539aa0 Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Mon, 5 Feb 2024 18:54:33 +0600
Subject: [PATCH] Add IoBoxed::take() method (#292)

---
 ntex-io/CHANGES.md  |  4 ++++
 ntex-io/Cargo.toml  |  2 +-
 ntex-io/src/io.rs   |  9 ++++++++-
 ntex-io/src/seal.rs | 10 ++++++++++
 4 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/ntex-io/CHANGES.md b/ntex-io/CHANGES.md
index 1f25051d..3ddfa4a4 100644
--- a/ntex-io/CHANGES.md
+++ b/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
diff --git a/ntex-io/Cargo.toml b/ntex-io/Cargo.toml
index 429c9ae5..a3eb36b1 100644
--- a/ntex-io/Cargo.toml
+++ b/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"]
diff --git a/ntex-io/src/io.rs b/ntex-io/src/io.rs
index 8a079547..937bbe8c 100644
--- a/ntex-io/src/io.rs
+++ b/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);
     }
 }
diff --git a/ntex-io/src/seal.rs b/ntex-io/src/seal.rs
index 12a527f4..0dbb8886 100644
--- a/ntex-io/src/seal.rs
+++ b/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)