netspeed-cli 0.10.3

Command-line interface for testing internet bandwidth using speedtest.net
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

//! Registry for phase functions allowing open/closed extension.
use crate::orchestrator::Orchestrator;
use crate::phases::{PhaseContext, PhaseFn, PhaseOutcome};
use std::collections::HashMap;
// Arc not needed – phases are stored in a Vec

/// Holds the ordered list of phases.
#[derive(Default)]
pub struct PhaseRegistry {
    phases: Vec<PhaseFn>,
    // optional map for named lookup if needed later
    #[allow(dead_code)]
    map: HashMap<String, usize>,
}

impl PhaseRegistry {
    /// Create a new empty registry.
    pub fn new() -> Self {
        Self::default()
    }

    /// Register a phase function (pushes to end of execution order).
    pub fn register(&mut self, name: impl Into<String>, phase: PhaseFn) {
        let idx = self.phases.len();
        self.map.insert(name.into(), idx);
        self.phases.push(phase);
    }

    /// Iterate over registered phases.
    pub fn iter(&self) -> impl Iterator<Item = &PhaseFn> {
        self.phases.iter()
    }
}

/// Execute all registered phases using the provided orchestrator.
pub async fn run_all_registered(
    orch: &Orchestrator,
    registry: &PhaseRegistry,
) -> Result<(), crate::error::Error> {
    let mut ctx = PhaseContext::new(orch.services_arc());
    for phase in registry.iter() {
        match phase(orch, &mut ctx).await {
            PhaseOutcome::PhaseCompleted => {}
            PhaseOutcome::PhaseEarlyExit => break,
            PhaseOutcome::PhaseError(e) => return Err(e),
        }
    }
    Ok(())
}