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