Examples with IpcClientSpanBuilder org.apache.hadoop.hbase.client.trace.IpcClientSpanBuilder used on opensource projects

Example 1 with IpcClientSpanBuilder

use of org.apache.hadoop.hbase.client.trace.IpcClientSpanBuilder in project hbase by apache.

the class AbstractRpcClient method callMethod.

private Call callMethod(final Descriptors.MethodDescriptor md, final HBaseRpcController hrc, final Message param, Message returnType, final User ticket, final Address addr, final RpcCallback<Message> callback) {
    Span span = new IpcClientSpanBuilder().setMethodDescriptor(md).setRemoteAddress(addr).build();
    try (Scope scope = span.makeCurrent()) {
        final MetricsConnection.CallStats cs = MetricsConnection.newCallStats();
        cs.setStartTime(EnvironmentEdgeManager.currentTime());
        if (param instanceof ClientProtos.MultiRequest) {
            ClientProtos.MultiRequest req = (ClientProtos.MultiRequest) param;
            int numActions = 0;
            for (ClientProtos.RegionAction regionAction : req.getRegionActionList()) {
                numActions += regionAction.getActionCount();
            }
            cs.setNumActionsPerServer(numActions);
        }
        final AtomicInteger counter = concurrentCounterCache.getUnchecked(addr);
        Call call = new Call(nextCallId(), md, param, hrc.cellScanner(), returnType, hrc.getCallTimeout(), hrc.getPriority(), new RpcCallback<Call>() {

            @Override
            public void run(Call call) {
                try (Scope scope = call.span.makeCurrent()) {
                    counter.decrementAndGet();
                    onCallFinished(call, hrc, addr, callback);
                } finally {
                    if (hrc.failed()) {
                        TraceUtil.setError(span, hrc.getFailed());
                    } else {
                        span.setStatus(StatusCode.OK);
                    }
                    span.end();
                }
            }
        }, cs);
        ConnectionId remoteId = new ConnectionId(ticket, md.getService().getName(), addr);
        int count = counter.incrementAndGet();
        try {
            if (count > maxConcurrentCallsPerServer) {
                throw new ServerTooBusyException(addr, count);
            }
            cs.setConcurrentCallsPerServer(count);
            T connection = getConnection(remoteId);
            connection.sendRequest(call, hrc);
        } catch (Exception e) {
            call.setException(toIOE(e));
            span.end();
        }
        return call;
    }
}