Source file `log_service.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
open Base
module Sig = Log_service_sig

module Make () : Sig.SERVICE = struct
  include Logs

  let get_level () =
    let level = Sys.getenv "LOG_LEVEL" |> Option.map ~f:String.lowercase in
    match level with
    | Some "info" -> Info
    | Some "debug" -> Debug
    | Some "warning" -> Warning
    | Some "error" -> Error
    | _ -> Warning

  let lifecycle =
    Core.Container.Lifecycle.make "log"
      (fun ctx ->
        let log_level = Some (get_level ()) in
        Logs_fmt.reporter () |> set_reporter;
        set_level log_level;
        debug (fun m -> m "LOGGER: Logger set up");
        Lwt.return ctx)
      (fun _ -> Lwt.return ())
end