Jędrzej Stuczyński
4897cb0ce4
This includes: - mixnet contract cache - described nodes cache - nodes annotations cache (performance) those changes include taking some code developed for the purposes of #6277
67 lines
2.3 KiB
Rust
67 lines
2.3 KiB
Rust
// Copyright 2025 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
use nym_api_requests::models::{DescribedNodeTypeV2, NymNodeDataV2, NymNodeDescriptionV2};
|
|
use nym_mixnet_contract_common::NodeId;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::collections::HashMap;
|
|
use std::net::IpAddr;
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
pub struct DescribedNodes {
|
|
pub(crate) nodes: HashMap<NodeId, NymNodeDescriptionV2>,
|
|
pub(crate) addresses_cache: HashMap<IpAddr, NodeId>,
|
|
}
|
|
|
|
impl DescribedNodes {
|
|
pub fn force_update(&mut self, node: NymNodeDescriptionV2) {
|
|
for ip in &node.description.host_information.ip_address {
|
|
self.addresses_cache.insert(*ip, node.node_id);
|
|
}
|
|
self.nodes.insert(node.node_id, node);
|
|
}
|
|
|
|
pub fn get_description(&self, node_id: &NodeId) -> Option<&NymNodeDataV2> {
|
|
self.nodes.get(node_id).map(|n| &n.description)
|
|
}
|
|
|
|
pub fn get_node(&self, node_id: &NodeId) -> Option<&NymNodeDescriptionV2> {
|
|
self.nodes.get(node_id)
|
|
}
|
|
|
|
pub fn all_nodes(&self) -> impl Iterator<Item = &NymNodeDescriptionV2> {
|
|
self.nodes.values()
|
|
}
|
|
|
|
pub fn all_nym_nodes(&self) -> impl Iterator<Item = &NymNodeDescriptionV2> {
|
|
self.nodes
|
|
.values()
|
|
.filter(|n| n.contract_node_type == DescribedNodeTypeV2::NymNode)
|
|
}
|
|
|
|
pub fn mixing_nym_nodes(&self) -> impl Iterator<Item = &NymNodeDescriptionV2> {
|
|
self.nodes
|
|
.values()
|
|
.filter(|n| n.contract_node_type == DescribedNodeTypeV2::NymNode)
|
|
.filter(|n| n.description.declared_role.mixnode)
|
|
}
|
|
|
|
pub fn entry_capable_nym_nodes(&self) -> impl Iterator<Item = &NymNodeDescriptionV2> {
|
|
self.nodes
|
|
.values()
|
|
.filter(|n| n.contract_node_type == DescribedNodeTypeV2::NymNode)
|
|
.filter(|n| n.description.declared_role.entry)
|
|
}
|
|
|
|
pub fn exit_capable_nym_nodes(&self) -> impl Iterator<Item = &NymNodeDescriptionV2> {
|
|
self.nodes
|
|
.values()
|
|
.filter(|n| n.contract_node_type == DescribedNodeTypeV2::NymNode)
|
|
.filter(|n| n.description.declared_role.can_operate_exit_gateway())
|
|
}
|
|
|
|
pub fn node_with_address(&self, address: IpAddr) -> Option<NodeId> {
|
|
self.addresses_cache.get(&address).copied()
|
|
}
|
|
}
|