unikernel/duniverse/ocaml-caqti/caqti/lib/caqti_connect_config.ml

@@ -0,0 +1,65 @@
+(* Copyright (C) 2023--2025  Petter A. Urkedal <paurkedal@gmail.com>
+ *
+ * This library is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or (at your
+ * option) any later version, with the LGPL-3.0 Linking Exception.
+ *
+ * This library is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
+ * License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * and the LGPL-3.0 Linking Exception along with this library.  If not, see
+ * <http://www.gnu.org/licenses/> and <https://spdx.org>, respectively.
+ *)
+
+(* Map Implementation *)
+
+type 'a tag = ..
+
+module type KEY = sig
+  type value
+  type 'a tag += Tag : value tag
+  val name : string [@@warning "-32"]
+  val default : value
+end
+
+type 'a key = (module KEY with type value = 'a)
+
+let create_key (type a) name (default : a) : a key =
+  let module Key = struct
+    type value = a
+    type _ tag += Tag : value tag
+    let name = name
+    let default = default
+  end in
+  (module Key : KEY with type value = a)
+
+module String_map = Map.Make (String)
+
+type binding = Binding : 'a tag * 'a -> binding
+
+type t = binding String_map.t
+
+let default = String_map.empty
+
+let mem_name key_name = String_map.mem key_name
+
+let get : type a. a key -> t -> a = fun (module Key) m ->
+  (match String_map.find_opt Key.name m with
+   | Some (Binding (Key.Tag, v)) -> v
+   | _ -> Key.default)
+
+let set : type a. a key -> a -> t -> t = fun (module Key) v m ->
+  String_map.add Key.name (Binding (Key.Tag, v)) m
+
+let reset : type a. a key -> t -> t = fun (module Key) m ->
+  String_map.remove Key.name m
+
+(* Configuration Keys *)
+
+let tweaks_version : (int * int) key = create_key "tweaks_version" (1, 7)
+
+let dynamic_prepare_capacity = create_key "dynamic_prepare_capacity" 32