diff --git a/beacon_node/lighthouse_network/src/behaviour/mod.rs b/beacon_node/lighthouse_network/src/behaviour/mod.rs index 29f5197616..1276db5e7e 100644 --- a/beacon_node/lighthouse_network/src/behaviour/mod.rs +++ b/beacon_node/lighthouse_network/src/behaviour/mod.rs @@ -1088,6 +1088,10 @@ impl Behaviour { if let Some(event) = self.internal_events.pop_front() { match event { InternalBehaviourMessage::DialPeer(peer_id) => { + // For any dial event, inform the peer manager + let enr = self.discovery_mut().enr_of_peer(&peer_id); + self.peer_manager.inject_dialing(&peer_id, enr); + // Submit the event let handler = self.new_handler(); return Poll::Ready(NBAction::Dial { opts: DialOpts::peer_id(peer_id) diff --git a/beacon_node/lighthouse_network/src/service.rs b/beacon_node/lighthouse_network/src/service.rs index 03732e8e61..3ecd32f3d9 100644 --- a/beacon_node/lighthouse_network/src/service.rs +++ b/beacon_node/lighthouse_network/src/service.rs @@ -367,18 +367,7 @@ impl Service { return Libp2pEvent::ZeroListeners; } } - SwarmEvent::Dialing(peer_id) => { - // We require the ENR to inject into the peer db, if it exists. - let enr = self - .swarm - .behaviour_mut() - .discovery_mut() - .enr_of_peer(&peer_id); - self.swarm - .behaviour_mut() - .peer_manager_mut() - .inject_dialing(&peer_id, enr); - } + SwarmEvent::Dialing(_peer_id) => {} } } }