Jędrzej Stuczyński
98 lines
2.9 KiB
Rust
98 lines
2.9 KiB
Rust
// Copyright 2023 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
use async_trait::async_trait;
|
|
use nym_sdk::{NymApiTopologyProvider, NymApiTopologyProviderConfig, UserAgent};
|
|
use nym_topology::{gateway, NymTopology, TopologyProvider};
|
|
use std::sync::Arc;
|
|
use std::time::Duration;
|
|
use time::OffsetDateTime;
|
|
use tokio::sync::Mutex;
|
|
use tracing::debug;
|
|
use url::Url;
|
|
|
|
#[derive(Clone)]
|
|
pub struct GatewayTopologyProvider {
|
|
inner: Arc<Mutex<GatewayTopologyProviderInner>>,
|
|
}
|
|
|
|
impl GatewayTopologyProvider {
|
|
pub fn new(
|
|
gateway_node: gateway::LegacyNode,
|
|
cache_ttl: Duration,
|
|
user_agent: UserAgent,
|
|
nym_api_url: Vec<Url>,
|
|
) -> GatewayTopologyProvider {
|
|
GatewayTopologyProvider {
|
|
inner: Arc::new(Mutex::new(GatewayTopologyProviderInner {
|
|
inner: NymApiTopologyProvider::new(
|
|
NymApiTopologyProviderConfig {
|
|
min_mixnode_performance: 50,
|
|
min_gateway_performance: 0,
|
|
},
|
|
nym_api_url,
|
|
env!("CARGO_PKG_VERSION").to_string(),
|
|
Some(user_agent),
|
|
),
|
|
cache_ttl,
|
|
cached_at: OffsetDateTime::UNIX_EPOCH,
|
|
cached: None,
|
|
gateway_node,
|
|
})),
|
|
}
|
|
}
|
|
}
|
|
|
|
struct GatewayTopologyProviderInner {
|
|
inner: NymApiTopologyProvider,
|
|
cache_ttl: Duration,
|
|
cached_at: OffsetDateTime,
|
|
cached: Option<NymTopology>,
|
|
gateway_node: gateway::LegacyNode,
|
|
}
|
|
|
|
impl GatewayTopologyProviderInner {
|
|
fn cached_topology(&self) -> Option<NymTopology> {
|
|
if let Some(cached_topology) = &self.cached {
|
|
if self.cached_at + self.cache_ttl > OffsetDateTime::now_utc() {
|
|
return Some(cached_topology.clone());
|
|
}
|
|
}
|
|
|
|
None
|
|
}
|
|
|
|
async fn update_cache(&mut self) -> Option<NymTopology> {
|
|
let updated_cache = match self.inner.get_new_topology().await {
|
|
None => None,
|
|
Some(mut base) => {
|
|
if !base.gateway_exists(&self.gateway_node.identity_key) {
|
|
debug!(
|
|
"{} didn't exist in topology. inserting it.",
|
|
self.gateway_node.identity_key
|
|
);
|
|
base.insert_gateway(self.gateway_node.clone());
|
|
}
|
|
Some(base)
|
|
}
|
|
};
|
|
|
|
self.cached_at = OffsetDateTime::now_utc();
|
|
self.cached = updated_cache.clone();
|
|
|
|
updated_cache
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl TopologyProvider for GatewayTopologyProvider {
|
|
async fn get_new_topology(&mut self) -> Option<NymTopology> {
|
|
let mut guard = self.inner.lock().await;
|
|
// check the cache
|
|
if let Some(cached) = guard.cached_topology() {
|
|
return Some(cached);
|
|
}
|
|
guard.update_cache().await
|
|
}
|
|
}
|