Package com.linkedin.venice.client.store

Class StatTrackingStoreClient<K,V>

java.lang.Object

com.linkedin.venice.client.store.InternalAvroStoreClient<K,V>

com.linkedin.venice.client.store.DelegatingStoreClient<K,V>

com.linkedin.venice.client.store.StatTrackingStoreClient<K,V>

Type Parameters:

V -

All Implemented Interfaces:

AvroGenericReadComputeStoreClient<K,V>, AvroGenericStoreClient<K,V>, Closeable, AutoCloseable

Direct Known Subclasses:

SpecificStatTrackingStoreClient

public class StatTrackingStoreClient<K,V>
extends DelegatingStoreClient<K,V>

This class is used to handle all the metric related logic.

Constructor Summary

Constructors

Constructor	Description
StatTrackingStoreClient(InternalAvroStoreClient<K,V> innerStoreClient, ClientConfig clientConfig)

Method Summary

Modifier and Type	Method	Description
CompletableFuture<Map<K,V>>	batchGet(Set<K> keys)	Get the values associated with the given keys and return them in a map of keys to values.
ComputeRequestBuilder<K>	compute()	This API allows performing transformations (projection, vector arithmetic and aggregations like count) on the values associated with the given set of keys.
void	compute(ComputeRequestWrapper computeRequestWrapper, Set<K> keys, org.apache.avro.Schema resultSchema, StreamingCallback<K,ComputeGenericRecord> callback, long preRequestTimeInNS)
CompletableFuture<V>	get(K key)	Lookup the value by given key, and get(key).get() will return null if it doesn't exist.
CompletableFuture<byte[]>	getRaw(String requestPath)
static <T> BiFunction<? super T,Throwable,? extends T>	getStatCallback(ClientStats clientStats, long startTimeInNS)
static void	handleStoreExceptionInternally(Throwable throwable)
void	recordRetryCount(RequestType requestType)
CompletableFuture<VeniceResponseMap<K,V>>	streamingBatchGet(Set<K> keys)	Get the values associated with the given keys and return them in a map of keys to values.
void	streamingBatchGet(Set<K> keys, StreamingCallback<K,V> callback)	Streaming interface for AvroGenericStoreClient.batchGet(Set).
String	toString()

Methods inherited from class com.linkedin.venice.client.store.DelegatingStoreClient

close, compute, computeWithKeyPrefixFilter, get, getDeserializationExecutor, getInnerStoreClient, getKeySchema, getLatestValueSchema, getRaw, getSchemaReader, getStoreName, isProjectionFieldValidationEnabled, start, startWithExceptionThrownWhenFail

Methods inherited from class com.linkedin.venice.client.store.InternalAvroStoreClient

getStreamingCallback

Methods inherited from class java.lang.Object

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

Methods inherited from interface com.linkedin.venice.client.store.AvroGenericReadComputeStoreClient

compute, compute, computeAggregation

Methods inherited from interface com.linkedin.venice.client.store.AvroGenericStoreClient

get

Constructor Details

StatTrackingStoreClient

public StatTrackingStoreClient(InternalAvroStoreClient<K,V> innerStoreClient,
 ClientConfig clientConfig)

Method Details

get

public CompletableFuture<V> get(K key)

Description copied from interface: AvroGenericStoreClient

Lookup the value by given key, and get(key).get() will return null if it doesn't exist. For now, if any backend error/exception happens, get(Object key).get() will throw ExecutionException, which is a wrapper of the real exception.

Specified by:

get in interface AvroGenericStoreClient<K,V>

Overrides:

get in class InternalAvroStoreClient<K,V>

Returns:

getRaw

public CompletableFuture<byte[]> getRaw(String requestPath)

Overrides:

getRaw in class InternalAvroStoreClient<K,V>

batchGet

public CompletableFuture<Map<K,V>> batchGet(Set<K> keys)
                                     throws VeniceClientException

Description copied from interface: AvroGenericStoreClient

Get the values associated with the given keys and return them in a map of keys to values. Note that the returned map will only contain entries for the keys which have a value associated with them.

Specified by:

batchGet in interface AvroGenericStoreClient<K,V>

Overrides:

batchGet in class DelegatingStoreClient<K,V>

Returns:

Throws:

VeniceClientException

streamingBatchGet

public CompletableFuture<VeniceResponseMap<K,V>> streamingBatchGet(Set<K> keys)
                                                            throws VeniceClientException

Description copied from interface: AvroGenericStoreClient

Get the values associated with the given keys and return them in a map of keys to values. When time-out happens for the following invocation: streamingBatchGet(keys).get(waitingTime, unit); This function will return the available response instead of throwing a TimeoutException. It means this function could return either full response or partial response.

Returns:

Throws:

VeniceClientException

recordRetryCount

public void recordRetryCount(RequestType requestType)

streamingBatchGet

public void streamingBatchGet(Set<K> keys,
 StreamingCallback<K,V> callback)
                       throws VeniceClientException

Description copied from interface: AvroGenericStoreClient

Streaming interface for AvroGenericStoreClient.batchGet(Set). You can find more info in StreamingCallback.

Specified by:

streamingBatchGet in interface AvroGenericStoreClient<K,V>

Overrides:

streamingBatchGet in class DelegatingStoreClient<K,V>

Throws:

VeniceClientException

compute

public void compute(ComputeRequestWrapper computeRequestWrapper,
 Set<K> keys,
 org.apache.avro.Schema resultSchema,
 StreamingCallback<K,ComputeGenericRecord> callback,
 long preRequestTimeInNS)
             throws VeniceClientException

Specified by:

compute in interface AvroGenericReadComputeStoreClient<K,V>

Overrides:

compute in class DelegatingStoreClient<K,V>

Throws:

VeniceClientException

compute

public ComputeRequestBuilder<K> compute()
                                 throws VeniceClientException

Description copied from interface: AvroGenericStoreClient

This API allows performing transformations (projection, vector arithmetic and aggregations like count) on the values associated with the given set of keys. Check out ComputeRequestBuilder for details on the available operations.

Throws:

VeniceClientException

See Also:

• ComputeRequestBuilder

getStatCallback

public static <T>
BiFunction<? super T,Throwable,? extends T> getStatCallback(ClientStats clientStats,
 long startTimeInNS)

handleStoreExceptionInternally

public static void handleStoreExceptionInternally(Throwable throwable)

toString

public String toString()

Overrides:

toString in class Object