@@ -39,7 +39,7 @@ use libp2p_swarm::{
39
39
ExpiredListenAddr , FromSwarm ,
40
40
} ,
41
41
ConnectionDenied , ConnectionId , ExternalAddresses , ListenAddresses , NetworkBehaviour ,
42
- NetworkBehaviourAction , PollParameters , THandler , THandlerInEvent , THandlerOutEvent ,
42
+ PollParameters , THandler , THandlerInEvent , THandlerOutEvent , ToSwarm ,
43
43
} ;
44
44
use std:: {
45
45
collections:: { HashMap , VecDeque } ,
@@ -208,9 +208,7 @@ pub struct Behaviour {
208
208
209
209
last_probe : Option < Instant > ,
210
210
211
- pending_actions : VecDeque <
212
- NetworkBehaviourAction < <Self as NetworkBehaviour >:: OutEvent , THandlerInEvent < Self > > ,
213
- > ,
211
+ pending_actions : VecDeque < ToSwarm < <Self as NetworkBehaviour >:: OutEvent , THandlerInEvent < Self > > > ,
214
212
215
213
probe_id : ProbeId ,
216
214
@@ -336,9 +334,7 @@ impl Behaviour {
336
334
} => {
337
335
if let Some ( event) = self . as_server ( ) . on_outbound_connection ( & peer, address) {
338
336
self . pending_actions
339
- . push_back ( NetworkBehaviourAction :: GenerateEvent ( Event :: InboundProbe (
340
- event,
341
- ) ) ) ;
337
+ . push_back ( ToSwarm :: GenerateEvent ( Event :: InboundProbe ( event) ) ) ;
342
338
}
343
339
}
344
340
ConnectedPoint :: Dialer {
@@ -399,9 +395,7 @@ impl Behaviour {
399
395
} ) ) ;
400
396
if let Some ( event) = self . as_server ( ) . on_outbound_dial_error ( peer_id, error) {
401
397
self . pending_actions
402
- . push_back ( NetworkBehaviourAction :: GenerateEvent ( Event :: InboundProbe (
403
- event,
404
- ) ) ) ;
398
+ . push_back ( ToSwarm :: GenerateEvent ( Event :: InboundProbe ( event) ) ) ;
405
399
}
406
400
}
407
401
@@ -441,7 +435,7 @@ impl NetworkBehaviour for Behaviour {
441
435
}
442
436
443
437
match self . inner . poll ( cx, params) {
444
- Poll :: Ready ( NetworkBehaviourAction :: GenerateEvent ( event) ) => {
438
+ Poll :: Ready ( ToSwarm :: GenerateEvent ( event) ) => {
445
439
let actions = match event {
446
440
request_response:: Event :: Message {
447
441
message : request_response:: Message :: Response { .. } ,
@@ -474,9 +468,7 @@ impl NetworkBehaviour for Behaviour {
474
468
match self . as_client ( ) . poll_auto_probe ( cx) {
475
469
Poll :: Ready ( event) => {
476
470
self . pending_actions
477
- . push_back ( NetworkBehaviourAction :: GenerateEvent ( Event :: OutboundProbe (
478
- event,
479
- ) ) ) ;
471
+ . push_back ( ToSwarm :: GenerateEvent ( Event :: OutboundProbe ( event) ) ) ;
480
472
continue ;
481
473
}
482
474
Poll :: Pending => { }
@@ -601,8 +593,7 @@ impl NetworkBehaviour for Behaviour {
601
593
}
602
594
}
603
595
604
- type Action =
605
- NetworkBehaviourAction < <Behaviour as NetworkBehaviour >:: OutEvent , THandlerInEvent < Behaviour > > ;
596
+ type Action = ToSwarm < <Behaviour as NetworkBehaviour >:: OutEvent , THandlerInEvent < Behaviour > > ;
606
597
607
598
// Trait implemented for `AsClient` and `AsServer` to handle events from the inner [`request_response::Behaviour`] Protocol.
608
599
trait HandleInnerEvent {
0 commit comments