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>, Closeable, AutoCloseable

Direct Known Subclasses:

StatsAvroGenericDaVinciClient

public class DelegatingAvroGenericDaVinciClient<K,V>
extends 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
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.
void	close()	Release the internal resources.
void	compute(ComputeRequestWrapper computeRequestWrapper, Set<K> keys, org.apache.avro.Schema resultSchema, StreamingCallback<K,ComputeGenericRecord> callback, long preRequestTimeInNS)
ComputeRequestBuilder<K>	compute(Optional<ClientStats> stats, 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)
CompletableFuture<V>	get(K key)	Lookup the value by given key, and get(key).get() will return null if it doesn't exist.
CompletableFuture<V>	get(K key, V reusedValue)	Similar to AvroGenericStoreClient.get(Object) except that it allows passing in a instance, to minimize GC.
org.apache.avro.Schema	getKeySchema()	Get key schema.
org.apache.avro.Schema	getLatestValueSchema()	Deprecated.
int	getPartitionCount()	Get partition count of a store.
SchemaReader	getSchemaReader()
String	getStoreName()
boolean	isProjectionFieldValidationEnabled()
void	start()
void	streamingBatchGet(Set<K> keys, StreamingCallback<K,V> callback)	Streaming interface for AvroGenericStoreClient.batchGet(Set).
CompletableFuture<Void>	subscribe(Set<Integer> partitions)	Ingest specific partition/partitions locally.
CompletableFuture<Void>	subscribeAll()	Ingest the entire data (i.e.
void	unsubscribe(Set<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, computeAggregation

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

streamingBatchGet

Constructor Details

DelegatingAvroGenericDaVinciClient

public DelegatingAvroGenericDaVinciClient(AvroGenericDaVinciClient<K,V> delegate)

Method Details

subscribeAll

public CompletableFuture<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 CompletableFuture<Void> subscribe(Set<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(Set<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 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 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 instance, to minimize GC.

Specified by:

get in interface AvroGenericStoreClient<K,V>

Throws:

VeniceClientException

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>

Returns:

Throws:

VeniceClientException

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>

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 AutoCloseable

Specified by:

close in interface AvroGenericStoreClient<K,V>

Specified by:

close in interface Closeable

getStoreName

public 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

@Deprecated
public org.apache.avro.Schema getLatestValueSchema()

Deprecated.

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>

getSchemaReader

public SchemaReader getSchemaReader()

Specified by:

getSchemaReader in interface AvroGenericReadComputeStoreClient<K,V>

isProjectionFieldValidationEnabled

public boolean isProjectionFieldValidationEnabled()

Specified by:

isProjectionFieldValidationEnabled in interface AvroGenericReadComputeStoreClient<K,V>

compute

public ComputeRequestBuilder<K> compute(Optional<ClientStats> stats,
 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,
 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>