Drazen Urch
139 lines
4.1 KiB
Rust
139 lines
4.1 KiB
Rust
// Copyright 2022 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::cache::Cache;
|
|
use crate::mix_nodes::location::Location;
|
|
use mixnet_contract_common::{Addr, Coin, Layer, MixNode};
|
|
use serde::Deserialize;
|
|
use serde::Serialize;
|
|
use std::sync::Arc;
|
|
use std::time::SystemTime;
|
|
use tokio::sync::RwLock;
|
|
use validator_client::models::SelectionChance;
|
|
|
|
#[derive(Clone, Debug, Serialize, JsonSchema, PartialEq)]
|
|
#[serde(rename_all = "snake_case")]
|
|
pub(crate) enum MixnodeStatus {
|
|
Active, // in both the active set and the rewarded set
|
|
Standby, // only in the rewarded set
|
|
Inactive, // in neither the rewarded set nor the active set
|
|
}
|
|
|
|
#[derive(Clone, Debug, Serialize, JsonSchema)]
|
|
pub(crate) struct PrettyDetailedMixNodeBond {
|
|
pub location: Option<Location>,
|
|
pub status: MixnodeStatus,
|
|
pub pledge_amount: Coin,
|
|
pub total_delegation: Coin,
|
|
pub owner: Addr,
|
|
pub layer: Layer,
|
|
pub mix_node: MixNode,
|
|
}
|
|
|
|
pub(crate) struct MixNodeCache {
|
|
pub(crate) descriptions: Cache<NodeDescription>,
|
|
pub(crate) node_stats: Cache<NodeStats>,
|
|
pub(crate) econ_stats: Cache<EconomicDynamicsStats>,
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub(crate) struct ThreadsafeMixNodeCache {
|
|
inner: Arc<RwLock<MixNodeCache>>,
|
|
}
|
|
|
|
impl ThreadsafeMixNodeCache {
|
|
pub(crate) fn new() -> Self {
|
|
ThreadsafeMixNodeCache {
|
|
inner: Arc::new(RwLock::new(MixNodeCache {
|
|
descriptions: Cache::new(),
|
|
node_stats: Cache::new(),
|
|
econ_stats: Cache::new(),
|
|
})),
|
|
}
|
|
}
|
|
|
|
pub(crate) async fn get_description(&self, identity_key: &str) -> Option<NodeDescription> {
|
|
self.inner.read().await.descriptions.get(identity_key)
|
|
}
|
|
|
|
pub(crate) async fn get_node_stats(&self, identity_key: &str) -> Option<NodeStats> {
|
|
self.inner.read().await.node_stats.get(identity_key)
|
|
}
|
|
|
|
pub(crate) async fn get_econ_stats(&self, identity_key: &str) -> Option<EconomicDynamicsStats> {
|
|
self.inner.read().await.econ_stats.get(identity_key)
|
|
}
|
|
|
|
pub(crate) async fn set_description(&self, identity_key: &str, description: NodeDescription) {
|
|
self.inner
|
|
.write()
|
|
.await
|
|
.descriptions
|
|
.set(identity_key, description);
|
|
}
|
|
|
|
pub(crate) async fn set_node_stats(&self, identity_key: &str, node_stats: NodeStats) {
|
|
self.inner
|
|
.write()
|
|
.await
|
|
.node_stats
|
|
.set(identity_key, node_stats);
|
|
}
|
|
|
|
pub(crate) async fn set_econ_stats(
|
|
&self,
|
|
identity_key: &str,
|
|
econ_stats: EconomicDynamicsStats,
|
|
) {
|
|
self.inner
|
|
.write()
|
|
.await
|
|
.econ_stats
|
|
.set(identity_key, econ_stats);
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Deserialize, PartialEq, Serialize, JsonSchema)]
|
|
pub(crate) struct NodeDescription {
|
|
pub(crate) name: String,
|
|
pub(crate) description: String,
|
|
pub(crate) link: String,
|
|
pub(crate) location: String,
|
|
}
|
|
|
|
#[derive(Serialize, Clone, Deserialize, JsonSchema)]
|
|
pub(crate) struct NodeStats {
|
|
#[serde(
|
|
serialize_with = "humantime_serde::serialize",
|
|
deserialize_with = "humantime_serde::deserialize"
|
|
)]
|
|
update_time: SystemTime,
|
|
|
|
#[serde(
|
|
serialize_with = "humantime_serde::serialize",
|
|
deserialize_with = "humantime_serde::deserialize"
|
|
)]
|
|
previous_update_time: SystemTime,
|
|
|
|
packets_received_since_startup: u64,
|
|
packets_sent_since_startup: u64,
|
|
packets_explicitly_dropped_since_startup: u64,
|
|
packets_received_since_last_update: u64,
|
|
packets_sent_since_last_update: u64,
|
|
packets_explicitly_dropped_since_last_update: u64,
|
|
}
|
|
|
|
#[derive(Clone, Serialize, Deserialize, JsonSchema)]
|
|
pub(crate) struct EconomicDynamicsStats {
|
|
pub(crate) stake_saturation: f32,
|
|
|
|
pub(crate) active_set_inclusion_probability: SelectionChance,
|
|
pub(crate) reserve_set_inclusion_probability: SelectionChance,
|
|
|
|
pub(crate) estimated_total_node_reward: u64,
|
|
pub(crate) estimated_operator_reward: u64,
|
|
pub(crate) estimated_delegators_reward: u64,
|
|
|
|
pub(crate) current_interval_uptime: u8,
|
|
}
|