RocketmqDefaultClient

Struct RocketmqDefaultClient 

Source
pub struct RocketmqDefaultClient<PR = DefaultRemotingRequestProcessor> { /* private fields */ }
Expand description

High-performance async RocketMQ client with connection pooling and auto-reconnection.

§Architecture

┌─────────────────────────────────────────────────────────┐
│            RocketmqDefaultClient<PR>                    │
├─────────────────────────────────────────────────────────┤
│                                                         │
│  ┌────────────────┐      ┌──────────────────┐         │
│  │ Connection Pool│ ───► │NameServer Router │         │
│  │  (DashMap)     │      │  (Health-based)  │         │
│  └────────────────┘      └──────────────────┘         │
│         │                         │                    │
│         ↓                         ↓                    │
│  ┌────────────────┐      ┌──────────────────┐         │
│  │ Request Handler│ ───► │  Response Table  │         │
│  │  (async tasks) │      │   (oneshot rx)   │         │
│  └────────────────┘      └──────────────────┘         │
│                                                         │
└─────────────────────────────────────────────────────────┘

§Key Features

  • Connection Pooling: Reuses TCP connections to brokers/nameservers
  • Auto-Reconnection: Exponential backoff retry on connection failures
  • Smart Routing: Selects healthiest nameserver based on latency/errors
  • Request Multiplexing: Multiple concurrent requests per connection
  • Graceful Shutdown: Drains in-flight requests before closing

§Performance Characteristics

  • Lock Contention: Uses Arc<Mutex<HashMap>> for connection pool
    • ⚠️ TODO: Migrate to DashMap for lock-free reads
  • Memory: O(N) where N = number of unique broker addresses
  • Latency: Single async hop for cached connections, 2-3 hops for new

§Type Parameters

  • PR - Request processor type (default: DefaultRemotingRequestProcessor)

§Example

use std::sync::Arc;

use rocketmq_remoting::clients::RocketmqDefaultClient;
use rocketmq_remoting::runtime::config::client_config::TokioClientConfig;

let config = Arc::new(TokioClientConfig::default());
let processor = Default::default();
let client = RocketmqDefaultClient::new(config, processor);

// Update nameserver list
client
    .update_name_server_address_list(vec!["127.0.0.1:9876".into()])
    .await;

// Send request
let response = client
    .invoke_request(
        None, // use default nameserver
        request, 3000, // 3s timeout
    )
    .await?;

Implementations§

Source§

impl<PR: RequestProcessor + Sync + Clone + 'static> RocketmqDefaultClient<PR>

Source

pub fn new(tokio_client_config: Arc<TokioClientConfig>, processor: PR) -> Self

Source

pub fn new_with_cl( tokio_client_config: Arc<TokioClientConfig>, processor: PR, tx: Option<Sender<ConnectionNetEvent>>, ) -> Self

Source§

impl<PR: RequestProcessor + Sync + Clone + 'static> RocketmqDefaultClient<PR>

Source

pub fn enable_connection_pool( &mut self, max_connections: usize, max_idle_duration: Duration, cleanup_interval: Duration, ) -> JoinHandle<()>

Enable advanced connection pool with metrics and automatic cleanup.

§Arguments
  • max_connections - Maximum number of connections (0 = unlimited)
  • max_idle_duration - Idle timeout (e.g., 5 minutes)
  • cleanup_interval - Cleanup task interval (e.g., 30 seconds)
§Returns

Task handle for the cleanup background task (can be aborted)

§Example
let client =
    RocketmqDefaultClient::new(Arc::new(TokioClientConfig::default()), Default::default());

// Enable connection pool with:
// - Max 1000 connections
// - 5 minute idle timeout
// - 30 second cleanup interval
let cleanup_task =
    client.enable_connection_pool(1000, Duration::from_secs(300), Duration::from_secs(30));

// ... use client ...

// Stop cleanup when shutting down
cleanup_task.abort();
Source

pub fn get_pool_stats(&self) -> Option<PoolStats>

Get connection pool statistics (if enabled).

§Returns
  • Some(stats) - Pool statistics
  • None - Connection pool not enabled
§Example
if let Some(stats) = client.get_pool_stats() {
    println!(
        "Pool: {}/{} connections ({:.1}% util)",
        stats.active(),
        stats.max_connections,
        stats.utilization() * 100.0
    );
    println!("Error rate: {:.2}%", stats.error_rate() * 100.0);
}

Trait Implementations§

Source§

impl<PR: RequestProcessor + Sync + Clone + 'static> RemotingClient for RocketmqDefaultClient<PR>

Source§

async fn invoke_request( &self, addr: Option<&CheetahString>, request: RemotingCommand, timeout_millis: u64, ) -> RocketMQResult<RemotingCommand>

Send request and wait for response with timeout.

§HOT PATH

This is the primary client API - optimize for low latency and high throughput.

