Source file query.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
(* Logging fuzz queries *)

type t = out_channel

(* Constructor *)

let setup_signal_handler (queries : t) =
  let handler _ =
    print_endline ">>> Caught interrupt, flushing queries";
    close_out queries;
    exit 1
  in
  Sys.set_signal Sys.sigint (Sys.Signal_handle handler)

let init (queryname : string) : t =
  let queries = open_out queryname in
  setup_signal_handler queries;
  queries

(* Logging *)

let query (queries : t) (msg : string) : unit =
  let timestamp =
    let tm = Unix.localtime (Unix.time ()) in
    Printf.sprintf "[%02d:%02d:%02d]" tm.tm_hour tm.tm_min tm.tm_sec
  in
  Format.asprintf "[%s] Query\n%s\n" timestamp msg |> output_string queries;
  flush queries

let answer (queries : t) (msg : string) : unit =
  let timestamp =
    let tm = Unix.localtime (Unix.time ()) in
    Printf.sprintf "[%02d:%02d:%02d]" tm.tm_hour tm.tm_min tm.tm_sec
  in
  Format.asprintf "[%s] Answer\n%s\n" timestamp msg |> output_string queries;
  flush queries

(* Closing *)

let close (queries : t) = close_out queries