use std::any::Any;
use std::sync::Arc;
use streamweave::graph;
use streamweave::graph::Graph;
use streamweave::nodes::boolean_logic::NandNode;
use tokio::sync::mpsc;
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let (config_tx, config_rx) = mpsc::channel(1);
let (in1_tx, in1_rx) = mpsc::channel(10);
let (in2_tx, in2_rx) = mpsc::channel(10);
let (out_tx, mut out_rx) = mpsc::channel::<Arc<dyn Any + Send + Sync>>(10);
let (error_tx, mut error_rx) = mpsc::channel::<Arc<dyn Any + Send + Sync>>(10);
let mut graph: Graph = graph! {
nand: NandNode::new("nand".to_string()),
graph.configuration => nand.configuration,
graph.in1 => nand.in1,
graph.in2 => nand.in2,
nand.out => graph.output,
nand.error => graph.error
};
graph.connect_input_channel("configuration", config_rx)?;
graph.connect_input_channel("in1", in1_rx)?;
graph.connect_input_channel("in2", in2_rx)?;
graph.connect_output_channel("output", out_tx)?;
graph.connect_output_channel("error", error_tx)?;
println!("✓ Graph built with NandNode using graph! macro");
let _ = config_tx
.send(Arc::new(()) as Arc<dyn Any + Send + Sync>)
.await;
let test_pairs = vec![
(true, true), (true, false), (false, true), (false, false), ];
for (bool1, bool2) in test_pairs {
let _ = in1_tx
.send(Arc::new(bool1) as Arc<dyn Any + Send + Sync>)
.await;
let _ = in2_tx
.send(Arc::new(bool2) as Arc<dyn Any + Send + Sync>)
.await;
tokio::time::sleep(tokio::time::Duration::from_millis(50)).await;
}
println!("✓ Configuration sent and test data sent to input channels");
println!("Executing graph with NandNode...");
let start = std::time::Instant::now();
graph
.execute()
.await
.map_err(|e| format!("Graph execution failed: {:?}", e))?;
println!("✓ Graph execution completed in {:?}", start.elapsed());
drop(config_tx);
drop(in1_tx);
drop(in2_tx);
println!("Reading results from output channels...");
let mut success_count = 0;
let mut error_count = 0;
loop {
let out_result =
tokio::time::timeout(tokio::time::Duration::from_millis(100), out_rx.recv()).await;
let error_result =
tokio::time::timeout(tokio::time::Duration::from_millis(100), error_rx.recv()).await;
let mut has_data = false;
if let Ok(Some(item)) = out_result {
if let Ok(result_arc) = item.clone().downcast::<bool>() {
let result = *result_arc;
println!(" NAND result: {}", result);
success_count += 1;
has_data = true;
}
}
if let Ok(Some(item)) = error_result
&& let Ok(error_msg) = item.downcast::<String>()
{
let error = (**error_msg).to_string();
println!(" Error: {}", error);
error_count += 1;
has_data = true;
}
if !has_data {
break;
}
}
println!(
"✓ Received {} successful results via output channel",
success_count
);
println!("✓ Received {} errors via error channel", error_count);
println!("✓ Total completed in {:?}", start.elapsed());
if success_count == 4 && error_count == 0 {
println!("✓ NandNode correctly performed logical NAND operations");
} else {
println!(
"⚠ NandNode behavior may be unexpected (successes: {}, errors: {}, expected successes: 4, errors: 0)",
success_count, error_count
);
}
Ok(())
}