pezkuwi-telemetry/backend/src/node.rs

use bytes::Bytes;
use std::sync::Arc;

use crate::types::{NodeId, NodeDetails, NodeStats, NodeHardware, NodeLocation, BlockDetails, Block, Timestamp};
use crate::util::now;

pub mod message;
pub mod connector;

use message::SystemInterval;

/// Minimum time between block below broadcasting updates to the browser gets throttled, in ms.
const THROTTLE_THRESHOLD: u64 = 100;
/// Minimum time of intervals for block updates sent to the browser when throttled, in ms.
const THROTTLE_INTERVAL: u64 = 1000;

pub struct Node {
    /// Static details
    details: NodeDetails,
    /// Basic stats
    stats: NodeStats,
    /// Best block
    best: BlockDetails,
    /// Finalized block
    finalized: Block,
    /// Timer for throttling block updates
    throttle: u64,
    /// Hardware stats over time
    hardware: NodeHardware,
    /// Physical location details
    location: Option<Arc<NodeLocation>>,
    /// Flag marking if the node is stale (not syncing or producing blocks)
    stale: bool,
    /// Connected at timestamp
    connected_at: Timestamp,
    /// Network state
    network_state: Option<Bytes>,
}

impl Node {
    pub fn new(details: NodeDetails) -> Self {
        Node {

            details,
            stats: NodeStats {
                txcount: 0,
                peers: 0,
            },
            best: BlockDetails {
                block: Block::zero(),
                block_timestamp: now(),
                block_time: 0,
                propagation_time: None,
            },
            finalized: Block::zero(),
            throttle: 0,
            hardware: NodeHardware::default(),
            location: None,
            stale: false,
            connected_at: now(),
            network_state: None,
        }
    }

    pub fn details(&self) -> &NodeDetails {
        &self.details
    }

    pub fn stats(&self) -> &NodeStats {
        &self.stats
    }

    pub fn best(&self) -> &Block {
        &self.best.block
    }

    pub fn finalized(&self) -> &Block {
        &self.finalized
    }

    pub fn hardware(&self) -> &NodeHardware {
        &self.hardware
    }

    pub fn location(&self) -> Option<&NodeLocation> {
        match self.location {
            Some(ref location) => Some(&**location),
            None => None
        }
    }

    pub fn update_location(&mut self, location: Arc<NodeLocation>) {
        self.location = Some(location);
    }

    pub fn block_details(&self) -> &BlockDetails {
        &self.best
    }

    pub fn update_block(&mut self, block: Block, timestamp: u64, propagation_time: Option<u64>) -> Option<&BlockDetails> {
        if block.height > self.best.block.height {
            self.stale = false;
            self.best.block = block;
            self.best.block_time = timestamp - self.best.block_timestamp;
            self.best.block_timestamp = timestamp;
            self.best.propagation_time = propagation_time;

            if self.throttle < timestamp {
                if self.best.block_time <= THROTTLE_THRESHOLD {
                    self.throttle = timestamp + THROTTLE_INTERVAL;
                }

                return Some(&self.best);
            }
        }

        None
    }

    pub fn update_hardware(&mut self, interval: &SystemInterval) -> bool {
        let mut changed = false;

        if let Some(cpu) = interval.cpu {
            changed |= self.hardware.cpu.push(cpu);
        }
        if let Some(memory) = interval.memory {
            changed |= self.hardware.memory.push(memory);
        }
        if let Some(upload) = interval.bandwidth_upload {
            changed |= self.hardware.upload.push(upload);
        }
        if let Some(download) = interval.bandwidth_download {
            changed |= self.hardware.download.push(download);
        }
        self.hardware.chart_stamps.push(now() as f64);

        changed
    }

    pub fn update_stats(&mut self, interval: &SystemInterval) -> Option<&NodeStats> {
        if self.stats != interval.stats {
            self.stats = interval.stats;
            Some(&self.stats)
        } else {
            None
        }
    }

    pub fn update_finalized(&mut self, block: Block) -> Option<&Block> {
        if block.height > self.finalized.height {
            self.finalized = block;
            Some(self.finalized())
        } else {
            None
        }
    }

    pub fn update_stale(&mut self, threshold: u64) -> bool {
        if self.best.block_timestamp < threshold {
            self.stale = true;
        }

        self.stale
    }

    pub fn stale(&self) -> bool {
        self.stale
    }

    pub fn set_validator_address(&mut self, addr: Box<str>) {
        self.details.validator = Some(addr);
    }

    pub fn set_network_state(&mut self, state: Bytes) {
        self.network_state = Some(state);
    }

    pub fn network_state(&self) -> Option<Bytes> {
        use serde::Deserialize;
        use serde_json::value::RawValue;

        #[derive(Deserialize)]
        struct Wrapper<'a> {
            #[serde(borrow)]
            #[serde(alias = "network_state")]
            state: &'a RawValue,
        }

        let raw = self.network_state.as_ref()?;
        let wrap: Wrapper = serde_json::from_slice(raw).ok()?;
        let json = wrap.state.get();

        // Handle old nodes that exposed network_state as stringified JSON
        if let Ok(stringified) = serde_json::from_str::<String>(json) {
            Some(stringified.into())
        } else {
            Some(json.into())
        }
    }

    pub fn connected_at(&self) -> Timestamp {
        self.connected_at
    }
}