File tree 2 files changed +17
-8
lines changed
2 files changed +17
-8
lines changed Original file line number Diff line number Diff line change 1
1
use std:: { fmt:: Debug , sync:: Arc } ;
2
2
3
3
use ahash:: { AHashMap , AHashSet } ;
4
- use anyhow:: Result ;
4
+ use anyhow:: { Context , Result } ;
5
5
use cid:: Cid ;
6
6
use derivative:: Derivative ;
7
7
use futures:: { future:: BoxFuture , FutureExt } ;
@@ -658,10 +658,15 @@ impl PeerManagerActor {
658
658
) {
659
659
if let Some ( session) = self . sessions . get ( & session) {
660
660
if session. peers . contains ( & peer) {
661
- self . network . protect_peer ( peer) . await ;
661
+ self . network
662
+ . protect_peer ( peer)
663
+ . await
664
+ . context ( "Failed to protect connection" )
665
+ . map_err ( |err| error ! ( "{err:#}" ) )
666
+ . ok ( ) ;
662
667
}
663
668
}
664
- let _ = response. send ( ( ) ) ;
669
+ response. send ( ( ) ) . ok ( ) ;
665
670
}
666
671
667
672
async fn remove_peer_from_session (
Original file line number Diff line number Diff line change @@ -297,18 +297,22 @@ impl Network {
297
297
trace ! ( "untag {}: {}" , peer, tag) ;
298
298
}
299
299
300
- pub async fn protect_peer ( & self , peer : PeerId ) {
300
+ pub async fn protect_peer ( & self , peer : PeerId ) -> Result < ( ) > {
301
301
trace ! ( "protect {}" , peer) ;
302
- let _ = self . network_out_sender . send ( OutEvent :: ProtectPeer { peer } ) ;
302
+ self . network_out_sender
303
+ . send ( OutEvent :: ProtectPeer { peer } )
304
+ . await ?;
305
+ Ok ( ( ) )
303
306
}
304
307
305
308
pub async fn unprotect_peer ( & self , peer : PeerId ) -> bool {
306
309
trace ! ( "unprotect {}" , peer) ;
307
310
308
311
let ( s, r) = oneshot:: channel ( ) ;
309
- let _ = self
310
- . network_out_sender
311
- . send ( OutEvent :: UnprotectPeer { peer, response : s } ) ;
312
+ self . network_out_sender
313
+ . send ( OutEvent :: UnprotectPeer { peer, response : s } )
314
+ . await
315
+ . ok ( ) ;
312
316
313
317
r. await . unwrap_or_default ( )
314
318
}
You can’t perform that action at this time.
0 commit comments