§Flow
1. Get/create client connection         (~100ns fast path, ~50ms slow)
2. Spawn send task on runtime           (~10μs)
3. Apply timeout wrapper                (~100ns)
4. Wait for response via oneshot        (network RTT + processing)
5. Unwrap nested Result layers          (~10ns)
§Performance Optimizations
  • Early validation: Check client availability before expensive spawn
  • Flat error handling: Reduce nested match overhead
  • Direct await: Spawn only for timeout enforcement (could be optimized further)
§Error Handling

Returns RocketmqError::RemoteError for all failures:

  • Client unavailable (no connection)
  • Network I/O error (send/recv failure)
  • Timeout (no response within deadline)
  • Task spawn error (runtime shutdown)
§Arguments
  • addr - Target address (None = use nameserver)
  • request - Command to send
  • timeout_millis - Max wait time for response
§Examples
let request = RemotingCommand::create_request_command(/* ... */);
let response = client.invoke_request(
    Some(&"127.0.0.1:10911".into()),
    request,
    3000 // 3 second timeout
).await?;
Source§

async fn update_name_server_address_list(&self, addrs: Vec<CheetahString>)

Updates the list of name remoting_server addresses. Read more
Source§

fn get_name_server_address_list(&self) -> &[CheetahString]

Retrieves the current list of name remoting_server addresses. Read more
Source§

fn get_available_name_srv_list(&self) -> Vec<CheetahString>

Retrieves a list of available name remoting_server addresses. Read more
Source§

async fn invoke_request_oneway( &self, addr: &CheetahString, request: RemotingCommand, timeout_millis: u64, )

Invokes a command on a specified address without waiting for a response. Read more
Source§

fn is_address_reachable(&mut self, addr: &CheetahString)

Checks if a specified address is reachable. Read more
Source§

fn close_clients(&mut self, addrs: Vec<String>)

Closes clients connected to the specified addresses. Read more
Source§

fn register_processor(&mut self, processor: impl RequestProcessor + Sync)

Source§

impl<PR: RequestProcessor + Sync + Clone + 'static> RemotingService for RocketmqDefaultClient<PR>

Source§

async fn start(&self, this: WeakArcMut<Self>)

Asynchronously starts the remoting service. Read more
Source§

fn shutdown(&mut self)

Shuts down the remoting service. Read more
Source§

fn register_rpc_hook(&mut self, hook: Arc<dyn RPCHook>)

Registers an RPC hook. Read more
Source§

fn clear_rpc_hook(&mut self)

Clears all registered RPC hooks. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> Conv for T

Source§

fn conv<T>(self) -> T
where Self: Into<T>,

Converts self into T using Into<T>. Read more
Source§

impl<T> FmtForward for T

Source§

fn fmt_binary(self) -> FmtBinary<Self>
where Self: Binary,

Causes self to use its Binary implementation when Debug-formatted.
Source§

fn fmt_display(self) -> FmtDisplay<Self>
where Self: Display,

Causes self to use its Display implementation when Debug-formatted.
Source§

fn fmt_lower_exp(self) -> FmtLowerExp<Self>
where Self: LowerExp,

Causes self to use its LowerExp implementation when Debug-formatted.
Source§

fn fmt_lower_hex(self) -> FmtLowerHex<Self>
where Self: LowerHex,

Causes self to use its LowerHex implementation when Debug-formatted.
Source§

fn fmt_octal(self) -> FmtOctal<Self>
where Self: Octal,

Causes self to use its Octal implementation when Debug-formatted.
Source§

fn fmt_pointer(self) -> FmtPointer<Self>
where Self: Pointer,

Causes self to use its Pointer implementation when Debug-formatted.
Source§

fn fmt_upper_exp(self) -> FmtUpperExp<Self>
where Self: UpperExp,

Causes self to use its UpperExp implementation when Debug-formatted.
Source§

fn fmt_upper_hex(self) -> FmtUpperHex<Self>
where Self: UpperHex,

Causes self to use its UpperHex implementation when Debug-formatted.
Source§

fn fmt_list(self) -> FmtList<Self>
where &'a Self: for<'a> IntoIterator,

Formats each item in a sequence. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> Pipe for T
where T: ?Sized,

Source§

fn pipe<R>(self, func: impl FnOnce(Self) -> R) -> R
where Self: Sized,

Pipes by value. This is generally the method you want to use. Read more
Source§

