Skip to main content

bi_consumer_once_demo/
bi_consumer_once_demo.rs

1/*******************************************************************************
2 *
3 *    Copyright (c) 2025 - 2026.
4 *    Haixing Hu, Qubit Co. Ltd.
5 *
6 *    All rights reserved.
7 *
8 ******************************************************************************/
9
10//! BiConsumerOnce demonstration
11//!
12//! This example demonstrates the usage of BoxBiConsumerOnce type, which
13//! consumes itself on first call.
14
15use qubit_function::{
16    BiConsumerOnce,
17    BoxBiConsumerOnce,
18};
19use std::sync::{
20    Arc,
21    Mutex,
22};
23
24fn main() {
25    println!("=== BiConsumerOnce Demo ===\n");
26
27    // 1. Basic usage
28    println!("1. Basic usage:");
29    let log = Arc::new(Mutex::new(Vec::new()));
30    let l = log.clone();
31    let consumer = BoxBiConsumerOnce::new(move |x: &i32, y: &i32| {
32        l.lock().unwrap().push(*x + *y);
33        println!("  Sum: {}", x + y);
34    });
35    consumer.accept(&10, &5);
36    println!("  Log: {:?}\n", *log.lock().unwrap());
37
38    // 2. Method chaining
39    println!("2. Method chaining:");
40    let log = Arc::new(Mutex::new(Vec::new()));
41    let l1 = log.clone();
42    let l2 = log.clone();
43    let chained = BoxBiConsumerOnce::new(move |x: &i32, y: &i32| {
44        l1.lock().unwrap().push(*x + *y);
45        println!("  First: sum={}", x + y);
46    })
47    .and_then(move |x: &i32, y: &i32| {
48        l2.lock().unwrap().push(*x * *y);
49        println!("  Second: product={}", x * y);
50    });
51    chained.accept(&5, &3);
52    println!("  Log: {:?}\n", *log.lock().unwrap());
53
54    // 3. Conditional execution - true case
55    println!("3. Conditional execution - true case:");
56    let log = Arc::new(Mutex::new(Vec::new()));
57    let l = log.clone();
58    let conditional = BoxBiConsumerOnce::new(move |x: &i32, y: &i32| {
59        l.lock().unwrap().push(*x + *y);
60    })
61    .when(|x: &i32, y: &i32| *x > 0 && *y > 0);
62    conditional.accept(&5, &3);
63    println!("  Positive values: {:?}\n", *log.lock().unwrap());
64
65    // 4. Conditional execution - false case
66    println!("4. Conditional execution - false case:");
67    let log = Arc::new(Mutex::new(Vec::new()));
68    let l = log.clone();
69    let conditional = BoxBiConsumerOnce::new(move |x: &i32, y: &i32| {
70        l.lock().unwrap().push(*x + *y);
71    })
72    .when(|x: &i32, y: &i32| *x > 0 && *y > 0);
73    conditional.accept(&-5, &3);
74    println!("  Negative value (unchanged): {:?}\n", *log.lock().unwrap());
75
76    // 5. Conditional branching
77    println!("5. Conditional branching:");
78    let log = Arc::new(Mutex::new(Vec::new()));
79    let l1 = log.clone();
80    let l2 = log.clone();
81    let branch = BoxBiConsumerOnce::new(move |x: &i32, _y: &i32| {
82        l1.lock().unwrap().push(*x);
83    })
84    .when(|x: &i32, y: &i32| *x > *y)
85    .or_else(move |_x: &i32, y: &i32| {
86        l2.lock().unwrap().push(*y);
87    });
88    branch.accept(&15, &10);
89    println!("  When x > y: {:?}\n", *log.lock().unwrap());
90
91    // 6. Working with closures directly
92    println!("6. Working with closures directly:");
93    let log = Arc::new(Mutex::new(Vec::new()));
94    let l = log.clone();
95    let closure = move |x: &i32, y: &i32| {
96        l.lock().unwrap().push(*x + *y);
97        println!("  Processed: {}", x + y);
98    };
99    closure.accept(&10, &20);
100    println!("  Log: {:?}\n", *log.lock().unwrap());
101
102    // 7. Moving captured values
103    println!("7. Moving captured values:");
104    let data = vec![1, 2, 3, 4, 5];
105    let consumer = BoxBiConsumerOnce::new(move |x: &i32, y: &i32| {
106        println!("  x={}, y={}", x, y);
107        println!("  Captured data: {:?}", data);
108        println!("  Data sum: {}", data.iter().sum::<i32>());
109    });
110    consumer.accept(&5, &3);
111    // data is no longer available here
112    println!();
113
114    // 8. Initialization callback
115    println!("8. Initialization callback:");
116    let log = Arc::new(Mutex::new(Vec::new()));
117    let l = log.clone();
118    let init_callback = BoxBiConsumerOnce::new(move |width: &i32, height: &i32| {
119        println!("  Initializing with dimensions: {}x{}", width, height);
120        l.lock().unwrap().push(*width * *height);
121    });
122    init_callback.accept(&800, &600);
123    println!("  Areas: {:?}\n", *log.lock().unwrap());
124
125    // 9. Cleanup callback
126    println!("9. Cleanup callback:");
127    let cleanup = BoxBiConsumerOnce::new(|count: &i32, total: &i32| {
128        println!("  Cleanup: processed {} out of {} items", count, total);
129        println!(
130            "  Success rate: {:.1}%",
131            (*count as f64 / *total as f64) * 100.0
132        );
133    });
134    cleanup.accept(&85, &100);
135    println!();
136
137    // 10. Name support
138    println!("10. Name support:");
139    let mut named_consumer = BoxBiConsumerOnce::<i32, i32>::noop();
140    println!("  Initial name: {:?}", named_consumer.name());
141
142    named_consumer.set_name("init_callback");
143    println!("  After setting name: {:?}", named_consumer.name());
144    println!("  Display: {}", named_consumer);
145    named_consumer.accept(&1, &2);
146    println!();
147
148    // 11. Print helpers
149    println!("11. Print helpers:");
150    let print = BoxBiConsumerOnce::new(|x: &i32, y: &i32| println!("{}, {}", x, y));
151    print.accept(&42, &10);
152
153    let print_with =
154        BoxBiConsumerOnce::new(|x: &i32, y: &i32| println!("Dimensions: {}, {}", x, y));
155    print_with.accept(&800, &600);
156    println!();
157
158    // 12. Converting to function
159    println!("12. Converting to function:");
160    let log = Arc::new(Mutex::new(Vec::new()));
161    let l = log.clone();
162    let consumer = BoxBiConsumerOnce::new(move |x: &i32, y: &i32| {
163        l.lock().unwrap().push(*x + *y);
164    });
165    let func = consumer.into_fn();
166    func(&7, &3);
167    println!("  Log: {:?}\n", *log.lock().unwrap());
168
169    println!("=== Demo Complete ===");
170}