Add try_map_codec

mxinden · Sep 29, 2023 · b3b8633 · b3b8633
1 parent 5abbc48
commit b3b8633
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/framed.rs b/src/framed.rs
@@ -215,4 +215,19 @@ impl<T, U> FramedParts<T, U> {
             _priv: (),
         }
     }
+
+    /// Changes the codec used in this `FramedParts`.
+    pub fn try_map_codec<V, F, E>(self, f: F) -> Result<FramedParts<T, V>, E>
+    where
+        V: Encoder + Decoder,
+        F: FnOnce(U) -> Result<V, E>,
+    {
+        Ok(FramedParts {
+            io: self.io,
+            codec: f(self.codec)?,
+            read_buffer: self.read_buffer,
+            write_buffer: self.write_buffer,
+            _priv: (),
+        })
+    }
 }