fp/apps/aquatic/crates/http_load_test/src/config.rs

use std::net::SocketAddr;

use aquatic_common::cli::LogLevel;
use aquatic_toml_config::TomlConfig;
use serde::Deserialize;

/// aquatic_http_load_test configuration
#[derive(Clone, Debug, PartialEq, TomlConfig, Deserialize)]
#[serde(default, deny_unknown_fields)]
pub struct Config {
    pub server_address: SocketAddr,
    pub log_level: LogLevel,
    pub num_workers: usize,
    /// Maximum number of connections to keep open
    pub num_connections: usize,
    /// How often to check if num_connections connections are open, and
    /// open a new one otherwise. A value of 0 means that connections are
    /// opened as quickly as possible, which is useful when the tracker
    /// does not keep connections alive.
    pub connection_creation_interval_ms: u64,
    /// Announce/scrape url suffix. Use `/my_token/` to get `/announce/my_token/`
    pub url_suffix: String,
    pub duration: usize,
    pub keep_alive: bool,
    pub enable_tls: bool,
    pub torrents: TorrentConfig,
}

impl aquatic_common::cli::Config for Config {
    fn get_log_level(&self) -> Option<LogLevel> {
        Some(self.log_level)
    }
}

impl Default for Config {
    fn default() -> Self {
        Self {
            server_address: "127.0.0.1:3000".parse().unwrap(),
            log_level: LogLevel::Error,
            num_workers: 1,
            num_connections: 128,
            connection_creation_interval_ms: 10,
            url_suffix: "".into(),
            duration: 0,
            keep_alive: true,
            enable_tls: true,
            torrents: TorrentConfig::default(),
        }
    }
}

#[derive(Clone, Debug, PartialEq, TomlConfig, Deserialize)]
#[serde(default, deny_unknown_fields)]
pub struct TorrentConfig {
    pub number_of_torrents: usize,
    /// Probability that a generated peer is a seeder
    pub peer_seeder_probability: f64,
    /// Probability that a generated request is a announce request, as part
    /// of sum of the various weight arguments.
    pub weight_announce: usize,
    /// Probability that a generated request is a scrape request, as part
    /// of sum of the various weight arguments.
    pub weight_scrape: usize,
    /// Peers choose torrents according to this Gamma distribution shape
    pub torrent_gamma_shape: f64,
    /// Peers choose torrents according to this Gamma distribution scale
    pub torrent_gamma_scale: f64,
}

impl Default for TorrentConfig {
    fn default() -> Self {
        Self {
            number_of_torrents: 10_000,
            peer_seeder_probability: 0.25,
            weight_announce: 5,
            weight_scrape: 0,
            torrent_gamma_shape: 0.2,
            torrent_gamma_scale: 100.0,
        }
    }
}

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

    ::aquatic_toml_config::gen_serialize_deserialize_test!(Config);
}