Struct ShardClientPool 

pub struct ShardClientPool { /* private fields */ }

A pool of shard client connections.

The connection pool maintains a cached set of connections to shards, creating new connections on demand and reusing existing ones.

Thread Safety

The pool is thread-safe and can be shared across multiple tasks.

Example

use phago_distributed::rpc::client::ShardClientPool;

let pool = ShardClientPool::new();
pool.register_shard(ShardId::new(0), "127.0.0.1:8080".parse().unwrap());

let client = pool.get_client(ShardId::new(0)).await?;
let health = client.health_check(tarpc::context::current()).await?;

Implementations

impl ShardClientPool

pub fn new() -> Self

Create a new empty connection pool.

pub fn with_config(config: ClientConfig) -> Self

Create a new connection pool with custom configuration.

pub async fn register_shard(&self, shard_id: ShardId, addr: SocketAddr)

Register a shard’s address.

This does not establish a connection immediately; connections are created lazily when get_client is called.

pub async fn unregister_shard(&self, shard_id: ShardId)

Unregister a shard and close any cached connection.

pub async fn get_client( &self, shard_id: ShardId, ) -> Result<ShardServiceClient, Error>

Get a client for the specified shard.

Returns a cached client if available, otherwise creates a new connection.

Errors

Returns an error if the shard is not registered or if the connection fails.

pub async fn get_all_clients(&self) -> Vec<(ShardId, ShardServiceClient)>

Get clients for all registered shards.

Attempts to connect to all shards, returning successfully connected clients. Failed connections are logged but do not cause the entire operation to fail.

pub async fn has_shard(&self, shard_id: ShardId) -> bool

Check if a shard is registered.

pub async fn shard_count(&self) -> usize

Get the number of registered shards.

pub async fn cached_connection_count(&self) -> usize

Get the number of cached connections.

pub async fn clear_cache(&self)

Clear all cached connections.

This forces new connections to be created on the next get_client call.

pub async fn invalidate_client(&self, shard_id: ShardId)

Remove a cached connection for a specific shard.

Useful for forcing a reconnection after a failure.

Trait Implementations

impl Clone for ShardClientPool

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for ShardClientPool

fn default() -> Self

Returns the “default value” for a type.