ocaml-multicore / reagents

Reagents for multicore OCaml
ISC License
125 stars 6 forks source link

= 📢 Note 📢 =

kcas is the recommended framework for composable concurrency. It provides a favourable performance and safer interface than Reagents.

Reagents are a research project and not ready for production use. This repository is no longer actively developed or maintained.

See comparison for details.

Reagents — Composable lock-free data and synchronization structures

API reference

Reagents are an experimental library for writing multicore programs. Reagents provide an expressive framework for composable multithreading. They support both fine- and coarse-grained multithreading and incorporate mechanisms for efficient retrying.

Contents

Motivation

Reagents strive to be a comprehensive framework for all things concurrent and parallel. In particular, they have the following advantages over traditional approaches:

Limitations

Reagents are weaker than transactional memory. A reagent must be decomposable into a list of compare-and-set operations. This eliminates the need for any extra accounting for performed CASes.

Getting Reagents

Reagents require OCaml 5 (opam switch create 5.0.0).

Install Reagents from this repository.

opam pin -y https://github.com/ocaml-multicore/reagents.git

Soon Reagents will be available in the opam repository.

Test the setup in utop with the following snippet.

# #require "reagents";;

# module Scheduler = (val Reagents.Toy_scheduler.make 1 ())
  open Reagents.Make (Scheduler);;

# let s = Ref.mk_ref "hello world\n" in
  Scheduler.run (run (Ref.read s >>> lift print_string));;

hello world
- : unit = ()

Key concepts

This section briefly explains all the key concepts required for using Reagents.

Scheduler

Reagents are parametrised over a minimal scheduler interface. If an active reagent cannot make progress, Reagents automatically suspend the fiber. Once someone else updates the state, Reagents trigger required resumptions. This behavior comes for free.

  type 'a cont

  val suspend : ('a cont -> 'a option) -> 'a
  val resume : 'a cont -> 'a -> unit
  val get_tid : unit -> int

A toy scheduler for experimenting and running tests is available in Reagents.Toy_scheduler.

Reagent Type

A computation within Reagents framework has the following type ('a, 'b) Reagents.t. Here, the computation takes a parameter of type 'a and returns a value of type 'b. Internally, it may consist of any number of operations and any number of side effects. Crucially, regardless of the construction of a reagent, all of its operations execute atomically and entirely or none at all.

Combinators

Reagents can be composed in arbitrary ways. The following three main combinators are exposed by the library. These can be composed into more complex combinators.

Sequential composition runs one reagent after another. It passes the result of the first one as a parameter to the second one.

Conjunction executes both reagents at once and returns both results. Note, this combinator still attempts to execute its components sequentially. It differs with [>>>] in information flow only.

Disjunction tries to execute the first reagent and if it blocks, it attempts the second one. If both block, the first one to unblock is executed. Also referred to as a left-biased choice.

Running a Reagent

Once the desired reagent has been composed, it can be run.

val run : ('a, 'b) t -> 'a -> 'b

run r v executes the reagent r with value v.

Note, this function has to be executed from within a scheduler for the suspension and resumption effects to be handled correctly.

Others

There are several other values defined in the public interface that serve as units, helpers, or transformations for existing reagents. Perhaps the most notable one is attempt, which converts a blocking reagent into a non-blocking one.

val attempt : ('a, 'b) t -> ('a, 'b option) t

attempt r converts a blocking reagent into a non-blocking one. If the reagent blocks, then attempt returns None. Otherwise, the reagent is committed immediately and the returned option value is non-empty.

Data structures

Reagents expose two core data structures. Complex data structures should utilise these as building blocks, if possible.

The library also provides many higher-level data structures (e.g. counter, stack, queue) and synchronisation primitives (e.g. lock, conditional variable). See interface.

Sample programs

This section showcases a few applications of Reagents.

Counter

See a simple example of creating a synchronized counter below.

# let counter = Counter.create 0 in
  let a = run (Counter.inc counter) () in
  let b = run (Counter.inc counter) () in
  let c = run (Counter.dec counter) () in
  (a, b, c);;
- : int * int * int = (0, 1, 2)

Both inc and dec operations are of type (unit, int) Reagents.t since they take a unit as input and return the previous value. Now, imagine there are several counters representing different statistics about the system, balances of bank accounts, etc.

Reagents let us take a consistent snapshot of the system without locks.

# let c1 = Counter.create 0 in
  let c2 = Counter.create 0 in

  run (Counter.get c1 <*> Counter.get c2) ();;
- : int * int = (0, 0)

Reference

Continuing from counter, we can update any number of locations at once as well. This example uses references, which are similar to an atomic variable.

# let account_1 = Ref.mk_ref 100 in
  let account_2 = Ref.mk_ref 0 in

  let transfer a b amount =
    Ref.upd a (fun acc () -> Some (acc - amount, ()))
    >>> Ref.upd b (fun acc () -> Some (acc + amount, ()))
  in

  run (transfer account_1 account_2 100) ();

  ((Ref.read_imm account_1), (Ref.read_imm account_2));;
- : int * int = (0, 100)

Note, in the example above the function passed to Ref.upd returns an option type. If the observed value of account is not appropriate for the requested operation (e.g. the transfer would make account 1 negative), it may choose to return None. In such a case, reagent will block until the value of the reference is updated by another actor. Alternatively, it can be attempted, to simply return with failure if the reagent cannot proceed. See counter_test.ml for example.

Channel

Channel is the building block for sharing memory by communication. Reagents offer a two-way channel (but we can pass units in one direction).

# Scheduler.run (fun () ->
    let endpoint_a, endpoint_b = Channel.mk_chan () in
    Scheduler.fork (fun () -> run (Channel.swap endpoint_a) 12345);
    print_int (run (Channel.swap endpoint_b) ()));;
12345- : unit = ()

There are a couple of nuances worth keeping in mind:

Catalyst

Catalyst is a passively invoked reagent. It does not react on its own, instead, it remains ready to react with others as many times as needed until canceled. Catalysts let us link multiple data structures to form a graph of computations. See catalyst_test.ml for examples of linking channels.

More

More sample programs and tests are located in the tests directory of the distribution. They can be built and run with:

dune build @runtest

Individual tests are built as executables (available in Dune's _build directory).

Development

Formatting

This project uses ocamlformat (for OCaml) and prettier (for Markdown).

Internals quick start

Reagents are largely driven by kcas. kcas is a software solution for executing multiple atomic operations as a single transaction on architectures providing a single-word CAS only. The current implementation of kcas requires k+1 atomic operations for k-location update.

In the non-blocking case, Reagents constitute a convenient abstraction over the specification and aggregation of individual atomic operations. If the list of required atomic operations can be constructed and committed immediately, a reagent succeeds using the fast-path.

However, an operation may be unable to proceed. If fast-path found that the operation cannot finish (e.g. pop on an empty stack), Reagents core generates an offer. The offer is then published in a relevant queue with extra information and fiber suspends on it. Once another thread comes, it sees the offer and resumes fibers that are now unblocked. This logic is reagent-specific. In the case of reference, it's going to wake up all waiters. In the case of channel, it will take suspended thread's transaction, merge it with its own, and try to commit everything at once. If the commit succeeds, it provides the suspended thread with the result and resumes it. Both actions cancel the offer.

These two mechanisms are the key design choices behind Reagents.

License

Reagents are distributed under ISC license.

Further Reading

Talks:

Papers in the order of increasing detail: