Package com.linkedin.venice.unit.kafka.producer

Class MockInMemoryProducerAdapter

java.lang.Object

com.linkedin.venice.unit.kafka.producer.MockInMemoryProducerAdapter

All Implemented Interfaces:

PubSubProducerAdapter

public class MockInMemoryProducerAdapter
extends java.lang.Object
implements PubSubProducerAdapter

A PubSubProducerAdapter implementation which interacts with the InMemoryKafkaBroker in order to make unit tests more lightweight.

Constructor Summary

Constructors

Constructor	Description
MockInMemoryProducerAdapter(InMemoryKafkaBroker broker)

Method Summary

Modifier and Type	Method	Description
void	close(int closeTimeOutMs, boolean flush)
void	flush()
java.lang.String	getBrokerAddress()
it.unimi.dsi.fastutil.objects.Object2DoubleMap<java.lang.String>	getMeasurableProducerMetrics()
int	getNumberOfPartitions(java.lang.String topic)	The support for the following two getNumberOfPartitions APIs will be removed.
java.util.concurrent.CompletableFuture<PubSubProduceResult>	sendMessage(java.lang.String topic, java.lang.Integer partition, KafkaKey key, KafkaMessageEnvelope value, PubSubMessageHeaders headers, PubSubProducerCallback callback)	Sends a message to a PubSub topic asynchronously and returns a Future representing the result of the produce operation.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface com.linkedin.venice.pubsub.api.PubSubProducerAdapter

close, close

Constructor Detail

MockInMemoryProducerAdapter

public MockInMemoryProducerAdapter(InMemoryKafkaBroker broker)

Method Detail

getNumberOfPartitions

public int getNumberOfPartitions(java.lang.String topic)

Description copied from interface: PubSubProducerAdapter

The support for the following two getNumberOfPartitions APIs will be removed.

Specified by:

getNumberOfPartitions in interface PubSubProducerAdapter

sendMessage

public java.util.concurrent.CompletableFuture<PubSubProduceResult> sendMessage(java.lang.String topic,
                                                                               java.lang.Integer partition,
                                                                               KafkaKey key,
                                                                               KafkaMessageEnvelope value,
                                                                               PubSubMessageHeaders headers,
                                                                               PubSubProducerCallback callback)

Description copied from interface: PubSubProducerAdapter

Sends a message to a PubSub topic asynchronously and returns a Future representing the result of the produce operation.

Specified by:

sendMessage in interface PubSubProducerAdapter

Parameters:

topic - The name of the Kafka topic to which the message will be sent.

partition - The partition to which the message should be sent.

key - The key associated with the message, used for partitioning and message retrieval.

value - The message payload to be sent to the PubSubTopic topic.

headers - Additional headers to be included with the message.

callback - An optional callback to handle the result of the produce operation.

Returns:

A Future representing the result of the produce operation.

flush

public void flush()

Specified by:

flush in interface PubSubProducerAdapter

close

public void close(int closeTimeOutMs,
                  boolean flush)

Specified by:

close in interface PubSubProducerAdapter

getMeasurableProducerMetrics

public it.unimi.dsi.fastutil.objects.Object2DoubleMap<java.lang.String> getMeasurableProducerMetrics()

Specified by:

getMeasurableProducerMetrics in interface PubSubProducerAdapter

getBrokerAddress

public java.lang.String getBrokerAddress()

Specified by:

getBrokerAddress in interface PubSubProducerAdapter