flowbuilder-core 0.1.0

Core flow building functionality for FlowBuilder
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

use crate::{FlowExecutor, Step};
use anyhow::Result;
use flowbuilder_context::FlowContext;
use std::sync::Arc;
use tokio::sync::Mutex;

/// Represents a flow that can be executed
pub struct Flow {
    steps: Vec<Step>,
}

impl Flow {
    pub(crate) fn new(steps: Vec<Step>) -> Self {
        Self { steps }
    }

    /// Execute the flow with a default context
    pub async fn execute(self) -> Result<FlowContext> {
        let context = FlowContext::default();
        self.execute_with_context(context).await
    }

    /// Execute the flow with a custom context
    pub async fn execute_with_context(
        self,
        context: FlowContext,
    ) -> Result<FlowContext> {
        let shared_context = Arc::new(Mutex::new(context));
        let executor = FlowExecutor::new();

        executor
            .execute_steps(self.steps, shared_context.clone())
            .await?;

        let final_context = Arc::try_unwrap(shared_context)
            .map_err(|_| anyhow::anyhow!("Failed to unwrap shared context"))?
            .into_inner();

        Ok(final_context)
    }
}