fp-library 0.7.0

A functional programming library for Rust featuring your favourite higher-kinded types and type classes.
Documentation

fp-library

crates.io docs.rs GitHub License

A functional programming library for Rust featuring your favourite higher-kinded types and type classes.

Features

  • Higher-Kinded Types (HKT): Implemented using lightweight higher-kinded polymorphism (type-level defunctionalization/brands).
  • Macros: Procedural macros (def_kind!, impl_kind!, Apply!) to simplify HKT boilerplate and type application.
  • Type Classes: A comprehensive collection of standard type classes including:
    • Functor, Applicative, Monad
    • Semigroup, Monoid
    • Foldable, Traversable
    • Compactable, Filterable, Witherable
    • Category, Semigroupoid
    • Pointed, Lift
    • ApplyFirst, ApplySecond, Semiapplicative, Semimonad
    • MonadRec, RefFunctor
    • Function, CloneableFn, SendCloneableFn, ParFoldable (Function wrappers and thread-safe operations)
    • Pointer, RefCountedPointer, SendRefCountedPointer (Pointer abstraction)
    • Defer, SendDefer
  • Helper Functions: Standard FP utilities:
    • compose, constant, flip, identity
  • Data Types: Implementations for standard and custom types:
    • Option, Result, Vec, String
    • Identity, Lazy, Pair
    • Trampoline, Thunk, Free
    • Endofunction, Endomorphism, SendEndofunction
    • RcBrand, ArcBrand, FnBrand

Motivation

