74 lines
2.1 KiB
Rust
74 lines
2.1 KiB
Rust
use super::blockchain::SavedBlock;
|
|
use super::connection::Web3Connection;
|
|
use super::connections::Web3Connections;
|
|
use ethers::prelude::{H256, U64};
|
|
use serde::Serialize;
|
|
use std::fmt;
|
|
use std::sync::Arc;
|
|
|
|
/// A collection of Web3Connections that are on the same block.
|
|
/// Serialize is so we can print it on our debug endpoint
|
|
#[derive(Clone, Default, Serialize)]
|
|
pub struct ConsensusConnections {
|
|
// TODO: store ArcBlock instead?
|
|
pub(super) head_block: Option<SavedBlock>,
|
|
// TODO: this should be able to serialize, but it isn't
|
|
#[serde(skip_serializing)]
|
|
pub(super) conns: Vec<Arc<Web3Connection>>,
|
|
pub(super) num_checked_conns: usize,
|
|
pub(super) includes_backups: bool,
|
|
}
|
|
|
|
impl ConsensusConnections {
|
|
pub fn num_conns(&self) -> usize {
|
|
self.conns.len()
|
|
}
|
|
|
|
pub fn sum_soft_limit(&self) -> u32 {
|
|
self.conns.iter().fold(0, |sum, rpc| sum + rpc.soft_limit)
|
|
}
|
|
|
|
// TODO: sum_hard_limit?
|
|
}
|
|
|
|
impl fmt::Debug for ConsensusConnections {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
// TODO: the default formatter takes forever to write. this is too quiet though
|
|
// TODO: print the actual conns?
|
|
f.debug_struct("ConsensusConnections")
|
|
.field("head_block", &self.head_block)
|
|
.field("num_conns", &self.conns.len())
|
|
.finish_non_exhaustive()
|
|
}
|
|
}
|
|
|
|
impl Web3Connections {
|
|
pub fn head_block(&self) -> Option<SavedBlock> {
|
|
self.synced_connections.load().head_block.clone()
|
|
}
|
|
|
|
pub fn head_block_hash(&self) -> Option<H256> {
|
|
self.synced_connections
|
|
.load()
|
|
.head_block
|
|
.as_ref()
|
|
.map(|head_block| head_block.hash())
|
|
}
|
|
|
|
pub fn head_block_num(&self) -> Option<U64> {
|
|
self.synced_connections
|
|
.load()
|
|
.head_block
|
|
.as_ref()
|
|
.map(|head_block| head_block.number())
|
|
}
|
|
|
|
pub fn synced(&self) -> bool {
|
|
!self.synced_connections.load().conns.is_empty()
|
|
}
|
|
|
|
pub fn num_synced_rpcs(&self) -> usize {
|
|
self.synced_connections.load().conns.len()
|
|
}
|
|
}
|