All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.hadoop.hbase.client.AsyncTableImpl Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.hadoop.hbase.client;
import static java.util.stream.Collectors.toList;
import com.google.protobuf.RpcChannel;
import io.opentelemetry.api.trace.Span;
import io.opentelemetry.context.Context;
import io.opentelemetry.context.Scope;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.CompareOperator;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.filter.Filter;
import org.apache.hadoop.hbase.io.TimeRange;
import org.apache.hadoop.hbase.util.FutureUtils;
import org.apache.yetus.audience.InterfaceAudience;
/**
* Just a wrapper of {@link RawAsyncTableImpl}. The difference is that users need to provide a
* thread pool when constructing this class, and the callback methods registered to the returned
* {@link CompletableFuture} will be executed in this thread pool. So usually it is safe for users
* to do anything they want in the callbacks without breaking the rpc framework.
*/
@InterfaceAudience.Private
class AsyncTableImpl implements AsyncTable {
private final RawAsyncTableImpl rawTable;
private final ExecutorService pool;
AsyncTableImpl(RawAsyncTableImpl rawTable, ExecutorService pool) {
this.rawTable = rawTable;
this.pool = pool;
}
@Override
public TableName getName() {
return rawTable.getName();
}
@Override
public Configuration getConfiguration() {
return rawTable.getConfiguration();
}
@Override
public CompletableFuture getDescriptor() {
return wrap(rawTable.getDescriptor());
}
@Override
public AsyncTableRegionLocator getRegionLocator() {
return rawTable.getRegionLocator();
}
@Override
public long getRpcTimeout(TimeUnit unit) {
return rawTable.getRpcTimeout(unit);
}
@Override
public long getReadRpcTimeout(TimeUnit unit) {
return rawTable.getReadRpcTimeout(unit);
}
@Override
public long getWriteRpcTimeout(TimeUnit unit) {
return rawTable.getWriteRpcTimeout(unit);
}
@Override
public long getOperationTimeout(TimeUnit unit) {
return rawTable.getOperationTimeout(unit);
}
@Override
public long getScanTimeout(TimeUnit unit) {
return rawTable.getScanTimeout(unit);
}
@Override
public Map getRequestAttributes() {
return rawTable.getRequestAttributes();
}
private CompletableFuture wrap(CompletableFuture future) {
return FutureUtils.wrapFuture(future, pool);
}
@Override
public CompletableFuture get(Get get) {
return wrap(rawTable.get(get));
}
@Override
public CompletableFuture put(Put put) {
return wrap(rawTable.put(put));
}
@Override
public CompletableFuture delete(Delete delete) {
return wrap(rawTable.delete(delete));
}
@Override
public CompletableFuture append(Append append) {
return wrap(rawTable.append(append));
}
@Override
public CompletableFuture increment(Increment increment) {
return wrap(rawTable.increment(increment));
}
@Override
public CheckAndMutateBuilder checkAndMutate(byte[] row, byte[] family) {
return new CheckAndMutateBuilder() {
private final CheckAndMutateBuilder builder = rawTable.checkAndMutate(row, family);
@Override
public CompletableFuture thenPut(Put put) {
return wrap(builder.thenPut(put));
}
@Override
public CompletableFuture thenMutate(RowMutations mutation) {
return wrap(builder.thenMutate(mutation));
}
@Override
public CompletableFuture thenDelete(Delete delete) {
return wrap(builder.thenDelete(delete));
}
@Override
public CheckAndMutateBuilder qualifier(byte[] qualifier) {
builder.qualifier(qualifier);
return this;
}
@Override
public CheckAndMutateBuilder timeRange(TimeRange timeRange) {
builder.timeRange(timeRange);
return this;
}
@Override
public CheckAndMutateBuilder ifNotExists() {
builder.ifNotExists();
return this;
}
@Override
public CheckAndMutateBuilder ifMatches(CompareOperator compareOp, byte[] value) {
builder.ifMatches(compareOp, value);
return this;
}
};
}
@Override
public CheckAndMutateWithFilterBuilder checkAndMutate(byte[] row, Filter filter) {
return new CheckAndMutateWithFilterBuilder() {
private final CheckAndMutateWithFilterBuilder builder = rawTable.checkAndMutate(row, filter);
@Override
public CheckAndMutateWithFilterBuilder timeRange(TimeRange timeRange) {
builder.timeRange(timeRange);
return this;
}
@Override
public CompletableFuture thenPut(Put put) {
return wrap(builder.thenPut(put));
}
@Override
public CompletableFuture thenDelete(Delete delete) {
return wrap(builder.thenDelete(delete));
}
@Override
public CompletableFuture thenMutate(RowMutations mutation) {
return wrap(builder.thenMutate(mutation));
}
};
}
@Override
public CompletableFuture checkAndMutate(CheckAndMutate checkAndMutate) {
return wrap(rawTable.checkAndMutate(checkAndMutate));
}
@Override
public List>
checkAndMutate(List checkAndMutates) {
return rawTable.checkAndMutate(checkAndMutates).stream().map(this::wrap).collect(toList());
}
@Override
public CompletableFuture mutateRow(RowMutations mutation) {
return wrap(rawTable.mutateRow(mutation));
}
@Override
public CompletableFuture> scanAll(Scan scan) {
return wrap(rawTable.scanAll(scan));
}
@Override
public ResultScanner getScanner(Scan scan) {
return rawTable.getScanner(scan);
}
private void scan0(Scan scan, ScanResultConsumer consumer) {
Span span = null;
try (AsyncTableResultScanner scanner = rawTable.getScanner(scan)) {
span = scanner.getSpan();
try (Scope ignored = span.makeCurrent()) {
if (scan.isScanMetricsEnabled()) {
consumer.onScanMetricsCreated(scanner.getScanMetrics());
}
for (Result result; (result = scanner.next()) != null;) {
if (!consumer.onNext(result)) {
break;
}
}
consumer.onComplete();
}
} catch (IOException e) {
try (Scope ignored = span.makeCurrent()) {
consumer.onError(e);
}
}
}
@Override
public void scan(Scan scan, ScanResultConsumer consumer) {
final Context context = Context.current();
pool.execute(context.wrap(() -> scan0(scan, consumer)));
}
@Override
public List> get(List gets) {
return rawTable.get(gets).stream().map(this::wrap).collect(toList());
}
@Override
public List> put(List puts) {
return rawTable.put(puts).stream().map(this::wrap).collect(toList());
}
@Override
public List> delete(List deletes) {
return rawTable.delete(deletes).stream().map(this::wrap).collect(toList());
}
@Override
public List> batch(List actions) {
return rawTable. batch(actions).stream().map(this::wrap).collect(toList());
}
@Override
public CompletableFuture coprocessorService(Function stubMaker,
ServiceCaller callable, byte[] row) {
return wrap(rawTable.coprocessorService(stubMaker, callable, row));
}
@Override
public CoprocessorServiceBuilder coprocessorService(
Function stubMaker, ServiceCaller callable,
CoprocessorCallback callback) {
final Context context = Context.current();
CoprocessorCallback wrappedCallback = new CoprocessorCallback() {
@Override
public void onRegionComplete(RegionInfo region, R resp) {
pool.execute(context.wrap(() -> callback.onRegionComplete(region, resp)));
}
@Override
public void onRegionError(RegionInfo region, Throwable error) {
pool.execute(context.wrap(() -> callback.onRegionError(region, error)));
}
@Override
public void onComplete() {
pool.execute(context.wrap(callback::onComplete));
}
@Override
public void onError(Throwable error) {
pool.execute(context.wrap(() -> callback.onError(error)));
}
};
CoprocessorServiceBuilder builder =
rawTable.coprocessorService(stubMaker, callable, wrappedCallback);
return new CoprocessorServiceBuilder() {
@Override
public CoprocessorServiceBuilder fromRow(byte[] startKey, boolean inclusive) {
builder.fromRow(startKey, inclusive);
return this;
}
@Override
public CoprocessorServiceBuilder toRow(byte[] endKey, boolean inclusive) {
builder.toRow(endKey, inclusive);
return this;
}
@Override
public void execute() {
builder.execute();
}
};
}
}
