Source file caqti_pool_sig.ml

(* Copyright (C) 2017--2018  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 OCaml static compilation 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
 * along with this library.  If not, see <http://www.gnu.org/licenses/>.
 *)

(** Resource pool signature. *)

module type S = sig

  type 'a future

  type ('a, +'e) t

  val create :
    ?max_size: int ->
    ?check: ('a -> (bool -> unit) -> unit) ->
    ?validate: ('a -> bool future) ->
    (unit -> ('a, 'e) result future) -> ('a -> unit future) -> ('a, 'e) t
  (** (internal) [create alloc free] is a pool of resources allocated by
      [alloc] and freed by [free]. This is primarily indented for implementing
      the [connect_pool] functions.

      @param max_size
        Maximum number of resources to allocate at any given time.

      @param check
        A function used to check a resource after use.

      @param validate
        A function to check before use that a resource is still valid. *)

  val size : ('a, 'e) t -> int
  (** [size pool] is the current number of open resources in [pool]. *)

  val use :
    ?priority: float ->
    ('a -> ('b, 'e) result future) -> ('a, 'e) t -> ('b, 'e) result future
  (** [use f pool] calls [f] on a resource drawn from [pool], handing back the
      resource to the pool when [f] exits.

      @param priority
        Requests for the resource are handled in decreasing order of priority.
        The default priority is [0.0]. *)

  val drain : ('a, 'e) t -> unit future
  (** [drain pool] closes all resources in [pool]. The pool is still usable, as
      new resources will be created on demand. *)

end