Examples with ThriftIOException alluxio.thrift.ThriftIOException used on opensource projects

Example 1 with ThriftIOException

use of alluxio.thrift.ThriftIOException in project alluxio by Alluxio.

the class RetryHandlingFileSystemWorkerClient method sessionHeartbeat.

@Override
public void sessionHeartbeat(RetryPolicy retryPolicy) throws IOException, InterruptedException {
    TException exception;
    do {
        FileSystemWorkerClientService.Client client = mClientHeartbeatPool.acquire();
        try {
            client.sessionHeartbeat(mSessionId, null);
            Metrics.FILE_SYSTEM_WORKER_HEARTBEATS.inc();
            return;
        } catch (AlluxioTException | ThriftIOException e) {
            exception = e;
            LOG.warn(e.getMessage());
        } catch (TException e) {
            client.getOutputProtocol().getTransport().close();
            exception = e;
            LOG.warn(e.getMessage());
        } finally {
            mClientHeartbeatPool.release(client);
        }
    } while (retryPolicy.attemptRetry());
    Preconditions.checkNotNull(exception);
    throw new IOException(exception);
}

Example 2 with ThriftIOException

use of alluxio.thrift.ThriftIOException in project alluxio by Alluxio.

the class RetryHandlingBlockWorkerClient method sessionHeartbeat.

/**
   * sessionHeartbeat is not retried because it is supposed to be called periodically.
   *
   * @throws IOException if it fails to heartbeat
   * @throws InterruptedException if heartbeat is interrupted
   */
@Override
public void sessionHeartbeat(RetryPolicy retryPolicy) throws IOException, InterruptedException {
    TException exception;
    do {
        BlockWorkerClientService.Client client = mClientHeartbeatPool.acquire();
        try {
            client.sessionHeartbeat(mSessionId, null);
            Metrics.BLOCK_WORKER_HEATBEATS.inc();
            return;
        } catch (AlluxioTException e) {
            AlluxioException ae = AlluxioException.fromThrift(e);
            LOG.warn(ae.getMessage());
            throw new IOException(ae);
        } catch (ThriftIOException e) {
            LOG.warn(e.getMessage());
            throw new IOException(e);
        } catch (TException e) {
            client.getOutputProtocol().getTransport().close();
            exception = e;
            LOG.warn(e.getMessage());
        } finally {
            mClientHeartbeatPool.release(client);
        }
    } while (retryPolicy.attemptRetry());
    Preconditions.checkNotNull(exception);
    throw new IOException(exception);
}

Example 3 with ThriftIOException

use of alluxio.thrift.ThriftIOException in project alluxio by Alluxio.

the class AbstractThriftClient method retryRPC.

/**
   * Similar to {@link #retryRPC(RpcCallable)} except that the RPC call may throw
   * {@link AlluxioTException} and once it is thrown, it will be transformed into
   * {@link AlluxioException} and be thrown.
   *
   * @param rpc the RPC call to be executed
   * @param <V> type of return value of the RPC call
   * @return the return value of the RPC call
   * @throws AlluxioException when {@link AlluxioTException} is thrown by the RPC call
   * @throws IOException when retries exceeds {@link #RPC_MAX_NUM_RETRY} or some server
   *         side IOException occurred.
   */
protected <V> V retryRPC(RpcCallableThrowsAlluxioTException<V, C> rpc) throws AlluxioException, IOException {
    TException exception = null;
    RetryPolicy retryPolicy = new ExponentialBackoffRetry(BASE_SLEEP_MS, MAX_SLEEP_MS, RPC_MAX_NUM_RETRY);
    do {
        C client = acquireClient();
        try {
            return rpc.call(client);
        } catch (AlluxioTException e) {
            AlluxioException ae = AlluxioException.fromThrift(e);
            processException(client, ae);
            exception = new TException(ae);
        } catch (ThriftIOException e) {
            throw new IOException(e);
        } catch (TException e) {
            LOG.error(e.getMessage(), e);
            closeClient(client);
            exception = e;
        } finally {
            releaseClient(client);
        }
    } while (retryPolicy.attemptRetry());
    LOG.error("Failed after " + retryPolicy.getRetryCount() + " retries.");
    Preconditions.checkNotNull(exception);
    throw new IOException(exception);
}

Example 4 with ThriftIOException

use of alluxio.thrift.ThriftIOException in project alluxio by Alluxio.

the class AbstractThriftClient method retryRPC.

/**
   * Tries to execute an RPC defined as a {@link RpcCallable}.
   *
   * @param rpc the RPC call to be executed
   * @param <V> type of return value of the RPC call
   * @return the return value of the RPC call
   * @throws IOException when retries exceeds {@link #RPC_MAX_NUM_RETRY} or some server
   *         side IOException occurred.
   */
protected <V> V retryRPC(RpcCallable<V, C> rpc) throws IOException {
    TException exception;
    RetryPolicy retryPolicy = new ExponentialBackoffRetry(BASE_SLEEP_MS, MAX_SLEEP_MS, RPC_MAX_NUM_RETRY);
    do {
        C client = acquireClient();
        try {
            return rpc.call(client);
        } catch (ThriftIOException e) {
            throw new IOException(e);
        } catch (AlluxioTException e) {
            AlluxioException ae = AlluxioException.fromThrift(e);
            try {
                processException(client, ae);
            } catch (AlluxioException ee) {
                throw new IOException(ee);
            }
            exception = new TException(ae);
        } catch (TException e) {
            LOG.warn(e.getMessage());
            closeClient(client);
            exception = e;
        } finally {
            releaseClient(client);
        }
    } while (retryPolicy.attemptRetry());
    LOG.error("Failed after " + retryPolicy.getRetryCount() + " retries.");
    Preconditions.checkNotNull(exception);
    throw new IOException(exception);
}