com.google.cloud.bigtable.grpc.async

Class AsyncExecutor

java.lang.Object

com.google.cloud.bigtable.grpc.async.AsyncExecutor

public class AsyncExecutor
extends Object

This class provides management of asynchronous Bigtable RPCs. It ensures that there aren't too many concurrent, in flight asynchronous RPCs and also makes sure that the memory used by the requests doesn't exceed a threshold.

Version:

$Id: $Id

Author:

sduskis

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static interface	AsyncExecutor.AsyncCall<RequestT,ResponseT>

Field Summary

Fields

Modifier and Type	Field and Description
protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.CheckAndMutateRowRequest,com.google.bigtable.v2.CheckAndMutateRowResponse>	CHECK_AND_MUTATE_ASYNC Calls BigtableDataClient.checkAndMutateRowAsync(CheckAndMutateRowRequest).
protected static Logger	LOG Constant LOG
protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.MutateRowRequest,com.google.bigtable.v2.MutateRowResponse>	MUTATE_ROW_ASYNC Calls BigtableDataClient.mutateRowAsync(MutateRowRequest).
protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.MutateRowsRequest,List<com.google.bigtable.v2.MutateRowsResponse>>	MUTATE_ROWS_ASYNC Calls BigtableDataClient.mutateRowsAsync(MutateRowsRequest).
protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.ReadRowsRequest,List<FlatRow>>	READ_FLAT_ROWS_ASYNC Calls BigtableDataClient.readRowsAsync(ReadRowsRequest).
protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.ReadModifyWriteRowRequest,com.google.bigtable.v2.ReadModifyWriteRowResponse>	READ_MODIFY_WRITE_ASYNC Calls BigtableDataClient.readModifyWriteRowAsync(ReadModifyWriteRowRequest).
protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.ReadRowsRequest,List<com.google.bigtable.v2.Row>>	READ_ROWS_ASYNC Calls BigtableDataClient.readRowsAsync(ReadRowsRequest).

Constructor Summary

Constructors

Constructor and Description
AsyncExecutor(BigtableDataClient client) Constructor for AsyncExecutor.

Method Summary

Modifier and Type	Method and Description
com.google.common.util.concurrent.ListenableFuture<com.google.bigtable.v2.CheckAndMutateRowResponse>	checkAndMutateRowAsync(com.google.bigtable.v2.CheckAndMutateRowRequest request) Performs a BigtableDataClient.checkAndMutateRowAsync(CheckAndMutateRowRequest) on the CheckAndMutateRowRequest.
void	flush() Waits until all operations managed by the OperationAccountant complete.
BigtableDataClient	getClient() Getter for the field client.
OperationAccountant	getOperationAccountant() Getter for the field operationsAccountant.
boolean	hasInflightRequests() hasInflightRequests.
com.google.common.util.concurrent.ListenableFuture<com.google.bigtable.v2.MutateRowResponse>	mutateRowAsync(com.google.bigtable.v2.MutateRowRequest request) Performs a BigtableDataClient.mutateRowAsync(MutateRowRequest) on the MutateRowRequest.
com.google.common.util.concurrent.ListenableFuture<List<com.google.bigtable.v2.MutateRowsResponse>>	mutateRowsAsync(com.google.bigtable.v2.MutateRowsRequest request) Performs a BigtableDataClient.mutateRowsAsync(MutateRowsRequest) on the MutateRowsRequest.
com.google.common.util.concurrent.ListenableFuture<List<FlatRow>>	readFlatRowsAsync(com.google.bigtable.v2.ReadRowsRequest request) Performs a BigtableDataClient.readRowsAsync(ReadRowsRequest) on the ReadRowsRequest.
com.google.common.util.concurrent.ListenableFuture<com.google.bigtable.v2.ReadModifyWriteRowResponse>	readModifyWriteRowAsync(com.google.bigtable.v2.ReadModifyWriteRowRequest request) Performs a BigtableDataClient.readModifyWriteRow(ReadModifyWriteRowRequest) on the ReadModifyWriteRowRequest.
com.google.common.util.concurrent.ListenableFuture<List<com.google.bigtable.v2.Row>>	readRowsAsync(com.google.bigtable.v2.ReadRowsRequest request) Performs a BigtableDataClient.readRowsAsync(ReadRowsRequest) on the ReadRowsRequest.

Methods inherited from class java.lang.Object

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

Field Detail

LOG

protected static final Logger LOG

Constant LOG

MUTATE_ROW_ASYNC

protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.MutateRowRequest,com.google.bigtable.v2.MutateRowResponse> MUTATE_ROW_ASYNC

Calls BigtableDataClient.mutateRowAsync(MutateRowRequest).

MUTATE_ROWS_ASYNC

protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.MutateRowsRequest,List<com.google.bigtable.v2.MutateRowsResponse>> MUTATE_ROWS_ASYNC

Calls BigtableDataClient.mutateRowsAsync(MutateRowsRequest).

READ_MODIFY_WRITE_ASYNC

protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.ReadModifyWriteRowRequest,com.google.bigtable.v2.ReadModifyWriteRowResponse> READ_MODIFY_WRITE_ASYNC

Calls BigtableDataClient.readModifyWriteRowAsync(ReadModifyWriteRowRequest).

CHECK_AND_MUTATE_ASYNC

protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.CheckAndMutateRowRequest,com.google.bigtable.v2.CheckAndMutateRowResponse> CHECK_AND_MUTATE_ASYNC

Calls BigtableDataClient.checkAndMutateRowAsync(CheckAndMutateRowRequest).

READ_ROWS_ASYNC

protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.ReadRowsRequest,List<com.google.bigtable.v2.Row>> READ_ROWS_ASYNC

Calls BigtableDataClient.readRowsAsync(ReadRowsRequest).

READ_FLAT_ROWS_ASYNC

protected static AsyncExecutor.AsyncCall<com.google.bigtable.v2.ReadRowsRequest,List<FlatRow>> READ_FLAT_ROWS_ASYNC

Calls BigtableDataClient.readRowsAsync(ReadRowsRequest).

Constructor Detail

AsyncExecutor

public AsyncExecutor(BigtableDataClient client)

Constructor for AsyncExecutor.

Parameters:

client - a BigtableDataClient object for executing RPCs.

Method Detail

mutateRowAsync

public com.google.common.util.concurrent.ListenableFuture<com.google.bigtable.v2.MutateRowResponse> mutateRowAsync(com.google.bigtable.v2.MutateRowRequest request)
                                                                                                            throws InterruptedException

Performs a BigtableDataClient.mutateRowAsync(MutateRowRequest) on the MutateRowRequest. This method may block if OperationAccountant.registerOperation(ListenableFuture) blocks.

Parameters:

request - The MutateRowRequest to send.

Returns:

a ListenableFuture which can be listened to for completion events.

Throws:

InterruptedException - if any.

mutateRowsAsync

public com.google.common.util.concurrent.ListenableFuture<List<com.google.bigtable.v2.MutateRowsResponse>> mutateRowsAsync(com.google.bigtable.v2.MutateRowsRequest request)
                                                                                                                    throws InterruptedException

Performs a BigtableDataClient.mutateRowsAsync(MutateRowsRequest) on the MutateRowsRequest. This method may block if OperationAccountant.registerOperation(ListenableFuture) blocks.

Parameters:

request - The MutateRowRequest to send.

Returns:

a ListenableFuture which can be listened to for completion events.

Throws:

InterruptedException - if any.

checkAndMutateRowAsync

public com.google.common.util.concurrent.ListenableFuture<com.google.bigtable.v2.CheckAndMutateRowResponse> checkAndMutateRowAsync(com.google.bigtable.v2.CheckAndMutateRowRequest request)
                                                                                                                            throws InterruptedException

Performs a BigtableDataClient.checkAndMutateRowAsync(CheckAndMutateRowRequest) on the CheckAndMutateRowRequest. This method may block if OperationAccountant.registerOperation(ListenableFuture) blocks.

Parameters:

request - The CheckAndMutateRowRequest to send.

Returns:

a ListenableFuture which can be listened to for completion events.

Throws:

InterruptedException - if any.

readModifyWriteRowAsync

public com.google.common.util.concurrent.ListenableFuture<com.google.bigtable.v2.ReadModifyWriteRowResponse> readModifyWriteRowAsync(com.google.bigtable.v2.ReadModifyWriteRowRequest request)
                                                                                                                              throws InterruptedException

Performs a BigtableDataClient.readModifyWriteRow(ReadModifyWriteRowRequest) on the ReadModifyWriteRowRequest. This method may block if OperationAccountant.registerOperation(ListenableFuture) blocks.

Parameters:

request - The ReadModifyWriteRowRequest to send.

Returns:

a ListenableFuture which can be listened to for completion events.

Throws:

InterruptedException - if any.

readRowsAsync

public com.google.common.util.concurrent.ListenableFuture<List<com.google.bigtable.v2.Row>> readRowsAsync(com.google.bigtable.v2.ReadRowsRequest request)
                                                                                                   throws InterruptedException

Performs a BigtableDataClient.readRowsAsync(ReadRowsRequest) on the ReadRowsRequest. This method may block if OperationAccountant.registerOperation(ListenableFuture) blocks.

Parameters:

request - The ReadRowsRequest to send.

Returns:

a ListenableFuture which can be listened to for completion events.

Throws:

InterruptedException - if any.

readFlatRowsAsync

public com.google.common.util.concurrent.ListenableFuture<List<FlatRow>> readFlatRowsAsync(com.google.bigtable.v2.ReadRowsRequest request)
                                                                                    throws InterruptedException

Performs a BigtableDataClient.readRowsAsync(ReadRowsRequest) on the ReadRowsRequest. This method may block if OperationAccountant.registerOperation(ListenableFuture) blocks.

Parameters:

request - The ReadRowsRequest to send.

Returns:

a ListenableFuture which can be listened to for completion events.

Throws:

InterruptedException - if any.

flush

public void flush()
           throws IOException

Waits until all operations managed by the OperationAccountant complete. See OperationAccountant.awaitCompletion() for more information.

Throws:

IOException - if something goes wrong.

hasInflightRequests

public boolean hasInflightRequests()

hasInflightRequests.

Returns:

a boolean.

getClient

public BigtableDataClient getClient()

Getter for the field client.

Returns:

a BigtableDataClient object.

getOperationAccountant

public OperationAccountant getOperationAccountant()

Getter for the field operationsAccountant.

Returns:

a OperationAccountant object.