56 lines
2.1 KiB
OCaml
56 lines
2.1 KiB
OCaml
(* Copyright (C) 2022--2024 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.
|
|
*)
|
|
|
|
module type DRIVER_FUNCTOR =
|
|
functor (System : Caqti_platform.System_sig.S) ->
|
|
functor (_ : System_sig.S
|
|
with type 'a fiber := 'a System.Fiber.t
|
|
and type stdenv := System.stdenv) ->
|
|
Caqti_platform.Driver_loader.DRIVER
|
|
with type 'a fiber := 'a System.Fiber.t
|
|
and type ('a, 'err) stream := ('a, 'err) System.Stream.t
|
|
and type switch := System.Switch.t
|
|
and type stdenv := System.stdenv
|
|
|
|
let drivers = Hashtbl.create 5
|
|
let register scheme p = Hashtbl.add drivers scheme p
|
|
|
|
module Make
|
|
(System : Caqti_platform.System_sig.S)
|
|
(System_unix : System_sig.S
|
|
with type 'a fiber := 'a System.Fiber.t
|
|
and type stdenv := System.stdenv) =
|
|
struct
|
|
module Core_loader = Caqti_platform.Driver_loader.Make (System)
|
|
|
|
module type DRIVER = Core_loader.DRIVER
|
|
module type CONNECTION = Core_loader.CONNECTION
|
|
|
|
let provides_unix = true
|
|
|
|
let find_and_apply' scheme =
|
|
(match Hashtbl.find_opt drivers scheme with
|
|
| None -> None
|
|
| Some (module F : DRIVER_FUNCTOR) ->
|
|
let module Driver = F (System) (System_unix) in
|
|
Some (module Driver : DRIVER))
|
|
|
|
let find_and_apply scheme =
|
|
(match Core_loader.find_and_apply scheme with
|
|
| Some _ as r -> r
|
|
| None -> find_and_apply' scheme)
|
|
end
|