Add derive node to graph

This commit is contained in:
2024-09-07 14:07:32 +09:30
parent 3a0bfd6ad0
commit 3d8ce9c498
2 changed files with 8 additions and 9 deletions

View File

@@ -296,7 +296,7 @@ fn derive(
}
pub struct DeriveNodeRunner {
derive_node: DeriveNode,
pub derive_node: DeriveNode,
}
#[async_trait]

View File

@@ -2,7 +2,7 @@ use std::{
cmp::{min, Ordering},
collections::{HashMap, HashSet},
sync::{
mpsc, Arc
mpsc, Arc,
},
};
@@ -13,6 +13,7 @@ use serde::{Deserialize, Serialize};
use split::{SplitNode, SplitNodeRunner};
use tokio::sync::mpsc::Sender;
use crate::graph::derive::DeriveNodeRunner;
use {
derive::DeriveNode,
filter::{FilterNode, FilterNodeRunner},
@@ -138,7 +139,7 @@ fn get_runnable_node(node: Node) -> Box<dyn RunnableNode + Send> {
NodeConfiguration::FileNode => todo!(),
NodeConfiguration::MoveMoneyNode(_) => todo!(),
NodeConfiguration::MergeNode(_) => todo!(),
NodeConfiguration::DeriveNode(_) => todo!(),
NodeConfiguration::DeriveNode(derive_node) => Box::new(DeriveNodeRunner { derive_node }),
NodeConfiguration::CodeRuleNode(_) => todo!(),
NodeConfiguration::FilterNode(filter_node) => Box::new(FilterNodeRunner { filter_node }),
NodeConfiguration::UploadNode(upload_node) => Box::new(UploadNodeRunner { upload_node }),
@@ -182,7 +183,6 @@ impl RunnableGraph {
).await
}
// Make this not mutable, emit node status when required in a function or some other message
pub async fn run<'a, F, StatusChanged>(
&self,
num_threads: usize,
@@ -308,7 +308,6 @@ impl RunnableGraph {
#[cfg(test)]
mod tests {
use chrono::Local;
use super::{NodeConfiguration, RunnableGraph};