P2p: improve cancelation in maintenance

2019-02-20 15:02:01 +01:00 · 2019-02-20 15:02:01 +01:00 · 359f2d6932
commit 359f2d6932
parent 07595c9e1f
1 changed files with 3 additions and 1 deletions
--- a/src/lib_p2p/p2p_maintenance.ml
+++ b/src/lib_p2p/p2p_maintenance.ml
@ -114,7 +114,9 @@ let rec try_to_contact
    else
      List.fold_left
        (fun acc point ->
-           P2p_pool.connect pool point >>= function
+           protect ~canceler:st.canceler begin fun () ->
+             P2p_pool.connect pool point
+           end >>= function
           | Ok _ -> acc >|= succ
           | Error _ -> acc)
        (Lwt.return 0)