Struct mz_kafka_util::client::BrokerRewritingClientContext

pub struct BrokerRewritingClientContext<C> {
    inner: C,
    rewrites: BTreeMap<BrokerAddr, Arc<dyn Fn() -> BrokerRewrite + Send + Sync>>,
}

A client context that supports rewriting broker addresses.

Fields

inner: C

rewrites: BTreeMap<BrokerAddr, Arc<dyn Fn() -> BrokerRewrite + Send + Sync>>

Implementations

impl<C> BrokerRewritingClientContext<C>

pub fn new(inner: C) -> BrokerRewritingClientContext<C>

Constructs a new context that wraps inner.

pub fn add_broker_rewrite<F>(&mut self, broker: BrokerAddr, rewrite: F)where F: Fn() -> BrokerRewrite + Send + Sync + 'static,

Adds a broker rewrite rule.

rewrite is a function that returns a BrokerRewrite that specifies how to rewrite the address for broker.

The function is invoked by librdkafka on every connection attempt to the broker. This permits the rewrite to evolve over time, for example, if the rewrite is for a tunnel whose address changes if the tunnel fails and restarts.

pub fn inner(&self) -> &C

Returns a reference to the wrapped context.

Trait Implementations

impl<C> ClientContext for BrokerRewritingClientContext<C>where C: ClientContext,

const ENABLE_REFRESH_OAUTH_TOKEN: bool = C::ENABLE_REFRESH_OAUTH_TOKEN

Whether to periodically refresh the SASL OAUTHBEARER token by calling ClientContext::generate_oauth_token.

fn rewrite_broker_addr(&self, addr: BrokerAddr) -> BrokerAddr

Rewrites a broker address for DNS resolution.

fn log(&self, level: RDKafkaLogLevel, fac: &str, log_message: &str)

Receives log lines from librdkafka.

fn error(&self, error: KafkaError, reason: &str)

Receives global errors from the librdkafka client.

fn stats(&self, statistics: Statistics)

Receives the decoded statistics of the librdkafka client. To enable, the statistics.interval.ms configuration parameter must be specified.

fn stats_raw(&self, statistics: &[u8])

Receives the JSON-encoded statistics of the librdkafka client. To enable, the statistics.interval.ms configuration parameter must be specified.

fn generate_oauth_token( &self, oauthbearer_config: Option<&str> ) -> Result<OAuthToken, Box<dyn Error>>

Generates an OAuth token from the provided configuration.

impl<C: Clone> Clone for BrokerRewritingClientContext<C>

fn clone(&self) -> BrokerRewritingClientContext<C>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<C> ConsumerContext for BrokerRewritingClientContext<C>where C: ConsumerContext,

fn rebalance( &self, native_client: &NativeClient, err: RDKafkaRespErr, tpl: &mut TopicPartitionList )

Implements the default rebalancing strategy and calls the pre_rebalance and post_rebalance methods. If this method is overridden, it will be responsibility of the user to call them if needed.

fn pre_rebalance<'a>(&self, rebalance: &Rebalance<'a>)

Pre-rebalance callback. This method will run before the rebalance and should terminate its execution quickly.

fn post_rebalance<'a>(&self, rebalance: &Rebalance<'a>)

Post-rebalance callback. This method will run after the rebalance and should terminate its execution quickly.

fn commit_callback(&self, result: KafkaResult<()>, offsets: &TopicPartitionList)

Post commit callback. This method will run after a group of offsets was committed to the offset store.

fn main_queue_min_poll_interval(&self) -> Timeout

Returns the minimum interval at which to poll the main queue, which services the logging, stats, and error callbacks.

impl<C> ProducerContext for BrokerRewritingClientContext<C>where C: ProducerContext,

type DeliveryOpaque = <C as ProducerContext>::DeliveryOpaque

A DeliveryOpaque is a user-defined structure that will be passed to the producer when producing a message, and returned to the delivery method once the message has been delivered, or failed to.

fn delivery( &self, delivery_result: &DeliveryResult<'_>, delivery_opaque: Self::DeliveryOpaque )

This method will be called once the message has been delivered (or failed to). The DeliveryOpaque will be the one provided by the user when calling send.