swactor/crates/swactor-gossip/src/trace.rs

use std::collections::HashMap;
use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::{Arc, Mutex};

use swactor::actor::ActorAddress;

use crate::state::VersionedValue;

// ── Shared handles ───────────────────────────────────────────────────────

/// Shared, append-only event log.
pub type EventLog = Arc<Mutex<Vec<GossipEvent>>>;

/// Shared tick counter — the simulation harness increments this.
pub type TickCounter = Arc<AtomicU64>;

/// Maps actor addresses to human-readable names like `"node-0"`.
pub type NameRegistry = Arc<Mutex<HashMap<ActorAddress, String>>>;

// ── Event types ──────────────────────────────────────────────────────────

#[derive(Debug, Clone)]
pub struct GossipEvent {
    pub tick: u64,
    pub node_name: String,
    pub node_addr: ActorAddress,
    pub kind: GossipEventKind,
}

#[derive(Debug, Clone)]
pub enum GossipEventKind {
    /// A local `Set { key, .. }` was processed.
    LocalSet { key: String },
    /// `DoGossipRound` chose a peer and sent a Push.
    GossipRoundStarted { target_name: String },
    /// `DoGossipRound` had no peers.
    GossipRoundNoPeers,
    /// Received a Push from another node.
    PushReceived {
        from_name: String,
        keys_updated: usize,
    },
    /// Received a Query.
    QueryReceived { key: String },
    /// A peer was added.
    PeerAdded { peer_name: String },
    /// A peer was removed.
    PeerRemoved { peer_name: String },
    /// Full state snapshot (requested via `TakeSnapshot`).
    StateSnapshot { snapshot: NodeSnapshot },
}

#[derive(Debug, Clone)]
pub struct NodeSnapshot {
    pub entries: HashMap<String, VersionedValue>,
    pub peer_count: usize,
}

// ── Simulation trace (complete run output) ───────────────────────────────

#[derive(Debug, Clone)]
pub struct SimulationTrace {
    pub name: String,
    pub node_names: Vec<String>,
    pub node_addrs: Vec<ActorAddress>,
    pub topology_edges: Vec<(String, String)>,
    pub events: Vec<GossipEvent>,
    pub snapshots_per_round: Vec<Vec<(String, NodeSnapshot)>>,
    pub num_rounds: usize,
    pub total_keys: usize,
}

// ── Helpers ──────────────────────────────────────────────────────────────

pub fn current_tick(counter: &TickCounter) -> u64 {
    counter.load(Ordering::Relaxed)
}

pub fn resolve_name(registry: &NameRegistry, addr: ActorAddress) -> String {
    registry
        .lock()
        .unwrap()
        .get(&addr)
        .cloned()
        .unwrap_or_else(|| format!("{:?}", &addr.0[..4]))
}

pub fn record_event(log: &EventLog, event: GossipEvent) {
    log.lock().unwrap().push(event);
}