mte/src/devices.ml

let db_connection caqti_switch : (unit, Caqti_miou.connection) Vif.Device.device
    =
  let f () =
    let db_uri = Config.Exchangedb_postgres.config in
    match Caqti_miou_unix.connect ~sw:caqti_switch db_uri with
    | Error err ->
        Fmt.failwith "Database connection failure: %a." Caqti_error.pp err
    | Ok conn -> (
        match Pg.preflight conn with
        | Error err ->
            Fmt.failwith "Database preflight failure: %a." Caqti_error.pp err
        | Ok () ->
            Logs.info (fun m -> m "database connection initialized");
            conn)
  in
  let finally (module Conn : Caqti_miou.CONNECTION) = Conn.disconnect () in
  Vif.Device.v ~name:"db_connection" ~finally [] f

let keys storage =
  let f (module Conn : Pg.CONN) () =
    let (module Fs) =
      (module struct
        let t = storage
      end)
    in
    let keys : (module Keys.S) = (module Keys.Make (Conn) (Fs)) in
    keys
  in
  let finally _key = () in
  Vif.Device.v ~name:"keys" ~finally [ Vif.Device.value db_connection ] f