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

Class MockInMemoryTransformingProducerAdapter

java.lang.Object

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

All Implemented Interfaces:

PubSubProducerAdapter

public class MockInMemoryTransformingProducerAdapter
extends Object
implements PubSubProducerAdapter

This PubSubProducerAdapter implementation allows tests to perform arbitrary transformations on the messages that are about to be written to PubSub. This can be used in unit tests to inject corrupt data.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MockInMemoryTransformingProducerAdapter.SendMessageParameters
static interface	MockInMemoryTransformingProducerAdapter.SendMessageParametersTransformer

Constructor Summary

Constructors

Constructor	Description
MockInMemoryTransformingProducerAdapter(PubSubProducerAdapter baseProducer, MockInMemoryTransformingProducerAdapter.SendMessageParametersTransformer transformer)

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.
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

MockInMemoryTransformingProducerAdapter

public MockInMemoryTransformingProducerAdapter(PubSubProducerAdapter baseProducer,
 MockInMemoryTransformingProducerAdapter.SendMessageParametersTransformer transformer)

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