hatchet-sdk 0.2.4

This is an unofficial Rust SDK for Hatchet, a distributed, fault-tolerant task queue.
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use hatchet_sdk::serde::{Deserialize, Serialize};
use hatchet_sdk::{
    Context, Hatchet, Runnable, TriggerWorkflowOptionsBuilder, anyhow, serde_json, tokio,
};

#[derive(Serialize, Deserialize, Clone)]
#[serde(crate = "hatchet_sdk::serde")]
pub struct ParentInput {
    n: i32,
}

#[derive(Serialize, Deserialize, Clone)]
#[serde(crate = "hatchet_sdk::serde")]
pub struct ChildInput {
    a: String,
}

pub async fn create_child_spawning_workflow() -> (
    hatchet_sdk::Workflow<ParentInput, serde_json::Value>,
    hatchet_sdk::Workflow<ChildInput, serde_json::Value>,
) {
    let hatchet = Hatchet::from_env().await.unwrap();

    let child_task: hatchet_sdk::Task<ChildInput, serde_json::Value> = hatchet
        .task(
            "child_task",
            async move |input: ChildInput, _ctx: Context| -> anyhow::Result<serde_json::Value> {
                println!("child process {}", input.a);
                Ok(serde_json::json!({"status": format!("Got input {}", input.a)}))
            },
        )
        .build()
        .unwrap();

    let child_workflow = hatchet
        .workflow::<ChildInput, serde_json::Value>("fanout-child")
        .build()
        .unwrap()
        .add_task(&child_task);

    let child_workflow_clone = child_workflow.clone();

    let spawn_task = hatchet
        .task(
            "spawn_task",
            async move |input: ParentInput, _ctx: Context| -> anyhow::Result<serde_json::Value> {
                let mut child_tasks = vec![];
                for i in 0..input.n {
                    let workflow_clone = child_workflow_clone.clone();
                    let options = TriggerWorkflowOptionsBuilder::default()
                        .additional_metadata(Some(serde_json::json!({
                            "child_index": i.to_string(),
                        })))
                        .build()
                        .unwrap();
                    let handle = async move {
                        workflow_clone
                            .run(&ChildInput { a: i.to_string() }, Some(&options))
                            .await
                            .unwrap()
                            .get("child_task")
                            .unwrap()
                            .to_owned()
                    };
                    child_tasks.push(handle);
                }
                let results = futures::future::join_all(child_tasks).await;
                Ok(serde_json::json!({"results": results}))
            },
        )
        .build()
        .unwrap();

    let parent_workflow = hatchet
        .workflow::<ParentInput, serde_json::Value>("fanout-parent")
        .build()
        .unwrap()
        .add_task(&spawn_task);

    (parent_workflow, child_workflow)
}

#[tokio::main]
#[allow(dead_code)]
async fn main() {
    dotenvy::dotenv().ok();

    let (parent_workflow, _child_workflow) = create_child_spawning_workflow().await;

    let input = ParentInput { n: 10 };
    let result = parent_workflow.run(&input, None).await.unwrap();
    println!("Result: {}", result);
}