b6449cae87
* `lib_stdlib`: basic extended OCaml stdlib and generic data structures * `lib_data_encoding`: almost independant 'Data_encoding' * `lib_error_monad`: almost independant 'Error_monad' * `lib_stdlib_lwt`: extended Lwt library * `lib_crypto`: all the crypto stuff (hashing, signing, cryptobox). * `lib_base`: - basic type definitions (Block_header, Operation, ...) - a module `TzPervasives` to bind them all and to be the single module opened everywhere. In the process, I splitted `Tezos_data` and `Hash` in multiple submodules, thus removing a lot of `-open`. The following two modules may not have found their place yet: - Base58 (currently in `lib_crypto`) - Cli_entries (currently in `lib_stdlib_lwt`)
54 lines
1.7 KiB
OCaml
54 lines
1.7 KiB
OCaml
(**************************************************************************)
|
|
(* *)
|
|
(* Copyright (c) 2014 - 2017. *)
|
|
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
|
|
(* *)
|
|
(* All rights reserved. No warranty, explicit or implicit, provided. *)
|
|
(* *)
|
|
(**************************************************************************)
|
|
|
|
open Lwt.Infix
|
|
|
|
type t = {
|
|
cancelation: unit Lwt_condition.t ;
|
|
cancelation_complete: unit Lwt_condition.t ;
|
|
mutable cancel_hook: unit -> unit Lwt.t ;
|
|
mutable canceling: bool ;
|
|
mutable canceled: bool ;
|
|
}
|
|
|
|
let create () =
|
|
let cancelation = Lwt_condition.create () in
|
|
let cancelation_complete = Lwt_condition.create () in
|
|
{ cancelation ; cancelation_complete ;
|
|
cancel_hook = (fun () -> Lwt.return ()) ;
|
|
canceling = false ;
|
|
canceled = false ;
|
|
}
|
|
|
|
let cancel st =
|
|
if st.canceled then
|
|
Lwt.return ()
|
|
else if st.canceling then
|
|
Lwt_condition.wait st.cancelation_complete
|
|
else begin
|
|
st.canceling <- true ;
|
|
Lwt_condition.broadcast st.cancelation () ;
|
|
Lwt.finalize
|
|
st.cancel_hook
|
|
(fun () ->
|
|
st.canceled <- true ;
|
|
Lwt_condition.broadcast st.cancelation_complete () ;
|
|
Lwt.return ())
|
|
end
|
|
|
|
let on_cancel st cb =
|
|
let hook = st.cancel_hook in
|
|
st.cancel_hook <- (fun () -> hook () >>= cb)
|
|
|
|
let cancelation st =
|
|
if st.canceling then Lwt.return ()
|
|
else Lwt_condition.wait st.cancelation
|
|
|
|
let canceled st = st.canceling
|