Source file `env.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
open! Base
open! Import

type t =
  | Heartbeat_promotions
  | Heartbeat_interval_us

let to_string = function
  | Heartbeat_promotions -> "PARALLEL_HEARTBEAT_PROMOTIONS"
  | Heartbeat_interval_us -> "PARALLEL_HEARTBEAT_INTERVAL_US"
;;

let get ?(min = Int.min_value) ?(max = Int.max_value) t ~default =
  let i =
    match Sys.getenv (to_string t) with
    | None -> default
    | Some i -> Int.of_string i
  in
  if i < min then Printf.failwithf !"%{} < %d" t min ();
  if i > max then Printf.failwithf !"%{} > %d" t max ();
  i
;;

(* cf https://dl.acm.org/doi/pdf/10.1145/3632880 for the defaults here *)

let heartbeat_promotions = get Heartbeat_promotions ~default:15 ~min:0
let heartbeat_interval_us = get Heartbeat_interval_us ~default:250 ~min:1 ~max:999_999