tempest 0.1.1

Realtime message handling framework inspired by Apache Storm and built with Actix
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

use crate::common::logger::*;
use config;
use serde_derive::Deserialize;

/// Config is used to parse `Topology.toml` files

static TARGET_SERVICE_CONFIG: &'static str = "tempest::service::config";

/// Read a config from a file path
fn read_config(path: &str) -> Result<config::Config, config::ConfigError> {
    let mut file = config::Config::default();
    match file.merge(config::File::with_name(path)) {
        Ok(file) => Ok(file.to_owned()),
        Err(err) => Err(err),
    }
}

/// Try and parse a config file path into a `TopologyConfig`
pub fn get_topology_config(path: &str) -> Result<TopologyConfig, config::ConfigError> {
    match read_config(path) {
        Ok(config) => {
            let topology: TopologyConfig = config.try_into()?;
            debug!(
                target: TARGET_SERVICE_CONFIG,
                "Topology config: {:?}", &topology
            );
            Ok(topology)
        }
        Err(err) => Err(err),
    }
}

/// The TopologyConfig is used to deserialize the main Topology
/// section from a `Topology.toml` file.
#[derive(Debug, Deserialize)]
pub struct TopologyConfig {
    pub name: String,
    pub host: Option<String>,
    pub port: Option<String>,
    pub graceful_shutdown: Option<u64>,
    pub task: Vec<TaskConfig>,
    pub source: Option<SourceConfig>,
    pub metric: Option<MetricConfig>,
}

/// The TaskConfig is used to deserialize a `Task`
/// section (`[[task]]`) from a `Topology.toml` file.
#[derive(Debug, Deserialize)]
pub struct TaskConfig {
    pub name: String,
    pub workers: Option<u64>,
    pub poll_interval: Option<u64>,
    pub poll_count: Option<u16>,
    pub max_backoff: Option<u64>,
}

/// The SourceConfig is used to partially deserialize a `[source.config]`
/// section from a `Topology.toml` file.
///
/// The config is stored as a `config::Value` to generate Source configuration
/// options.
#[derive(Debug, Deserialize)]
pub struct SourceConfig {
    pub config: config::Value,
}

/// The MetricConfig is used to deserialize a `[metric]`
/// section from a `Topology.toml` file.
#[derive(Debug, Deserialize)]
pub struct MetricConfig {
    pub flush_interval: Option<u64>,
    pub target: Vec<config::Value>,
}