jellyflow-runtime 0.3.0

Headless store, rules, schema, profile, and change pipeline for Jellyflow.
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

use crate::runtime::xyflow::changes::NodeChange;
use crate::runtime::xyflow::dialect::{node_update_id, node_update_op};
use jellyflow_core::core::NodeId;
use jellyflow_core::ops::{GraphMutationPlanner, GraphOp, GraphTransaction};

use super::ApplyChangesPlanner;

impl<'a> ApplyChangesPlanner<'a> {
    pub(super) fn apply_nodes(&mut self, changes: &[NodeChange]) {
        for change in changes {
            self.apply_node_change(change);
        }
    }

    fn apply_node_change(&mut self, change: &NodeChange) {
        let Some(op) = self.node_change_op(change) else {
            self.mark_ignored();
            return;
        };
        if GraphTransaction::from_ops([op])
            .apply_to(self.graph)
            .is_ok()
        {
            self.mark_applied();
        } else {
            self.mark_ignored();
        }
    }

    fn node_change_op(&self, change: &NodeChange) -> Option<GraphOp> {
        Some(match change {
            NodeChange::Add { id, node } => {
                if self.graph.nodes().contains_key(id) {
                    return None;
                }
                GraphOp::AddNode {
                    id: *id,
                    node: node.clone(),
                }
            }
            NodeChange::Remove { id } => self.remove_node_change_op(*id)?,
            _ => self.node_update_change_op(change)?,
        })
    }

    fn node_update_change_op(&self, change: &NodeChange) -> Option<GraphOp> {
        let id = node_update_id(change)?;
        let node = self.graph.nodes().get(&id)?;
        node_update_op(change, node)
    }

    fn remove_node_change_op(&self, id: NodeId) -> Option<GraphOp> {
        GraphMutationPlanner::new(self.graph)
            .remove_node_op(id)
            .ok()
    }
}