Source file `mssql_error.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
open Core_kernel

type t =
  { msg : string
  ; reraised_exn : Exn.t option [@sexp.option]
  ; here : Source_code_position.t
  ; query : string option [@sexp.option]
  ; params : Db_field.t option list [@sexp.list]
  ; formatted_query : string option [@sexp.option]
  ; results : Row.t list list [@sexp.list]
  }
[@@deriving sexp_of]

exception Mssql_error of t [@@deriving sexp_of]

let make ~msg ?exn ~here ?query ?(params = []) ?formatted_query ?(results = []) () =
  Mssql_error { msg; reraised_exn = exn; here; query; params; formatted_query; results }
;;

let failwith ?query ?params ?formatted_query ?results ?exn ?backtrace here msg =
  let exn = make ~here ~msg ?query ?params ?formatted_query ?results ?exn () in
  match backtrace with
  | None -> raise exn
  | Some backtrace -> Caml.Printexc.raise_with_backtrace exn backtrace
;;

let failwithf ?query ?params ?formatted_query ?results ?exn ?backtrace here fmt =
  ksprintf
    (fun msg ->
      failwith ?query ?params ?formatted_query ?results ?exn ?backtrace here msg)
    fmt
;;

let with_wrap ?query ?(params = []) ?formatted_query ?(results = []) here f =
  try f () with
  | Mssql_error t ->
    let backtrace = Caml.Printexc.get_raw_backtrace () in
    let exn =
      (* Preserve original info if set, but override if not set *)
      (* Note that we never mess with the original [%here] *)
      Mssql_error
        { t with
          query = Option.first_some t.query query
        ; params =
            (match t.params with
            | [] -> params
            | _ -> t.params)
        ; formatted_query = Option.first_some t.formatted_query formatted_query
        ; results =
            (match t.results with
            | [] -> results
            | _ -> t.results)
        }
    in
    Caml.Printexc.raise_with_backtrace exn backtrace
  | Freetds.Dblib.Error (_, msg) as exn ->
    let backtrace = Caml.Printexc.get_raw_backtrace () in
    failwith here ?query ~params ?formatted_query ~backtrace ~exn msg
  | exn ->
    let backtrace = Caml.Printexc.get_raw_backtrace () in
    failwith
      here
      ?query
      ~params
      ?formatted_query
      ~backtrace
      ~exn
      "Unexpected error in Dblib"
;;