swarm/src/handler: Add impl IntoIterator for MultiHandler (#2572)

libp2p · Mar 21, 2022 · d81ad52 · d81ad52
1 parent 6b5744c
commit d81ad52
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/swarm/CHANGELOG.md b/swarm/CHANGELOG.md
@@ -1,7 +1,9 @@
 # 0.35.0 [unreleased]
 
+- Add impl `IntoIterator` for `MultiHandler`. See [PR 2572].
 - Remove `Send` bound from `NetworkBehaviour`. See [PR 2535].
 
+[PR 2572]: https://github.com/libp2p/rust-libp2p/pull/2572/
 [PR 2535]: https://github.com/libp2p/rust-libp2p/pull/2535/
 
 # 0.34.0 [2022-02-22]

diff --git a/swarm/src/handler/multi.rs b/swarm/src/handler/multi.rs
@@ -319,6 +319,16 @@ where
     }
 }
 
+/// Split [`MultiHandler`] into parts.
+impl<K, H> IntoIterator for MultiHandler<K, H> {
+    type Item = <Self::IntoIter as Iterator>::Item;
+    type IntoIter = std::collections::hash_map::IntoIter<K, H>;
+
+    fn into_iter(self) -> Self::IntoIter {
+        self.handlers.into_iter()
+    }
+}
+
 /// A [`IntoConnectionHandler`] for multiple other `IntoConnectionHandler`s.
 #[derive(Clone)]
 pub struct IntoMultiHandler<K, H> {