Use map_inbound_stream helper

libp2p · Jun 20, 2022 · 758800c · 758800c
1 parent 27638ac
commit 758800c
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/core/src/muxing/boxed.rs b/core/src/muxing/boxed.rs
@@ -50,16 +50,10 @@ where
         &self,
         cx: &mut Context<'_>,
     ) -> Poll<Result<StreamMuxerEvent<Self::Substream>, Self::Error>> {
-        let event = ready!(self.inner.poll_event(cx).map_err(into_io_error)?);
-
-        match event {
-            StreamMuxerEvent::AddressChange(a) => {
-                Poll::Ready(Ok(StreamMuxerEvent::AddressChange(a)))
-            }
-            StreamMuxerEvent::InboundSubstream(s) => {
-                Poll::Ready(Ok(StreamMuxerEvent::InboundSubstream(SubstreamBox::new(s))))
-            }
-        }
+        let event = ready!(self.inner.poll_event(cx).map_err(into_io_error)?)
+            .map_inbound_stream(SubstreamBox::new);
+
+        Poll::Ready(Ok(event))
     }
 
     #[inline]