Package com.linkedin.venice.pubsub.mock.adapter.producer

Class MockInMemoryProducerAdapter

java.lang.Object

com.linkedin.venice.pubsub.mock.adapter.producer.MockInMemoryProducerAdapter

All Implemented Interfaces:

PubSubProducerAdapter

public class MockInMemoryProducerAdapter
extends Object
implements PubSubProducerAdapter

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

Constructor Summary

Constructors

Constructor	Description
MockInMemoryProducerAdapter(InMemoryPubSubBroker broker)

Method Summary

Modifier and Type	Method	Description
void	close(long closeTimeOutMs)
void	flush()
String	getBrokerAddress()
it.unimi.dsi.fastutil.objects.Object2DoubleMap<String>	getMeasurableProducerMetrics()
int	getNumberOfPartitions(String topic)	The support for the following two getNumberOfPartitions APIs will be removed.
static InMemoryPubSubPosition	getPosition(Future<PubSubProduceResult> produceResultFuture)
CompletableFuture<PubSubProduceResult>	sendMessage(String topic, 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

sendMessage

Constructor Details

MockInMemoryProducerAdapter

public MockInMemoryProducerAdapter(InMemoryPubSubBroker broker)

Method Details

getNumberOfPartitions

public int getNumberOfPartitions(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 CompletableFuture<PubSubProduceResult> sendMessage(String topic,
 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(long closeTimeOutMs)

Specified by:

close in interface PubSubProducerAdapter

getMeasurableProducerMetrics

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

Specified by:

getMeasurableProducerMetrics in interface PubSubProducerAdapter

getBrokerAddress

public String getBrokerAddress()

Specified by:

getBrokerAddress in interface PubSubProducerAdapter

getPosition

public static InMemoryPubSubPosition getPosition(Future<PubSubProduceResult> produceResultFuture)
                                          throws ExecutionException,
InterruptedException

Throws:

ExecutionException

InterruptedException