client_pool.rs mod

2024-12-13 16:18:54 +01:00
parent c7b3bd1608
commit 747675f52f
1 changed files with 117 additions and 2 deletions
@@ -1,6 +1,121 @@
-//!
+//! use crate::mixnet::{MixnetClient, MixnetClientBuilder, NymNetworkDetails};
+//! use anyhow::Result;
+//! use std::fmt;
+//! use std::sync::Arc;
+//! use tokio::sync::RwLock;
+//! use tracing::{debug, info, warn};

-// TODO EXAMPLE ONCE FINISHED
+//! pub struct ClientPool {
+//!     clients: Arc<RwLock<Vec<Arc<MixnetClient>>>>, // Collection of clients waiting to be used which are popped off in  get_mixnet_client()
+//!     client_pool_reserve_number: usize, // Default # of clients to have available in pool in reserve waiting for incoming connections
+//! }
+
+//! // This is only necessary for when you're wanting to check the addresses of the clients that are currently in the pool.
+//! impl fmt::Debug for ClientPool {
+//!     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+//!         let clients_debug = match self.clients.try_read() {
+//!             Ok(clients) => {
+//!                 if f.alternate() {
+//!                     // pretty
+//!                     clients
+//!                         .iter()
+//!                         .enumerate()
+//!                         .map(|(i, client)| {
+//!                             format!("\n      {}: {}", i, client.nym_address().to_string())
+//!                         })
+//!                         .collect::<Vec<_>>()
+//!                         .join(",")
+//!                 } else {
+//!                     // compact
+//!                     clients
+//!                         .iter()
+//!                         .map(|client| client.nym_address().to_string())
+//!                         .collect::<Vec<_>>()
+//!                         .join(", ")
+//!                 }
+//!             }
+//!             Err(_) => "<locked>".to_string(),
+//!         };
+//!
+//!         let mut debug_struct = f.debug_struct("Pool");
+//!         debug_struct
+//!             .field(
+//!                 "client_pool_reserve_number",
+//!                 &self.client_pool_reserve_number,
+//!             )
+//!             .field("clients", &format_args!("[{}]", clients_debug));
+//!         debug_struct.finish()
+//!     }
+//! }
+
+//! impl ClientPool {
+//!     pub fn new(client_pool_reserve_number: usize) -> Self {
+//!         ClientPool {
+//!             clients: Arc::new(RwLock::new(Vec::new())),
+//!             client_pool_reserve_number,
+//!         }
+//!     }
+//!
+//!     // The loop here is simple: if there aren't enough clients, create more. If you set clients to 0, repeatedly just sleep.
+//!     pub async fn start(&self) -> Result<()> {
+//!         loop {
+//!             let spawned_clients = self.clients.read().await.len();
+//!             let addresses = self;
+//!             debug!(
+//!                 "Currently spawned clients: {}: {:?}",
+//!                 spawned_clients, addresses
+//!             );
+//!             if spawned_clients >= self.client_pool_reserve_number {
+//!                 debug!("Got enough clients already: sleeping");
+//!             } else {
+//!                 info!(
+//!                     "Clients in reserve = {}, reserve amount = {}, spawning new client",
+//!                     spawned_clients, self.client_pool_reserve_number
+//!                 );
+//!                 let client = loop {
+//!                     let net = NymNetworkDetails::new_from_env();
+//!                     match MixnetClientBuilder::new_ephemeral()
+//!                         .network_details(net)
+//!                         .build()?
+//!                         .connect_to_mixnet()
+//!                         .await
+//!                     {
+//!                         Ok(client) => break client,
+//!                         Err(err) => {
+//!                             warn!("Error creating client: {:?}, will retry in 100ms", err);
+//!                             tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
+//!                         }
+//!                     }
+//!                 };
+//!                 self.clients.write().await.push(Arc::new(client));
+//!             }
+//!             tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
+//!         }
+//!     }
+//!
+//!     pub async fn get_mixnet_client(&self) -> Option<MixnetClient> {
+//!         debug!("Grabbing client from pool");
+//!         let mut clients = self.clients.write().await;
+//!         clients
+//!             .pop()
+//!             .and_then(|arc_client| Arc::try_unwrap(arc_client).ok())
+//!     }
+//!
+//!     pub async fn get_client_count(&self) -> usize {
+//!         self.clients.read().await.len()
+//!     }
+//!
+//!     pub async fn get_pool_reserve(&self) -> usize {
+//!         self.client_pool_reserve_number
+//!     }
+//!
+//!     pub fn clone(&self) -> Self {
+//!         Self {
+//!             clients: Arc::clone(&self.clients),
+//!             client_pool_reserve_number: *&self.client_pool_reserve_number,
+//!         }
+//!     }
+//! }

 mod client_pool;