Source file caqti_common_priv.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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
open Printf
let ident x = x
let (%) g f x = g (f x)
let (%>) f g x = g (f x)
let (%>?) f g x = match f x with Ok y -> g y | Error _ as r -> r
let (|>?) r f = match r with Ok x -> f x | Error _ as r -> r
let rec ncompose n f acc = if n = 0 then acc else ncompose (n - 1) f (f acc)
module Option = struct
type 'a t = 'a option
let map f = function None -> None | Some x -> Some (f x)
let fold f = function None -> ident | Some x -> f x
let for_all f = function None -> true | Some x -> f x
end
module Result = struct
let map f = function Ok x -> Ok (f x) | Error e -> Error e
end
module List = struct
include List
let rec fold f = function
| [] -> fun acc -> acc
| x :: xs -> f x %> fold f xs
let rec fold_r f = function
| [] -> fun acc -> Ok acc
| x :: xs -> f x %>? fold_r f xs
let rec iter_r f = function
| [] -> Ok ()
| x :: xs -> (match f x with Ok () -> iter_r f xs | Error _ as r -> r)
let iteri_r f xs =
let rec loop i = function
| [] -> Ok ()
| x :: xs ->
(match f i x with Ok () -> loop (i + 1) xs | Error _ as r -> r)
in
loop 0 xs
let rec equal f xs ys =
(match xs, ys with
| [], [] -> true
| x :: xs', y :: ys' -> f x y && equal f xs' ys'
| [], _ :: _ | _ :: _, [] -> false)
end
let finally cleanup thunk =
let r = try thunk () with xc -> cleanup (); raise xc in
cleanup (); r
let datetuple_of_iso8601 s =
if String.length s = 10 && s.[4] = '-' && s.[7] = '-' then
try
(int_of_string (String.sub s 0 4),
int_of_string (String.sub s 5 2),
int_of_string (String.sub s 8 2))
with Failure _ ->
failwith "Caqti_internal.datetuple_of_iso8601"
else
failwith "Caqti_internal.datetuple_of_iso8601"
let iso8601_of_datetuple (y, m, d) =
sprintf "%04d-%02d-%02d" y m d
let string_of_rfc3339_error ~input err =
let buf = Buffer.create 64 in
let ppf = Format.formatter_of_buffer buf in
Ptime.pp_rfc3339_error ppf err;
Format.fprintf ppf " in value %S." input;
Format.pp_print_flush ppf ();
Buffer.contents buf
let ptime_of_rfc3339_utc s =
let n = String.length s in
let s' =
if n < 13 then s else
if s.[n - 1] = 'Z' then s else
if s.[n - 3] = '+' || s.[n - 3] = '-' then s ^ ":00" else
if s.[n - 6] = '+' || s.[n - 6] = '-' then s else
s ^ "Z"
in
(match Ptime.of_rfc3339 s' with
| Ok (t, _, _) -> Ok t
| Error (`RFC3339 (_, err)) ->
Error (string_of_rfc3339_error ~input:s' err))
let pdate_of_iso8601 s =
(match Ptime.of_date (datetuple_of_iso8601 s) with
| exception Failure _ ->
Error (sprintf "Cannot parse date %S." s)
| None ->
Error (sprintf "Date %s is out of range." s)
| Some pdate -> Ok pdate)
let iso8601_of_pdate x = iso8601_of_datetuple (Ptime.to_date x)
let default_log_src = Logs.Src.create "caqti"
let request_log_src = Logs.Src.create "caqti.request"
module Alog = (val Logs.src_log default_log_src)