bi_consumer_demo/
bi_consumer_demo.rs

1/*******************************************************************************
2 *
3 *    Copyright (c) 2025 - 2026 Haixing Hu.
4 *
5 *    SPDX-License-Identifier: Apache-2.0
6 *
7 *    Licensed under the Apache License, Version 2.0.
8 *
9 ******************************************************************************/
10
11//! BiConsumer demonstration
12//!
13//! This example demonstrates the usage of BiConsumer types after
14//! refactoring to use &T, &U semantics (not modifying input values).
15
16use qubit_function::{
17    ArcBiConsumer,
18    BiConsumer,
19    BoxBiConsumer,
20    BoxStatefulBiConsumer,
21    RcBiConsumer,
22    StatefulBiConsumer,
23};
24use std::cell::RefCell;
25use std::rc::Rc;
26use std::sync::{
27    Arc,
28    Mutex,
29};
30use std::thread;
31
32fn main() {
33    println!("=== BiConsumer Demo ===\n");
34
35    // 1. BoxBiConsumer - Single ownership
36    println!("1. BoxBiConsumer - Single ownership:");
37    let log = Arc::new(Mutex::new(Vec::new()));
38    let l = log.clone();
39    let box_consumer = BoxBiConsumer::new(move |x: &i32, y: &i32| {
40        println!("  Processing: x={}, y={}", x, y);
41        l.lock().unwrap().push(*x + *y);
42    });
43    box_consumer.accept(&10, &5);
44    println!("  Result log: {:?}\n", *log.lock().unwrap());
45
46    // 2. Method chaining with BoxBiConsumer
47    println!("2. BoxBiConsumer with method chaining:");
48    let log = Arc::new(Mutex::new(Vec::new()));
49    let l1 = log.clone();
50    let l2 = log.clone();
51    let chained = BoxBiConsumer::new(move |x: &i32, y: &i32| {
52        l1.lock().unwrap().push(*x + *y);
53        println!("  After first operation: sum = {}", x + y);
54    })
55    .and_then(move |x: &i32, y: &i32| {
56        l2.lock().unwrap().push(*x * *y);
57        println!("  After second operation: product = {}", x * y);
58    });
59    chained.accept(&5, &3);
60    println!("  Final log: {:?}\n", *log.lock().unwrap());
61
62    // 3. ArcBiConsumer - Thread-safe shared ownership
63    println!("3. ArcBiConsumer - Thread-safe shared ownership:");
64    let log = Arc::new(Mutex::new(Vec::new()));
65    let l = log.clone();
66    let arc_consumer = ArcBiConsumer::new(move |x: &i32, y: &i32| {
67        l.lock().unwrap().push(*x + *y);
68        println!("  Thread {:?}: sum = {}", thread::current().id(), x + y);
69    });
70
71    let consumer1 = arc_consumer.clone();
72    let consumer2 = arc_consumer.clone();
73
74    let handle1 = thread::spawn(move || {
75        let c = consumer1;
76        c.accept(&10, &5);
77    });
78
79    let handle2 = thread::spawn(move || {
80        let c = consumer2;
81        c.accept(&20, &8);
82    });
83
84    handle1.join().unwrap();
85    handle2.join().unwrap();
86    println!("  Final log: {:?}\n", *log.lock().unwrap());
87
88    // 4. RcBiConsumer - Single-threaded shared ownership
89    println!("4. RcBiConsumer - Single-threaded shared ownership:");
90    let log = Rc::new(RefCell::new(Vec::new()));
91    let l = log.clone();
92    let rc_consumer = RcBiConsumer::new(move |x: &i32, y: &i32| {
93        l.borrow_mut().push(*x + *y);
94    });
95
96    let clone1 = rc_consumer.clone();
97    let clone2 = rc_consumer.clone();
98
99    clone1.accept(&5, &3);
100    println!("  After first use: {:?}", *log.borrow());
101
102    clone2.accept(&7, &2);
103    println!("  After second use: {:?}\n", *log.borrow());
104
105    // 5. Working with closures directly
106    println!("5. Working with closures directly:");
107    let log = Arc::new(Mutex::new(Vec::new()));
108    let l = log.clone();
109    let closure = move |x: &i32, y: &i32| {
110        let sum = *x + *y;
111        l.lock().unwrap().push(sum);
112    };
113    closure.accept(&10, &20);
114    println!("  After closure: {:?}\n", *log.lock().unwrap());
115
116    // 6. Conditional BiConsumer
117    println!("6. Conditional BiConsumer:");
118    let log = Arc::new(Mutex::new(Vec::new()));
119    let l = log.clone();
120    let mut conditional = BoxStatefulBiConsumer::new(move |x: &i32, y: &i32| {
121        l.lock().unwrap().push(*x + *y);
122    })
123    .when(|x: &i32, y: &i32| *x > 0 && *y > 0);
124
125    conditional.accept(&5, &3);
126    println!("  Positive values: {:?}", *log.lock().unwrap());
127
128    conditional.accept(&-5, &3);
129    println!("  Negative value (unchanged): {:?}\n", *log.lock().unwrap());
130
131    // 7. Conditional branch BiConsumer
132    println!("7. Conditional branch BiConsumer:");
133    let log = Arc::new(Mutex::new(Vec::new()));
134    let l1 = log.clone();
135    let l2 = log.clone();
136    let mut branch = BoxStatefulBiConsumer::new(move |x: &i32, _y: &i32| {
137        l1.lock().unwrap().push(*x);
138    })
139    .when(|x: &i32, y: &i32| *x > *y)
140    .or_else(move |_x: &i32, y: &i32| {
141        l2.lock().unwrap().push(*y);
142    });
143
144    branch.accept(&15, &10);
145    println!("  When x > y: {:?}", *log.lock().unwrap());
146
147    branch.accept(&5, &10);
148    println!("  When x <= y: {:?}\n", *log.lock().unwrap());
149
150    // 8. Accumulating statistics
151    println!("8. Accumulating statistics:");
152    let count = Arc::new(Mutex::new(0));
153    let sum = Arc::new(Mutex::new(0));
154    let c = count.clone();
155    let s = sum.clone();
156    let stats_consumer = BoxBiConsumer::new(move |x: &i32, y: &i32| {
157        *c.lock().unwrap() += 1;
158        *s.lock().unwrap() += x + y;
159    });
160
161    stats_consumer.accept(&5, &3);
162    stats_consumer.accept(&10, &2);
163    stats_consumer.accept(&7, &8);
164
165    println!("  Count: {}", *count.lock().unwrap());
166    println!("  Sum: {}\n", *sum.lock().unwrap());
167
168    // 9. Name support
169    println!("9. Name support:");
170    let mut named_consumer = BoxBiConsumer::<i32, i32>::noop();
171    println!("  Initial name: {:?}", named_consumer.name());
172
173    named_consumer.set_name("sum_calculator");
174    println!("  After setting name: {:?}", named_consumer.name());
175    println!("  Display: {}\n", named_consumer);
176
177    println!("=== Demo Complete ===");
178}
bi_consumer_demo/bi_consumer_demo.rs

bi_consumer_demo/
bi_consumer_demo.rs
