pipebase 0.2.0

A tokio based runtime library for data integration app
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

use super::Select;
use crate::common::{ConfigInto, FromConfig, FromPath};
use async_trait::async_trait;
use serde::Deserialize;

#[derive(Deserialize)]
pub struct RoundRobinSelectorConfig {}

#[async_trait]
impl FromPath for RoundRobinSelectorConfig {
    async fn from_path<P>(_path: P) -> anyhow::Result<Self>
    where
        P: AsRef<std::path::Path> + Send,
    {
        Ok(RoundRobinSelectorConfig {})
    }
}

#[async_trait]
impl ConfigInto<RoundRobinSelector> for RoundRobinSelectorConfig {}

/// Select candidates use round robin
pub struct RoundRobinSelector {
    pub i: usize,
}

#[async_trait]
impl FromConfig<RoundRobinSelectorConfig> for RoundRobinSelector {
    async fn from_config(_config: RoundRobinSelectorConfig) -> anyhow::Result<Self> {
        Ok(RoundRobinSelector { i: 0 })
    }
}

/// # Parameters
/// * T: input
#[async_trait]
impl<T> Select<T, RoundRobinSelectorConfig> for RoundRobinSelector
where
    T: Sync,
{
    /// `candidates`: index of downstreams
    async fn select(&mut self, _t: &T, candidates: &[&usize]) -> anyhow::Result<Vec<usize>> {
        let i = self.i % candidates.len();
        let selected = vec![*candidates[i]];
        self.i = i + 1;
        Ok(selected)
    }
}

#[cfg(test)]
mod tests {}