async-reify 0.1.1

Async computation inspector: extract continuation structure as step graphs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! Demonstrates tracing an async workflow, extracting the step graph,
//! and rendering it as DOT.
//!
//! Run with: `cargo run -p async-reify --example trace_workflow`

use async_reify::{labeled_await, reify_execution, to_dot, Trace};

async fn simulate_work() {
    tokio::task::yield_now().await;
}

#[tokio::main]
async fn main() {
    let trace = Trace::shared();

    // Step 1: fetch
    labeled_await!(simulate_work(), trace).await;

    // Step 2: transform
    labeled_await!(simulate_work(), trace).await;

    // Step 3: store
    labeled_await!(async { 42 }, trace).await;

    let trace = Arc::try_unwrap(trace)
        .expect("single owner")
        .into_inner()
        .expect("not poisoned");

    println!("Collected {} poll events", trace.events.len());

    let graph = reify_execution(trace.events);
    println!(
        "Step graph: {} steps, {} edges",
        graph.steps.len(),
        graph.edges.len()
    );

    for step in &graph.steps {
        println!(
            "  Step {}: {} ({:?}, {}us)",
            step.id, step.label, step.outcome, step.duration_us
        );
    }

    let dot = to_dot(&graph);
    println!("\nDOT output:\n{dot}");

    #[cfg(feature = "serde")]
    {
        let json = serde_json::to_string_pretty(&graph).unwrap();
        println!("\nJSON:\n{json}");
    }
}

use std::sync::Arc;