kaspa-consensus 0.0.2

pub mod body_processor;
pub mod deps_manager;
pub mod header_processor;
pub mod pruning_processor;
pub mod virtual_processor;

use std::sync::atomic::{AtomicU64, Ordering};

#[derive(Default)]
pub struct ProcessingCounters {
    pub blocks_submitted: AtomicU64,
    pub header_counts: AtomicU64,
    pub dep_counts: AtomicU64,
    pub body_counts: AtomicU64,
    pub txs_counts: AtomicU64,
    pub chain_block_counts: AtomicU64,
}

impl ProcessingCounters {
    pub fn snapshot(&self) -> ProcessingCountersSnapshot {
        ProcessingCountersSnapshot {
            blocks_submitted: self.blocks_submitted.load(Ordering::SeqCst),
            header_counts: self.header_counts.load(Ordering::SeqCst),
            dep_counts: self.dep_counts.load(Ordering::SeqCst),
            body_counts: self.body_counts.load(Ordering::SeqCst),
            txs_counts: self.txs_counts.load(Ordering::SeqCst),
            chain_block_counts: self.chain_block_counts.load(Ordering::SeqCst),
        }
    }
}

#[derive(Debug, PartialEq, Eq)]
pub struct ProcessingCountersSnapshot {
    pub blocks_submitted: u64,
    pub header_counts: u64,
    pub dep_counts: u64,
    pub body_counts: u64,
    pub txs_counts: u64,
    pub chain_block_counts: u64,
}

impl core::ops::Sub for &ProcessingCountersSnapshot {
    type Output = ProcessingCountersSnapshot;

    fn sub(self, rhs: Self) -> Self::Output {
        Self::Output {
            blocks_submitted: self.blocks_submitted - rhs.blocks_submitted,
            header_counts: self.header_counts - rhs.header_counts,
            dep_counts: self.dep_counts - rhs.dep_counts,
            body_counts: self.body_counts - rhs.body_counts,
            txs_counts: self.txs_counts - rhs.txs_counts,
            chain_block_counts: self.chain_block_counts - rhs.chain_block_counts,
        }
    }
}