617 Advanced

Optics: prism pattern

Functional Programming

Tutorial Video

Text description (accessibility)

This video demonstrates the "Optics: prism pattern" functional Rust example. Difficulty level: Advanced. Key concepts covered: Functional Programming. Optics are composable data accessors originating from Haskell's lens library (Edward Kmett, 2012). Key difference from OCaml: 1. **HKT requirement**: Haskell's van Laarhoven encoding uses Functor/Applicative for optic unification requiring HKT; Rust uses explicit struct types per optic kind.

Tutorial

The Problem

Optics are composable data accessors originating from Haskell's lens library (Edward Kmett, 2012). They solve the deeply-nested update problem in immutable data: updating a field three levels deep requires rebuilding all intermediate values. Optics compose — a lens into a struct field composed with a prism for an enum variant gives a combined accessor that can get, set, and modify deeply nested optional values. The optic hierarchy includes Lens (exactly one focus), Prism (zero or one focus on enum variants), Traversal (zero or more foci), and Iso (lossless bidirectional conversion).

🎯 Learning Outcomes

• The specific optic demonstrated in this example and what it focuses on

• How to implement the optic manually using closures or structs in Rust

• How this optic composes with others in the hierarchy

• The laws the optic must satisfy for correct behavior

• Where optics are used: state management, config manipulation, nested data transformation

Code Example

#![allow(clippy::all)]
//! # Prism Pattern
//! Focus on one variant of an enum.

#[derive(Clone, Debug, PartialEq)]
pub enum Json {
    Null,
    Num(f64),
    Str(String),
    Array(Vec<Json>),
}

pub fn preview_num(j: &Json) -> Option<f64> {
    match j {
        Json::Num(n) => Some(*n),
        _ => None,
    }
}

pub fn review_num(n: f64) -> Json {
    Json::Num(n)
}

pub fn over_num(j: &Json, f: impl Fn(f64) -> f64) -> Json {
    match j {
        Json::Num(n) => Json::Num(f(*n)),
        other => other.clone(),
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_prism() {
        let j = Json::Num(5.0);
        assert_eq!(preview_num(&j), Some(5.0));
        let j2 = over_num(&j, |n| n * 2.0);
        assert_eq!(j2, Json::Num(10.0));
    }
}

(* Prism in OCaml *)
type ('s,'a) prism = {
  preview: 's -> 'a option;
  review:  'a -> 's;
}

(* Prisms for a JSON-like type *)
type json = Null | JBool of bool | JNum of float | JStr of string | JArr of json list

let json_bool = {
  preview = (function JBool b -> Some b | _ -> None);
  review  = (fun b -> JBool b);
}

let json_num = {
  preview = (function JNum n -> Some n | _ -> None);
  review  = (fun n -> JNum n);
}

let json_str = {
  preview = (function JStr s -> Some s | _ -> None);
  review  = (fun s -> JStr s);
}

(* Prism law checks *)
let preview_review p a = p.preview (p.review a) = Some a
let review_preview_consistency p s = match p.preview s with
  | None   -> true
  | Some a -> p.review a = s

let () =
  Printf.printf "law review_preview JBool: %b\n" (preview_review json_bool true);
  Printf.printf "law review_preview JNum:  %b\n" (preview_review json_num 42.0);
  Printf.printf "preview JBool on JNum: %s\n"
    (match json_bool.preview (JNum 1.0) with None->"None" | Some b->string_of_bool b)

Key Differences

HKT requirement: Haskell's van Laarhoven encoding uses Functor/Applicative for optic unification requiring HKT; Rust uses explicit struct types per optic kind.

Operator syntax: Haskell uses ^., .~, %~ for terse optic use; Rust uses method calls, more verbose but explicit.

Derive macros: lens-rs and similar crates provide derive macros for automatic lens generation; OCaml uses ppx_lens for the same.

Performance: Boxed closure implementations have runtime overhead; monomorphized generic versions compile to zero-cost abstractions.

OCaml Approach

OCaml optics use the same record-with-function approach:

type ('s, 'a) lens = { get: 's -> 'a; set: 's -> 'a -> 's }
let name_lens = { get = (fun u -> u.name); set = (fun u n -> { u with name = n }) }
let compose l1 l2 = { get = (fun s -> l2.get (l1.get s)); set = (fun s a -> l1.set s (l2.set (l1.get s) a)) }

Full Source

#![allow(clippy::all)]
//! # Prism Pattern
//! Focus on one variant of an enum.

#[derive(Clone, Debug, PartialEq)]
pub enum Json {
    Null,
    Num(f64),
    Str(String),
    Array(Vec<Json>),
}

pub fn preview_num(j: &Json) -> Option<f64> {
    match j {
        Json::Num(n) => Some(*n),
        _ => None,
    }
}

pub fn review_num(n: f64) -> Json {
    Json::Num(n)
}

pub fn over_num(j: &Json, f: impl Fn(f64) -> f64) -> Json {
    match j {
        Json::Num(n) => Json::Num(f(*n)),
        other => other.clone(),
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_prism() {
        let j = Json::Num(5.0);
        assert_eq!(preview_num(&j), Some(5.0));
        let j2 = over_num(&j, |n| n * 2.0);
        assert_eq!(j2, Json::Num(10.0));
    }
}

(* Prism in OCaml *)
type ('s,'a) prism = {
  preview: 's -> 'a option;
  review:  'a -> 's;
}

(* Prisms for a JSON-like type *)
type json = Null | JBool of bool | JNum of float | JStr of string | JArr of json list

let json_bool = {
  preview = (function JBool b -> Some b | _ -> None);
  review  = (fun b -> JBool b);
}

let json_num = {
  preview = (function JNum n -> Some n | _ -> None);
  review  = (fun n -> JNum n);
}

let json_str = {
  preview = (function JStr s -> Some s | _ -> None);
  review  = (fun s -> JStr s);
}

(* Prism law checks *)
let preview_review p a = p.preview (p.review a) = Some a
let review_preview_consistency p s = match p.preview s with
  | None   -> true
  | Some a -> p.review a = s

let () =
  Printf.printf "law review_preview JBool: %b\n" (preview_review json_bool true);
  Printf.printf "law review_preview JNum:  %b\n" (preview_review json_num 42.0);
  Printf.printf "preview JBool on JNum: %s\n"
    (match json_bool.preview (JNum 1.0) with None->"None" | Some b->string_of_bool b)

✓ Tests Rust test suite

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_prism() {
        let j = Json::Num(5.0);
        assert_eq!(preview_num(&j), Some(5.0));
        let j2 = over_num(&j, |n| n * 2.0);
        assert_eq!(j2, Json::Num(10.0));
    }
}

Deep Comparison

Prism

Focus on one enum variant

Exercises

Lens laws: Write tests for all three lens laws: get-set (get after set returns set value), set-get (set to current value is identity), set-set (second set wins).

Prism laws: Write tests for prism laws: preview after review returns Some, set via review then preview round-trips.

Compose two levels: Create a lens for a struct field and a prism for an enum variant in that field — compose them and modify the inner value when the variant is present.

Open Source Repos

functional-rust

View the source for this example on GitHub — OCaml and Rust side by side in the repo.

Rust