Rust is a multi-paradigm language with strong functional programming features like iterators, closures, and algebraic data types. However, it lacks native support for Higher-Kinded Types (HKT), which limits the ability to write generic code that abstracts over type constructors (e.g., writing a function that works for any Monad, whether it's Option, Result, or Vec).

fp-library aims to bridge this gap by providing:

  1. A robust encoding of HKTs in stable Rust.
  2. A comprehensive set of standard type classes (Functor, Monad, Traversable, etc.).
  3. Zero-cost abstractions that respect Rust's performance characteristics.

Usage

Add fp-library to your Cargo.toml:

[dependencies]
fp-library = "0.7"

Crate Features

The library offers optional features that can be enabled in your Cargo.toml:

  • rayon: Enables parallel folding operations (ParFoldable) and parallel execution support for VecBrand using the rayon library.

To enable this feature:

[dependencies]
fp-library = { version = "0.7", features = ["rayon"] }

Example: Using Functor with Option

use fp_library::{brands::*, functions::*};

fn main() {
	let x = Some(5);
	// Map a function over the `Option` using the `Functor` type class
	let y = map::<OptionBrand, _, _, _>(|i| i * 2, x);
	assert_eq!(y, Some(10));
}

How it Works

Higher-Kinded Types (HKT)

Since Rust doesn't support HKTs directly (e.g., trait Functor<F<_>>), this library uses Lightweight Higher-Kinded Polymorphism (also known as the "Brand" pattern or type-level defunctionalization).

Each type constructor has a corresponding Brand type (e.g., OptionBrand for Option). These brands implement the Kind traits, which map the brand and generic arguments back to the concrete type. The library provides macros to simplify this process.

use fp_library::{impl_kind, kinds::*};

pub struct OptionBrand;

impl_kind! {
	for OptionBrand {
		type Of<'a, A: 'a>: 'a = Option<A>;
	}
}

Zero-Cost Abstractions & Uncurried Semantics

Unlike many functional programming libraries that strictly adhere to curried functions (e.g., map(f)(fa)), fp-library adopts uncurried semantics (e.g., map(f, fa)) for its core abstractions.

Why? Traditional currying in Rust often requires:

  • Creating intermediate closures for each partial application.
  • Heap-allocating these closures (boxing) or wrapping them in reference counters (Rc/Arc) to satisfy type system constraints.
  • Dynamic dispatch (dyn Fn), which inhibits compiler optimizations like inlining.

By using uncurried functions with impl Fn or generic bounds, fp-library achieves zero-cost abstractions:

  • No Heap Allocation: Operations like map and bind do not allocate intermediate closures.
  • Static Dispatch: The compiler can fully monomorphize generic functions, enabling aggressive inlining and optimization.
  • Ownership Friendly: Better integration with Rust's ownership and borrowing system.

This approach ensures that using high-level functional abstractions incurs no runtime penalty compared to hand-written imperative code.

Exceptions: While the library strives for zero-cost abstractions, some operations inherently require dynamic dispatch or heap allocation due to Rust's type system:

  • Functions as Data: When functions are stored in data structures (e.g., inside a Vec for Semiapplicative::apply, or in Lazy thunks), they must often be "type-erased" (wrapped in Rc<dyn Fn> or Arc<dyn Fn>). This is because every closure in Rust has a unique, anonymous type. To store multiple different closures in the same container, or to compose functions dynamically (like in Endofunction), they must be coerced to a common trait object.
  • Lazy Evaluation: The Lazy type relies on storing a thunk that can be cloned and evaluated later, which typically requires reference counting and dynamic dispatch.

For these specific cases, the library provides Brand types (like RcFnBrand and ArcFnBrand) to let you choose the appropriate wrapper (single-threaded vs. thread-safe) while keeping the rest of your code zero-cost. The library uses a unified Pointer hierarchy to abstract over these choices.

Lazy Evaluation & Effect System

Rust is an eagerly evaluated language. To enable functional patterns like deferred execution and safe recursion, fp-library provides a granular set of types that let you opt-in to specific behaviors without paying for unnecessary overhead.

Type Primary Use Case Stack Safe? Memoized? Lifetimes? HKT Traits
Thunk<'a, A> Glue Code & Borrowing. Lightweight deferred computation. Best for short chains and working with references. ⚠️ Partial (tail_rec_m only) ❌ No 'a Functor, Applicative, Monad
Trampoline<A> Deep Recursion & Pipelines. Heavy-duty computation. Uses a trampoline to guarantee stack safety for infinite recursion. ✅ Yes ❌ No 'static ❌ No
Lazy<'a, A> Caching. Wraps a computation to ensure it runs at most once. N/A ✅ Yes 'a RefFunctor

The "Why" of Three Types

Unlike lazy languages (e.g., Haskell) where the runtime handles everything, Rust requires us to choose our trade-offs:

  1. Thunk vs Trampoline: Thunk is faster and supports borrowing (&'a T). Its tail_rec_m is stack-safe, but deep bind chains will overflow the stack. Trampoline guarantees stack safety for all operations via a trampoline (the Free monad) but requires types to be 'static and Send. A key distinction is that Thunk implements Functor, Applicative, and Monad directly, making it suitable for generic programming, while Trampoline does not.
  2. Computation vs Caching: Thunk and Trampoline describe computations—they re-run every time you call .run(). If you have an expensive operation (like a DB call), convert it to a Lazy to cache the result.

Workflow Example

A common pattern is to use Trampoline for the heavy lifting (recursion), Lazy to freeze the result, and Thunk to borrow it later.

use fp_library::types::*;

// 1. Use Trampoline for stack-safe recursion
let heavy_computation = Trampoline::tail_rec_m(|n| {
    if n < 1_000 { Trampoline::pure(Step::Loop(n + 1)) } else { Trampoline::pure(Step::Done(n)) }
}, 0);

// 2. Convert to Lazy to cache the result (runs once)
let cached = Lazy::<_, RcLazyConfig>::from(heavy_computation);

// 3. Use Thunk to borrow the cached value without re-running
let view = Thunk::new(|| {
    let val = cached.get();
    format!("Result: {}", val)
});

Thread Safety and Parallelism

The library supports thread-safe operations through the SendCloneableFn extension trait and parallel folding via ParFoldable.

  • SendCloneableFn: Extends CloneableFn to provide Send + Sync function wrappers. Implemented by ArcFnBrand.
  • ParFoldable: Provides par_fold_map and par_fold_right for parallel execution.
  • Rayon Support: VecBrand supports parallel execution using rayon when the rayon feature is enabled.
use fp_library::{brands::*, functions::*};

let v = vec![1, 2, 3, 4, 5];
// Create a thread-safe function wrapper
let f = send_cloneable_fn_new::<ArcFnBrand, _, _>(|x: i32| x.to_string());
// Fold in parallel (if rayon feature is enabled)
let result = par_fold_map::<ArcFnBrand, VecBrand, _, _>(f, v);
assert_eq!(result, "12345".to_string());

Documentation

Contributing

We welcome contributions! Please feel free to submit a Pull Request.

Development Environment

This project uses Nix to manage the development environment.

  1. Install Nix Package Manager.
  2. Install nix-direnv (recommended) for automatic environment loading.

To set up the environment:

# If using direnv
direnv allow

# Or manually enter the shell
nix develop

This will provide a shell with the correct Rust version and dependencies.

Project Structure

  • fp-library/src/classes: Contains the definitions of type classes (traits).
  • fp-library/src/types: Contains implementations of type classes for various data types.
  • fp-library/src/kinds: Contains the machinery for higher-kinded types.
  • fp-library/src/brands: Contains type brands used for HKT encoding.
  • fp-library/src/functions: Contains general helper functions.
  • fp-macros: Procedural macros for generating HKT traits and implementations.

Release Process

For maintainers, the release process is documented in docs/release-process.md.

Benchmarking

This project uses Criterion.rs for benchmarking to ensure zero-cost abstractions and detect performance regressions.

To run all benchmarks:

cargo bench -p fp-library

To list available benchmarks:

cargo bench -p fp-library --bench benchmarks -- --list

To run a specific benchmark (e.g., Vec):

cargo bench -p fp-library --bench benchmarks -- Vec

Benchmark reports are generated in target/criterion/report/index.html.

License

This project is licensed under the Blue Oak Model License 1.0.0.

References