17
17
use fnv:: FnvHashMap ;
18
18
use futures:: prelude:: * ;
19
19
use libp2p:: Multiaddr ;
20
- use libp2p:: core:: nodes :: listeners :: ListenerId ;
20
+ use libp2p:: core:: connection :: { ConnectionId , ListenerId } ;
21
21
use libp2p:: core:: { ConnectedPoint , either:: EitherOutput , PeerId , PublicKey } ;
22
22
use libp2p:: swarm:: { IntoProtocolsHandler , IntoProtocolsHandlerSelect , ProtocolsHandler } ;
23
23
use libp2p:: swarm:: { NetworkBehaviour , NetworkBehaviourAction , PollParameters } ;
@@ -205,26 +205,15 @@ impl NetworkBehaviour for DebugInfoBehaviour {
205
205
}
206
206
}
207
207
208
- fn inject_node_event (
208
+ fn inject_event (
209
209
& mut self ,
210
210
peer_id : PeerId ,
211
+ connection : ConnectionId ,
211
212
event : <<Self :: ProtocolsHandler as IntoProtocolsHandler >:: Handler as ProtocolsHandler >:: OutEvent
212
213
) {
213
214
match event {
214
- EitherOutput :: First ( event) => self . ping . inject_node_event ( peer_id, event) ,
215
- EitherOutput :: Second ( event) => self . identify . inject_node_event ( peer_id, event) ,
216
- }
217
- }
218
-
219
- fn inject_replaced ( & mut self , peer_id : PeerId , closed_endpoint : ConnectedPoint , new_endpoint : ConnectedPoint ) {
220
- self . ping . inject_replaced ( peer_id. clone ( ) , closed_endpoint. clone ( ) , new_endpoint. clone ( ) ) ;
221
- self . identify . inject_replaced ( peer_id. clone ( ) , closed_endpoint, new_endpoint. clone ( ) ) ;
222
-
223
- if let Some ( entry) = self . nodes_info . get_mut ( & peer_id) {
224
- entry. endpoint = new_endpoint;
225
- } else {
226
- error ! ( target: "sub-libp2p" ,
227
- "Disconnected from node we were not connected to {:?}" , peer_id) ;
215
+ EitherOutput :: First ( event) => self . ping . inject_event ( peer_id, connection, event) ,
216
+ EitherOutput :: Second ( event) => self . identify . inject_event ( peer_id, connection, event) ,
228
217
}
229
218
}
230
219
@@ -285,9 +274,10 @@ impl NetworkBehaviour for DebugInfoBehaviour {
285
274
return Poll :: Ready ( NetworkBehaviourAction :: DialAddress { address } ) ,
286
275
Poll :: Ready ( NetworkBehaviourAction :: DialPeer { peer_id } ) =>
287
276
return Poll :: Ready ( NetworkBehaviourAction :: DialPeer { peer_id } ) ,
288
- Poll :: Ready ( NetworkBehaviourAction :: SendEvent { peer_id, event } ) =>
289
- return Poll :: Ready ( NetworkBehaviourAction :: SendEvent {
277
+ Poll :: Ready ( NetworkBehaviourAction :: NotifyHandler { peer_id, handler , event } ) =>
278
+ return Poll :: Ready ( NetworkBehaviourAction :: NotifyHandler {
290
279
peer_id,
280
+ handler,
291
281
event : EitherOutput :: First ( event)
292
282
} ) ,
293
283
Poll :: Ready ( NetworkBehaviourAction :: ReportObservedAddr { address } ) =>
@@ -314,9 +304,10 @@ impl NetworkBehaviour for DebugInfoBehaviour {
314
304
return Poll :: Ready ( NetworkBehaviourAction :: DialAddress { address } ) ,
315
305
Poll :: Ready ( NetworkBehaviourAction :: DialPeer { peer_id } ) =>
316
306
return Poll :: Ready ( NetworkBehaviourAction :: DialPeer { peer_id } ) ,
317
- Poll :: Ready ( NetworkBehaviourAction :: SendEvent { peer_id, event } ) =>
318
- return Poll :: Ready ( NetworkBehaviourAction :: SendEvent {
307
+ Poll :: Ready ( NetworkBehaviourAction :: NotifyHandler { peer_id, handler , event } ) =>
308
+ return Poll :: Ready ( NetworkBehaviourAction :: NotifyHandler {
319
309
peer_id,
310
+ handler,
320
311
event : EitherOutput :: Second ( event)
321
312
} ) ,
322
313
Poll :: Ready ( NetworkBehaviourAction :: ReportObservedAddr { address } ) =>
0 commit comments