Package com.linkedin.venice.pubsub.mock

Class InMemoryPubSubBroker

java.lang.Object

com.linkedin.venice.pubsub.mock.InMemoryPubSubBroker

public class InMemoryPubSubBroker
extends Object

Used in unit tests in order to avoid spinning a full Kafka broker with network stack and disk IO. Instead, this Kafka broker keeps messages in memory. It can be used via the following mock classes:

See Also:

• MockInMemoryConsumerAdapter
• MockInMemoryProducerAdapter

Constructor Summary

Constructors

Constructor	Description
InMemoryPubSubBroker(String brokerNamePrefix)

Method Summary

Modifier and Type	Method	Description
Optional<InMemoryPubSubMessage>	consume(String topicName, int partition, InMemoryPubSubPosition position)
void	createTopic(String topicName, int partitionCount)
Long	endOffsets(String topicName, int partition)
InMemoryPubSubPosition	endPosition(String topicName, int partition)
int	getPartitionCount(String topicName)
int	getPort()
String	getPubSubBrokerAddress()
InMemoryPubSubPosition	produce(String topicName, int partition, InMemoryPubSubMessage message)

Methods inherited from class java.lang.Object

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

Constructor Details

InMemoryPubSubBroker

public InMemoryPubSubBroker(String brokerNamePrefix)

Method Details

createTopic

public void createTopic(String topicName,
 int partitionCount)

produce

public InMemoryPubSubPosition produce(String topicName,
 int partition,
 InMemoryPubSubMessage message)

Parameters:

topicName - The name of the topic in which to produce.

partition - The partition in which to produce a message.

message - The InMemoryPubSubMessage to produce into the partition.

Returns:

the offset of the produced message

Throws:

IllegalArgumentException - if the topic or partition does not exist.

consume

public Optional<InMemoryPubSubMessage> consume(String topicName,
 int partition,
 InMemoryPubSubPosition position)
                                        throws IllegalArgumentException

Parameters:

topicName - The name of the topic from which to consume.

partition - The partition from which to produce a message.

Returns:

Some InMemoryPubSubMessage instance, or the Optional.empty() instance if that partition is drained.

Throws:

IllegalArgumentException - if the topic or partition does not exist.

getPartitionCount

public int getPartitionCount(String topicName)

getPubSubBrokerAddress

public String getPubSubBrokerAddress()

Returns:

a synthetic broker server url.

endOffsets

public Long endOffsets(String topicName,
 int partition)

endPosition

public InMemoryPubSubPosition endPosition(String topicName,
 int partition)

getPort

public int getPort()