Update tests, examples, benches

src/lib.rs

@@ -24,6 +24,7 @@
 //!
 //! ```no_run
 //! use dagx::{task, DagRunner, Task};
+//! use futures::FutureExt; // for FutureExt::map
 //!
 //! // Source task with read-only state (tuple struct)
 //! struct Value(i32);
@@ -52,7 +53,7 @@
 //! let sum = dag.add_task(Add).depends_on((&x, &y));
 //!
 //! // Execute and retrieve results
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //! assert_eq!(dag.get(sum).unwrap(), 5);
 //! # };
 //! ```
@@ -290,6 +291,7 @@
 //!
 //! ```no_run
 //! use dagx::{task, DagRunner, Task};
+//! use futures::FutureExt;
 //!
 //! // Source task (tuple struct)
 //! struct Value(i32);
@@ -322,7 +324,7 @@
 //! let plus1 = dag.add_task(Add(1)).depends_on(&base);
 //! let times2 = dag.add_task(Scale(2)).depends_on(&base);
 //!
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //!
 //! assert_eq!(dag.get(plus1).unwrap(), 11);
 //! assert_eq!(dag.get(times2).unwrap(), 20);
@@ -335,6 +337,7 @@
 //!
 //! ```no_run
 //! use dagx::{task, DagRunner, Task};
+//! use futures::FutureExt;
 //!
 //! // Source task for String (tuple struct)
 //! struct Name(String);
@@ -378,7 +381,7 @@
 //! let active = dag.add_task(Active(true));
 //! let result = dag.add_task(FormatUser).depends_on((&name, &age, &active));
 //!
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //!
 //! assert_eq!(dag.get(result).unwrap(), "User: Alice, Age: 30, Active: true");
 //! # };
@@ -390,6 +393,7 @@
 //!
 //! ```no_run
 //! use dagx::{task, DagRunner, Task};
+//! use futures::FutureExt;
 //!
 //! // Source task (tuple struct)
 //! struct Value(i32);
@@ -430,7 +434,7 @@
 //! // Layer 3: Final result
 //! let total = dag.add_task(Add).depends_on((&sum_xy, &prod_yz)); // 5 + 15 = 20
 //!
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //!
 //! assert_eq!(dag.get(total).unwrap(), 20);
 //! # };
@@ -442,7 +446,8 @@
 //! the `#[task]` macro generates the necessary extraction logic:
 //!
 //! ```no_run
-//! use dagx::{task, DagRunner, Task};
+//! # use dagx::{task, DagRunner, Task};
+//! # use futures::FutureExt;
 //!
 //! // Just derive Clone - that's all you need!
 //! #[derive(Clone)]
@@ -473,7 +478,7 @@
 //! let user = dag.add_task(CreateUser);
 //! let formatted = dag.add_task(FormatUser).depends_on(&user);
 //!
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //!
 //! assert_eq!(dag.get(formatted).unwrap(), "Alice is 30 years old");
 //! # };
@@ -530,6 +535,7 @@
 //!
 //! ```no_run
 //! # use dagx::{task, DagRunner, Task};
+//! # use futures::FutureExt;
 //! # struct Value(i32);
 //! # #[task]
 //! # impl Value {
@@ -539,11 +545,11 @@
 //! # let dag = DagRunner::new();
 //! # let node = dag.add_task(Value(42));
 //! // Simple approach with .unwrap()
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //! let result = dag.get(node).unwrap();
 //!
 //! // Or handle errors explicitly
-//! match dag.run().await {
+//! match dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await {
 //!     Ok(_) => println!("DAG executed successfully"),
 //!     Err(e) => eprintln!("DAG execution failed: {}", e),
 //! }
@@ -626,7 +632,8 @@
 //! You output `T`, the framework handles the Arc wrapping:
 //!
 //! ```
-//! use dagx::{task, DagRunner, Task};
+//! # use dagx::{task, DagRunner, Task};
+//! # use futures::FutureExt;
 //!
 //! // ✅ CORRECT: Just output Vec<String>, framework wraps in Arc internally
 //! struct FetchData;
@@ -656,7 +663,7 @@
 //! let task2 = dag.add_task(ProcessData).depends_on(&data);
 //! let task3 = dag.add_task(ProcessData).depends_on(&data);
 //!
-//! dag.run().await.unwrap();
+//! dag.run(|fut| tokio::spawn(fut).map(Result::unwrap)).await.unwrap();
 //! # };
 //! ```
 //!