strest 0.1.10

Blazing-fast async HTTP load tester in Rust - lock-free design, real-time stats, distributed runs, and optional chart exports for high-load API testing.
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use std::sync::{Arc, OnceLock};

use crate::domain::run::{LoadMode, ProtocolKind};

use super::builtins;
use super::{ProtocolAdapterError, TransportAdapter};

#[derive(Clone)]
pub struct ProtocolRegistry {
    adapters: Vec<Arc<dyn TransportAdapter>>,
}

impl ProtocolRegistry {
    #[must_use]
    pub fn with_builtins() -> Self {
        let mut registry = Self {
            adapters: Vec::new(),
        };
        for builtin in builtins::builtins() {
            if let Err(err) = registry.register_adapter(builtin) {
                tracing::warn!(
                    "Skipping duplicate builtin protocol adapter: {}",
                    err.message
                );
            }
        }
        registry
    }

    /// Registers a protocol adapter into this registry.
    ///
    /// # Errors
    ///
    /// Returns an error when an adapter with the same `Protocol` key is already
    /// registered.
    pub fn register_adapter<P>(&mut self, adapter: P) -> Result<(), ProtocolAdapterError>
    where
        P: TransportAdapter + 'static,
    {
        let protocol = adapter.protocol();
        if self
            .adapters
            .iter()
            .any(|existing| existing.protocol() == protocol)
        {
            return Err(ProtocolAdapterError {
                message: format!("Protocol adapter already registered: {}", protocol.as_str()),
            });
        }
        self.adapters.push(Arc::new(adapter));
        Ok(())
    }

    pub fn adapter(&self, protocol: ProtocolKind) -> Option<&dyn TransportAdapter> {
        self.adapters
            .iter()
            .find(|adapter| adapter.protocol() == protocol)
            .map(Arc::as_ref)
    }

    #[must_use]
    pub fn supports_execution(&self, protocol: ProtocolKind) -> bool {
        self.adapter(protocol)
            .map(|adapter| adapter.executes_traffic())
            .unwrap_or(false)
    }

    #[must_use]
    pub fn supports_load_mode(&self, protocol: ProtocolKind, load_mode: LoadMode) -> bool {
        self.adapter(protocol)
            .map(|adapter| adapter.supported_load_modes().contains(&load_mode))
            .unwrap_or(false)
    }

    #[must_use]
    pub fn executable_protocols_csv(&self) -> String {
        let mut protocols: Vec<&'static str> = self
            .adapters
            .iter()
            .filter(|adapter| adapter.executes_traffic())
            .map(|adapter| adapter.protocol().as_str())
            .collect();
        protocols.sort_unstable();
        protocols.join(", ")
    }
}

pub fn protocol_registry() -> &'static ProtocolRegistry {
    static REGISTRY: OnceLock<ProtocolRegistry> = OnceLock::new();
    REGISTRY.get_or_init(ProtocolRegistry::with_builtins)
}