zen_engine/nodes/function/
mod.rs

1pub mod http_handler;
2pub(crate) mod v1;
3pub(crate) mod v2;
4
5use crate::nodes::definition::NodeHandler;
6use crate::nodes::function::v1::{FunctionV1NodeHandler, FunctionV1Trace};
7use crate::nodes::function::v2::{FunctionV2NodeHandler, FunctionV2Trace};
8use crate::nodes::result::NodeResult;
9use crate::nodes::NodeContext;
10use std::sync::Arc;
11use zen_types::decision::{FunctionContent, FunctionNodeContent};
12use zen_types::variable::Variable;
13
14#[derive(Debug, Clone)]
15pub struct FunctionNodeHandler;
16
17pub type FunctionNodeData = FunctionNodeContent;
18
19pub type FunctionNodeTrace = Variable;
20
21impl NodeHandler for FunctionNodeHandler {
22    type NodeData = FunctionNodeData;
23    type TraceData = FunctionNodeTrace;
24
25    async fn handle(&self, ctx: NodeContext<Self::NodeData, Self::TraceData>) -> NodeResult {
26        match &ctx.node {
27            FunctionNodeContent::Version1(source) => {
28                let v1_context = NodeContext::<Arc<str>, FunctionV1Trace> {
29                    id: ctx.id.clone(),
30                    name: ctx.name.clone(),
31                    input: ctx.input.clone(),
32                    extensions: ctx.extensions.clone(),
33                    trace: ctx.config.trace.then(|| Default::default()),
34                    iteration: ctx.iteration,
35                    config: ctx.config,
36                    node: source.clone(),
37                };
38
39                FunctionV1NodeHandler.handle(v1_context).await
40            }
41            FunctionNodeContent::Version2(content) => {
42                let v2_context = NodeContext::<FunctionContent, FunctionV2Trace> {
43                    id: ctx.id.clone(),
44                    name: ctx.name.clone(),
45                    input: ctx.input.clone(),
46                    extensions: ctx.extensions.clone(),
47                    trace: ctx.config.trace.then(|| Default::default()),
48                    iteration: ctx.iteration,
49                    config: ctx.config,
50                    node: content.clone(),
51                };
52
53                FunctionV2NodeHandler.handle(v2_context).await
54            }
55        }
56    }
57}
zen_engine/nodes/function/mod.rs

zen_engine/nodes/function/
mod.rs
