package oraft

  1. Overview
  2. Docs

Source file timer.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
open Core
open Lwt

type t = {
  timeout_millis : int;
  logger : Logger.t;
  mutable timeout : Time_ns.t;
  mutable should_stop : bool;
}

let span timeout_millis =
  float_of_int ((timeout_millis / 2) + Random.int timeout_millis)
  |> Time_ns.Span.of_ms


let update t = t.timeout <- Time_ns.add (Time_ns.now ()) (span t.timeout_millis)

let is_timed_out t = Time_ns.( < ) t.timeout (Time_ns.now ())

let start t ~on_stop =
  let rec check_election_timeout () =
    if is_timed_out t || t.should_stop
    then (
      Logger.debug t.logger "Election_timer timed out";
      Lwt.return (on_stop ())
    )
    else Lwt_unix.sleep 0.05 >>= fun () -> check_election_timeout ()
  in
  check_election_timeout ()


let create ~logger ~timeout_millis =
  let t =
    {
      timeout_millis;
      logger;
      timeout = Time_ns.add (Time_ns.now ()) (span timeout_millis);
      should_stop = false;
    }
  in
  t


let stop t = t.should_stop <- true
OCaml

Innovation. Community. Security.