Package com.linkedin.venice.fastclient

Class MultiKeyRequestContext<K,V>

java.lang.Object

com.linkedin.venice.fastclient.RequestContext

com.linkedin.venice.fastclient.MultiKeyRequestContext<K,V>

Type Parameters:

K - Key type

V - Value type

Direct Known Subclasses:

BatchGetRequestContext, ComputeRequestContext

public abstract class MultiKeyRequestContext<K,V>
extends RequestContext

Keep track of the progress of a multi-key requests - batch get and compute. This includes tracking all the scatter requests and utilities to gather responses.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MultiKeyRequestContext.KeyInfo<K>	represents a key requested on a route and tracks info related to it but specific to the route

Constructor Summary

Constructors

Constructor	Description
MultiKeyRequestContext(int numKeysInRequest, boolean isPartialSuccessAllowed)

Method Summary

Modifier and Type	Method	Description
void	addKey(String route, K key, byte[] serializedKey, int partitionId)
protected void	copyStateToRetryRequestContext(MultiKeyRequestContext<K,V> retryContext)
int	getFanoutSize()
Set<K>	getKeys()
Set<String>	getRoutes()
Map<Integer,Set<String>>	getRoutesForPartitionMapping()
void	setFanoutSize(int fanoutSize)
void	setKeys(Set<K> keys)
void	setRoutesForPartitionMapping(Map<Integer,Set<String>> routesForPartition)

Methods inherited from class com.linkedin.venice.fastclient.RequestContext

addNonAvailableReplicaPartition, getCurrentVersion, getHelixGroupId, getInstanceHealthMonitor, getNonAvailableReplicaPartitions, getRequestId, getRequestType, getResultFuture, getRouteRequestMap, getServerClusterName, hasNonAvailablePartition, isRetryRequest, setCurrentVersion, setHelixGroupId, setInstanceHealthMonitor, setRequestId, setRequestSerializationTime, setResultFuture, setRetryRequest, setServerClusterName

Methods inherited from class java.lang.Object

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

Constructor Details

MultiKeyRequestContext

public MultiKeyRequestContext(int numKeysInRequest,
 boolean isPartialSuccessAllowed)

Method Details

setKeys

public void setKeys(Set<K> keys)

getKeys

public Set<K> getKeys()

addKey

public void addKey(String route,
 K key,
 byte[] serializedKey,
 int partitionId)

getRoutes

public Set<String> getRoutes()

copyStateToRetryRequestContext

protected void copyStateToRetryRequestContext(MultiKeyRequestContext<K,V> retryContext)

getRoutesForPartitionMapping

public Map<Integer,Set<String>> getRoutesForPartitionMapping()

setRoutesForPartitionMapping

public void setRoutesForPartitionMapping(Map<Integer,Set<String>> routesForPartition)

setFanoutSize

public void setFanoutSize(int fanoutSize)

getFanoutSize

public int getFanoutSize()