private RetryingMutateRowsOperation createMutateRowsOperation(MutateRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } CallOptions callOptions = getCallOptions(mutateRowsRpc.getMethodDescriptor(), request); Metadata metadata = createMetadata(request.getTableName()); return new RetryingMutateRowsOperation(retryOptions, request, mutateRowsRpc, callOptions, retryExecutorService, metadata, CLOCK); }
/** {@inheritDoc} */ @Override public List<MutateRowsResponse> mutateRows(MutateRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createMutateRowsOperation(request).getBlockingResult(); }
/** {@inheritDoc} */ @Override public ScanHandler readFlatRows(ReadRowsRequest request, StreamObserver<FlatRow> observer) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } RetryingReadRowsOperation operation = createReadRowsRetryListener(request, observer); // Start the operation. operation.getAsyncResult(); return operation; }
private <ReqT, RespT> RetryingUnaryOperation<ReqT, RespT> createUnaryListener( ReqT request, BigtableAsyncRpc<ReqT, RespT> rpc, String tableName) { CallOptions callOptions = getCallOptions(rpc.getMethodDescriptor(), request); Metadata metadata = createMetadata(tableName); return new RetryingUnaryOperation<>( retryOptions, request, rpc, callOptions, retryExecutorService, metadata, CLOCK); }
/** {@inheritDoc} */ @Override public ListenableFuture<List<Row>> readRowsAsync(ReadRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return Futures.transform( createStreamingListener(request, readRowsAsync, request.getTableName()).getAsyncResult(), ROW_LIST_TRANSFORMER, MoreExecutors.directExecutor()); }
/** {@inheritDoc} */ @Override public ListenableFuture<MutateRowResponse> mutateRowAsync(MutateRowRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createUnaryListener(request, mutateRowRpc, request.getTableName()).getAsyncResult(); }
if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); final ResultScanner<FlatRow> delegate = readFlatRows(request); return new ResultScanner<Row>() {
this.mutateRowRpc = asyncUtilities.createAsyncRpc( BigtableGrpc.getMutateRowMethod(), getMutationRetryableFunction(IS_RETRYABLE_MUTATION)); this.mutateRowsRpc = asyncUtilities.createAsyncRpc( BigtableGrpc.getMutateRowsMethod(), getMutationRetryableFunction(ARE_RETRYABLE_MUTATIONS)); this.checkAndMutateRpc = asyncUtilities.createAsyncRpc( BigtableGrpc.getCheckAndMutateRowMethod(), getMutationRetryableFunction(Predicates.<CheckAndMutateRowRequest> alwaysFalse())); this.readWriteModifyRpc = asyncUtilities.createAsyncRpc( BigtableGrpc.getReadModifyWriteRowMethod(),
new CallOptionsFactory.ConfiguredCallOptionsFactory(options.getCallOptionsConfig()); dataClient = new BigtableDataGrpcClient(dataChannel, sharedPools.getRetryExecutor(), options); dataClient.setCallOptionsFactory(callOptionsFactory); ClientInterceptors.intercept(dataChannel, new ThrottlingClientInterceptor(resourceLimiter)); throttlingDataClient = new BigtableDataGrpcClient(asyncDataChannel, sharedPools.getRetryExecutor(), options);
/** {@inheritDoc} */ @Override public MutateRowResponse mutateRow(MutateRowRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createUnaryListener(request, mutateRowRpc, request.getTableName()).getBlockingResult(); }
/** {@inheritDoc} */ @Override public ListenableFuture<List<FlatRow>> readFlatRowsAsync(ReadRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return Futures.transform( createStreamingListener(request, readRowsAsync, request.getTableName()).getAsyncResult(), FLAT_ROW_LIST_TRANSFORMER, MoreExecutors.directExecutor()); }
private <ReqT, RespT> RetryingStreamOperation<ReqT, RespT> createStreamingListener( ReqT request, BigtableAsyncRpc<ReqT, RespT> rpc, String tableName) { CallOptions callOptions = getCallOptions(rpc.getMethodDescriptor(), request); Metadata metadata = createMetadata(tableName); return new RetryingStreamOperation<>( retryOptions, request, rpc, callOptions, retryExecutorService, metadata, CLOCK); }
/** {@inheritDoc} */ @Override public CheckAndMutateRowResponse checkAndMutateRow(CheckAndMutateRowRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createUnaryListener(request, checkAndMutateRpc, request.getTableName()) .getBlockingResult(); }
/** {@inheritDoc} */ @Override public List<FlatRow> readFlatRowsList(ReadRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return FLAT_ROW_LIST_TRANSFORMER.apply( createStreamingListener(request, readRowsAsync, request.getTableName()).getBlockingResult()); }
private RetryingReadRowsOperation createReadRowsRetryListener(ReadRowsRequest request, StreamObserver<FlatRow> observer) { return new RetryingReadRowsOperation( observer, retryOptions, request, readRowsAsync, getCallOptions(readRowsAsync.getMethodDescriptor(), request), retryExecutorService, createMetadata(request.getTableName()), CLOCK); }
/** {@inheritDoc} */ @Override public ListenableFuture<List<MutateRowsResponse>> mutateRowsAsync(MutateRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createMutateRowsOperation(request).getAsyncResult(); }
/** {@inheritDoc} */ @Override public ResultScanner<FlatRow> readFlatRows(ReadRowsRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } // Delegate all resumable operations to the scanner. It will request a non-resumable scanner // during operation. final ResponseQueueReader reader = new ResponseQueueReader(); RetryingReadRowsOperation operation = createReadRowsRetryListener(request, reader); operation.setResultObserver(new StreamObserver<ReadRowsResponse>(){ @Override public void onNext(ReadRowsResponse value) { reader.addRequestResultMarker(); } @Override public void onError(Throwable t) {} @Override public void onCompleted() {} }); // Start the operation. operation.getAsyncResult(); return new ResumingStreamingResultScanner(reader); }
/** {@inheritDoc} */ @Override public ListenableFuture<ReadModifyWriteRowResponse> readModifyWriteRowAsync( ReadModifyWriteRowRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createUnaryListener(request, readWriteModifyRpc, request.getTableName()) .getAsyncResult(); }
/** {@inheritDoc} */ @Override public ListenableFuture<List<SampleRowKeysResponse>> sampleRowKeysAsync( SampleRowKeysRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createStreamingListener(request, sampleRowKeysAsync, request.getTableName()) .getAsyncResult(); }
/** {@inheritDoc} */ @Override public ListenableFuture<CheckAndMutateRowResponse> checkAndMutateRowAsync( CheckAndMutateRowRequest request) { if (shouldOverrideAppProfile(request.getAppProfileId())) { request = request.toBuilder().setAppProfileId(clientDefaultAppProfileId).build(); } return createUnaryListener(request, checkAndMutateRpc, request.getTableName()).getAsyncResult(); }