Fix legacy substream fallback not working (#6826)

* Fix legacy substream fallback not working * Make it nicer
paritytech · Aug 5, 2020 · 35f3ba4 · 35f3ba4
1 parent 5ce8f09
commit 35f3ba4
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/client/network/src/protocol/generic_proto/handler/group.rs b/client/network/src/protocol/generic_proto/handler/group.rs
@@ -600,15 +600,10 @@ impl ProtocolsHandler for NotifsHandler {
 						message
 					} => {
 						for (handler, _) in &mut self.out_handlers {
-							if handler.protocol_name() != &protocol_name[..] {
-								continue;
-							}
-
-							if handler.is_open() {
+							if handler.protocol_name() == &protocol_name[..] && handler.is_open() {
 								handler.send_or_discard(message);
+								continue 'poll_notifs_sink;
 							}
-
-							continue 'poll_notifs_sink;
 						}
 
 						self.legacy.inject_event(LegacyProtoHandlerIn::SendCustomMessage {