fiddler 4.9.1

Data Stream processor written in rust
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use crate::config::register_plugin;
use crate::config::{parse_configuration_item, Item, ItemType};
use crate::config::{ConfigSpec, ExecutionType};
use crate::Message;
use crate::{Closer, Error, Output};
use async_trait::async_trait;
use serde_yaml::Value;
mod check;
use fiddler_macros::fiddler_registration_func;

pub struct Switch {
    steps: Vec<Box<dyn Output + Send + Sync>>,
}

#[async_trait]
impl Output for Switch {
    async fn write(&mut self, message: Message) -> Result<(), Error> {
        'steps: for p in &mut self.steps {
            match p.write(message.clone()).await {
                Ok(_) => return Ok(()),
                Err(e) => match e {
                    Error::ConditionalCheckfailed => continue 'steps,
                    _ => return Err(e),
                },
            };
        }
        Ok(())
    }
}

#[async_trait]
impl Closer for Switch {
    async fn close(&mut self) -> Result<(), Error> {
        for p in &mut self.steps {
            p.close().await?;
        }
        Ok(())
    }
}

#[fiddler_registration_func]
fn create_switch(conf: Value) -> Result<ExecutionType, Error> {
    let c: Vec<Item> = serde_yaml::from_value(conf.clone())?;
    let mut steps = Vec::new();
    for p in c {
        let ri = parse_configuration_item(ItemType::Output, &p.extra).await?;
        let step = ((ri.creator)(ri.config.clone())).await?;
        if let ExecutionType::Output(rp) = step {
            steps.push(rp);
        };
    }

    let s = Switch { steps };

    Ok(ExecutionType::Output(Box::new(s)))
}

pub(super) fn register_switch() -> Result<(), Error> {
    let config = "type: array";

    let conf_spec = ConfigSpec::from_schema(config)?;

    register_plugin("switch".into(), ItemType::Output, conf_spec, create_switch)?;
    check::register_check()
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn register_plugin() {
        register_switch().unwrap()
    }
}