Add Frame::map_data (#89)

hyperium · Feb 24, 2023 · 7bf321a · 7bf321a
1 parent 619ba28
commit 7bf321a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/http-body/src/frame.rs b/http-body/src/frame.rs
@@ -31,6 +31,21 @@ impl<T> Frame<T> {
         }
     }
 
+    /// Maps this frame's data to a different type.
+    pub fn map_data<F, D>(self, f: F) -> Frame<D>
+    where
+        F: FnOnce(T) -> D,
+    {
+        match self.kind {
+            Kind::Data(data) => Frame {
+                kind: Kind::Data(f(data)),
+            },
+            Kind::Trailers(trailers) => Frame {
+                kind: Kind::Trailers(trailers),
+            },
+        }
+    }
+
     /// Returns whether this is a DATA frame.
     pub fn is_data(&self) -> bool {
         matches!(self.kind, Kind::Data(..))