fn pipe_ref<'a, R>(&'a self, func: impl FnOnce(&'a Self) -> R) -> R
where R: 'a,

Borrows self and passes that borrow into the pipe function. Read more
Source§

fn pipe_ref_mut<'a, R>(&'a mut self, func: impl FnOnce(&'a mut Self) -> R) -> R
where R: 'a,

Mutably borrows self and passes that borrow into the pipe function. Read more
Source§

fn pipe_borrow<'a, B, R>(&'a self, func: impl FnOnce(&'a B) -> R) -> R
where Self: Borrow<B>, B: 'a + ?Sized, R: 'a,

Borrows self, then passes self.borrow() into the pipe function. Read more
Source§

fn pipe_borrow_mut<'a, B, R>( &'a mut self, func: impl FnOnce(&'a mut B) -> R, ) -> R
where Self: BorrowMut<B>, B: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.borrow_mut() into the pipe function. Read more
Source§

fn pipe_as_ref<'a, U, R>(&'a self, func: impl FnOnce(&'a U) -> R) -> R
where Self: AsRef<U>, U: 'a + ?Sized, R: 'a,

Borrows self, then passes self.as_ref() into the pipe function.
Source§

fn pipe_as_mut<'a, U, R>(&'a mut self, func: impl FnOnce(&'a mut U) -> R) -> R
where Self: AsMut<U>, U: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.as_mut() into the pipe function.
Source§

fn pipe_deref<'a, T, R>(&'a self, func: impl FnOnce(&'a T) -> R) -> R
where Self: Deref<Target = T>, T: 'a + ?Sized, R: 'a,

Borrows self, then passes self.deref() into the pipe function.
Source§

fn pipe_deref_mut<'a, T, R>( &'a mut self, func: impl FnOnce(&'a mut T) -> R, ) -> R
where Self: DerefMut<Target = T> + Deref, T: 'a + ?Sized, R: 'a,

Mutably borrows self, then passes self.deref_mut() into the pipe function.
Source§

impl<T> PolicyExt for T
where T: ?Sized,

Source§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
Source§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
Source§

impl<T> Tap for T

Source§

fn tap(self, func: impl FnOnce(&Self)) -> Self

Immutable access to a value. Read more
Source§

fn tap_mut(self, func: impl FnOnce(&mut Self)) -> Self

Mutable access to a value. Read more
Source§

fn tap_borrow<B>(self, func: impl FnOnce(&B)) -> Self
where Self: Borrow<B>, B: ?Sized,

Immutable access to the Borrow<B> of a value. Read more
Source§

fn tap_borrow_mut<B>(self, func: impl FnOnce(&mut B)) -> Self
where Self: BorrowMut<B>, B: ?Sized,

Mutable access to the BorrowMut<B> of a value. Read more
Source§

fn tap_ref<R>(self, func: impl FnOnce(&R)) -> Self
where Self: AsRef<R>, R: ?Sized,

Immutable access to the AsRef<R> view of a value. Read more
Source§

fn tap_ref_mut<R>(self, func: impl FnOnce(&mut R)) -> Self
where Self: AsMut<R>, R: ?Sized,

Mutable access to the AsMut<R> view of a value. Read more
Source§

fn tap_deref<T>(self, func: impl FnOnce(&T)) -> Self
where Self: Deref<Target = T>, T: ?Sized,

Immutable access to the Deref::Target of a value. Read more
Source§

fn tap_deref_mut<T>(self, func: impl FnOnce(&mut T)) -> Self
where Self: DerefMut<Target = T> + Deref, T: ?Sized,

Mutable access to the Deref::Target of a value. Read more
Source§

fn tap_dbg(self, func: impl FnOnce(&Self)) -> Self

Calls .tap() only in debug builds, and is erased in release builds.
Source§

fn tap_mut_dbg(self, func: impl FnOnce(&mut Self)) -> Self

Calls .tap_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_borrow_dbg<B>(self, func: impl FnOnce(&B)) -> Self
where Self: Borrow<B>, B: ?Sized,

Calls .tap_borrow() only in debug builds, and is erased in release builds.
Source§

fn tap_borrow_mut_dbg<B>(self, func: impl FnOnce(&mut B)) -> Self
where Self: BorrowMut<B>, B: ?Sized,

Calls .tap_borrow_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_ref_dbg<R>(self, func: impl FnOnce(&R)) -> Self
where Self: AsRef<R>, R: ?Sized,

Calls .tap_ref() only in debug builds, and is erased in release builds.
Source§

fn tap_ref_mut_dbg<R>(self, func: impl FnOnce(&mut R)) -> Self
where Self: AsMut<R>, R: ?Sized,

Calls .tap_ref_mut() only in debug builds, and is erased in release builds.
Source§

fn tap_deref_dbg<T>(self, func: impl FnOnce(&T)) -> Self
where Self: Deref<Target = T>, T: ?Sized,

Calls .tap_deref() only in debug builds, and is erased in release builds.
Source§

fn tap_deref_mut_dbg<T>(self, func: impl FnOnce(&mut T)) -> Self
where Self: DerefMut<Target = T> + Deref, T: ?Sized,

Calls .tap_deref_mut() only in debug builds, and is erased in release builds.
Source§

impl<T> TryConv for T

Source§

fn try_conv<T>(self) -> Result<T, Self::Error>
where Self: TryInto<T>,

Attempts to convert self into T using TryInto<T>. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

impl<T> ErasedDestructor for T
where T: 'static,

Source§

impl<T> MaybeSendSync for T