Add map_protocol

libp2p · Oct 24, 2018 · d77f092 · d77f092
1 parent 4c5b51b
commit d77f092
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/core/src/nodes/protocols_handler.rs b/core/src/nodes/protocols_handler.rs
@@ -213,6 +213,26 @@ impl<TConnectionUpgrade, TOutboundOpenInfo, TCustom>
         }
     }
 
+    /// If this is `OutboundSubstreamRequest`, maps the protocol to another.
+    #[inline]
+    pub fn map_protocol<F, I>(
+        self,
+        map: F,
+    ) -> ProtocolsHandlerEvent<I, TOutboundOpenInfo, TCustom>
+    where
+        F: FnOnce(TConnectionUpgrade) -> I,
+    {
+        match self {
+            ProtocolsHandlerEvent::OutboundSubstreamRequest { upgrade, info } => {
+                ProtocolsHandlerEvent::OutboundSubstreamRequest {
+                    upgrade: map(upgrade),
+                    info,
+                }
+            }
+            ProtocolsHandlerEvent::Custom(val) => ProtocolsHandlerEvent::Custom(val),
+        }
+    }
+
     /// If this is `Custom`, maps the content to something else.
     #[inline]
     pub fn map_custom<F, I>(