logos-blockchain-testing/cfgsync/adapter/src/registrations.rs

use cfgsync_core::NodeRegistration;
use serde::Serialize;

/// Immutable view of registrations currently known to cfgsync.
#[derive(Debug, Clone, Default, Serialize)]
pub struct RegistrationSnapshot {
    registrations: Vec<NodeRegistration>,
}

impl RegistrationSnapshot {
    /// Creates a stable registration snapshot sorted by node identifier.
    #[must_use]
    pub fn new(mut registrations: Vec<NodeRegistration>) -> Self {
        registrations.sort_by(|left, right| left.identifier.cmp(&right.identifier));

        Self { registrations }
    }

    /// Returns the number of registrations in the snapshot.
    #[must_use]
    pub fn len(&self) -> usize {
        self.registrations.len()
    }

    /// Returns `true` when the snapshot contains no registrations.
    #[must_use]
    pub fn is_empty(&self) -> bool {
        self.registrations.is_empty()
    }

    /// Iterates registrations in deterministic identifier order.
    #[must_use]
    pub fn iter(&self) -> impl Iterator<Item = &NodeRegistration> {
        self.registrations.iter()
    }

    /// Looks up a registration by its stable node identifier.
    #[must_use]
    pub fn get(&self, identifier: &str) -> Option<&NodeRegistration> {
        self.registrations
            .iter()
            .find(|registration| registration.identifier == identifier)
    }
}