P2P: Cancel MA computations on client disconnect

2016-11-30 14:48:21 +01:00 · 2016-11-30 14:48:21 +01:00 · 5f8b74e96c
commit 5f8b74e96c
parent 997d1972e0
1 changed files with 1 additions and 1 deletions
--- a/src/node/net/p2p.ml
+++ b/src/node/net/p2p.ml
@ -494,7 +494,7 @@ module Make (P: PARAMS) = struct
          let received = !counter in
          ma#add_int (received - old_received);
          inner received in
-        Lwt.async (fun () -> inner !counter)
+        Lwt.async (fun () -> Lwt.pick [cancelation (); inner !counter])
      in
      (* net object state *)
      let last = ref (Unix.gettimeofday ()) in