Package com.linkedin.davinci.client

Class DelegatingAvroGenericDaVinciClient<K,V>

java.lang.Object

com.linkedin.davinci.client.DelegatingAvroGenericDaVinciClient<K,V>

All Implemented Interfaces:

DaVinciClient<K,V>, AvroGenericReadComputeStoreClient<K,V>, AvroGenericStoreClient<K,V>, java.io.Closeable, java.lang.AutoCloseable

Direct Known Subclasses:

StatsAvroGenericDaVinciClient

public class DelegatingAvroGenericDaVinciClient<K,V>
extends java.lang.Object
implements DaVinciClient<K,V>, AvroGenericReadComputeStoreClient<K,V>

Delegating layer for DaVinciClient.

Constructor Summary

Constructors

Constructor	Description
DelegatingAvroGenericDaVinciClient(AvroGenericDaVinciClient<K,V> delegate)

Method Summary

Modifier and Type	Method	Description
java.util.concurrent.CompletableFuture<java.util.Map<K,V>>	batchGet(java.util.Set<K> keys)	Get the values associated with the given keys and return them in a map of keys to values.
void	close()	Release the internal resources.
void	compute(ComputeRequestWrapper computeRequestWrapper, java.util.Set<K> keys, org.apache.avro.Schema resultSchema, StreamingCallback<K,ComputeGenericRecord> callback, long preRequestTimeInNS)
ComputeRequestBuilder<K>	compute(java.util.Optional<ClientStats> stats, java.util.Optional<ClientStats> streamingStats, AvroGenericReadComputeStoreClient computeStoreClient, long preRequestTimeInNS)
void	computeWithKeyPrefixFilter(byte[] prefixBytes, ComputeRequestWrapper computeRequestWrapper, StreamingCallback<org.apache.avro.generic.GenericRecord,org.apache.avro.generic.GenericRecord> callback)
java.util.concurrent.CompletableFuture<V>	get(K key)	Lookup the value by given key, and get(key).get() will return null if it doesn't exist.
java.util.concurrent.CompletableFuture<V>	get(K key, V reusedValue)	Similar to AvroGenericStoreClient.get(Object) except that it allows passing in a {@param reusedValue} instance, to minimize GC.
org.apache.avro.Schema	getKeySchema()	Get key schema.
org.apache.avro.Schema	getLatestValueSchema()	Get the latest value schema known in current store client.
int	getPartitionCount()	Get partition count of a store.
java.lang.String	getStoreName()
boolean	isProjectionFieldValidationEnabled()
void	start()
void	streamingBatchGet(java.util.Set<K> keys, StreamingCallback<K,V> callback)	Streaming interface for AvroGenericStoreClient.batchGet(Set).
java.util.concurrent.CompletableFuture<java.lang.Void>	subscribe(java.util.Set<java.lang.Integer> partitions)	Ingest specific partition/partitions locally.
java.util.concurrent.CompletableFuture<java.lang.Void>	subscribeAll()	Ingest the entire data (i.e.
void	unsubscribe(java.util.Set<java.lang.Integer> partitions)	Stop ingesting a partition locally, and drop its associated local states/data.
void	unsubscribeAll()	Stop ingesting all subscribed partition locally, and drop their associated local states/data.

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.client.store.AvroGenericReadComputeStoreClient

compute, compute, compute

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

streamingBatchGet

Constructor Detail

DelegatingAvroGenericDaVinciClient

public DelegatingAvroGenericDaVinciClient(AvroGenericDaVinciClient<K,V> delegate)

Method Detail

subscribeAll

public java.util.concurrent.CompletableFuture<java.lang.Void> subscribeAll()

Description copied from interface: DaVinciClient

Ingest the entire data (i.e. all partitions) locally.

Specified by:

subscribeAll in interface DaVinciClient<K,V>

Returns:

a future which completes when the data is ready to serve

subscribe

public java.util.concurrent.CompletableFuture<java.lang.Void> subscribe(java.util.Set<java.lang.Integer> partitions)

Description copied from interface: DaVinciClient

Ingest specific partition/partitions locally.

Specified by:

subscribe in interface DaVinciClient<K,V>

Parameters:

partitions - the set of partition IDs to subscribe to

Returns:

a future which completes when the partitions are ready to serve

unsubscribeAll

public void unsubscribeAll()

Description copied from interface: DaVinciClient

Stop ingesting all subscribed partition locally, and drop their associated local states/data. If applications intend to keep the states/data for future use, no need to invoke this function before calling AvroGenericStoreClient.close().

Specified by:

unsubscribeAll in interface DaVinciClient<K,V>

unsubscribe

public void unsubscribe(java.util.Set<java.lang.Integer> partitions)

Description copied from interface: DaVinciClient

Stop ingesting a partition locally, and drop its associated local states/data. If applications intend to keep the states/data for future use, no need to invoke this function before calling AvroGenericStoreClient.close().

Specified by:

unsubscribe in interface DaVinciClient<K,V>

Parameters:

partitions - the set of partition IDs to unsubscribe from

getPartitionCount

public int getPartitionCount()

Description copied from interface: DaVinciClient

Get partition count of a store.

Specified by:

getPartitionCount in interface DaVinciClient<K,V>

Returns:

partition count

get

public java.util.concurrent.CompletableFuture<V> get(K key)
                                              throws VeniceClientException

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>

Returns:

Throws:

VeniceClientException

get

public java.util.concurrent.CompletableFuture<V> get(K key,
                                                     V reusedValue)
                                              throws VeniceClientException

Description copied from interface: AvroGenericStoreClient

Similar to AvroGenericStoreClient.get(Object) except that it allows passing in a {@param reusedValue} instance, to minimize GC.

Specified by:

get in interface AvroGenericStoreClient<K,V>

Throws:

VeniceClientException

batchGet

public java.util.concurrent.CompletableFuture<java.util.Map<K,V>> batchGet(java.util.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>

Returns:

Throws:

VeniceClientException

streamingBatchGet

public void streamingBatchGet(java.util.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>

Throws:

VeniceClientException

start

public void start()
           throws VeniceClientException

Specified by:

start in interface AvroGenericStoreClient<K,V>

Throws:

VeniceClientException

close

public void close()

Description copied from interface: AvroGenericStoreClient

Release the internal resources.

Specified by:

close in interface java.lang.AutoCloseable

Specified by:

close in interface AvroGenericStoreClient<K,V>

Specified by:

close in interface java.io.Closeable

getStoreName

public java.lang.String getStoreName()

Specified by:

getStoreName in interface AvroGenericStoreClient<K,V>

getKeySchema

public org.apache.avro.Schema getKeySchema()

Description copied from interface: AvroGenericStoreClient

Get key schema.

Specified by:

getKeySchema in interface AvroGenericStoreClient<K,V>

getLatestValueSchema

public org.apache.avro.Schema getLatestValueSchema()

Description copied from interface: AvroGenericStoreClient

Get the latest value schema known in current store client. This function doesn't guarantee it will return the latest schema if you add a new value schema when current store client is running.

Specified by:

getLatestValueSchema in interface AvroGenericStoreClient<K,V>

isProjectionFieldValidationEnabled

public boolean isProjectionFieldValidationEnabled()

Specified by:

isProjectionFieldValidationEnabled in interface AvroGenericReadComputeStoreClient<K,V>

compute

public ComputeRequestBuilder<K> compute(java.util.Optional<ClientStats> stats,
                                        java.util.Optional<ClientStats> streamingStats,
                                        AvroGenericReadComputeStoreClient computeStoreClient,
                                        long preRequestTimeInNS)
                                 throws VeniceClientException

Specified by:

compute in interface AvroGenericReadComputeStoreClient<K,V>

Throws:

VeniceClientException

compute

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

Specified by:

compute in interface AvroGenericReadComputeStoreClient<K,V>

Throws:

VeniceClientException

computeWithKeyPrefixFilter

public void computeWithKeyPrefixFilter(byte[] prefixBytes,
                                       ComputeRequestWrapper computeRequestWrapper,
                                       StreamingCallback<org.apache.avro.generic.GenericRecord,org.apache.avro.generic.GenericRecord> callback)

Specified by:

computeWithKeyPrefixFilter in interface AvroGenericReadComputeStoreClient<K,V>