org.apache.kudu.client.KuduSession Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camel-quarkus-kudu-client
org.apache.kudu:kudu-client with netty package relocations reverted and netty classes stripped away so that camel-quarkus-kudu can use quarkus-netty as a replacement
The newest version!
// 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.kudu.client;

import java.util.List;

import com.stumbleupon.async.Deferred;
import org.apache.yetus.audience.InterfaceAudience;
import org.apache.yetus.audience.InterfaceStability;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Synchronous version of {@link AsyncKuduSession}.
 * Offers the same API but with blocking methods.
 *
 * This class is not thread-safe.

 */
@InterfaceAudience.Public
@InterfaceStability.Evolving
public class KuduSession implements SessionConfiguration {

  private static final Logger LOG = LoggerFactory.getLogger(KuduSession.class);
  private final AsyncKuduSession session;

  KuduSession(AsyncKuduSession session) {
    this.session = session;
  }

  /**
   * Apply a given {@link Operation} to Kudu as part of this session.
   *
   * 

   * This is a blocking call that has different behavior based on the configured flush mode:
   *
   * 

   * {@link SessionConfiguration.FlushMode#AUTO_FLUSH_SYNC AUTO_FLUSH_SYNC}:
   * the call returns when the operation is persisted, else it throws an exception.
   *
   * 
{@link SessionConfiguration.FlushMode#AUTO_FLUSH_BACKGROUND AUTO_FLUSH_BACKGROUND}:
   * the call returns when the operation has been added to the buffer.
   * This call should normally perform only fast in-memory operations but
   * it may have to wait when the buffer is full and there's another buffer being flushed. Row
   * errors can be checked by calling {@link #countPendingErrors()} and can be retrieved by calling
   * {@link #getPendingErrors()}.
   *
   * 
{@link SessionConfiguration.FlushMode#MANUAL_FLUSH MANUAL_FLUSH}:
   * the call returns when the operation has been added to the buffer, else it throws a
   * {@link KuduException} if the buffer is full.
   * 
   *
   * 
   * Note: {@link PleaseThrottleException} is handled by this method and will not be thrown, unlike
   * with {@link AsyncKuduSession#apply AsyncKuduSession.apply()}.
   *
   * @param operation operation to apply
   * @return an OperationResponse for the applied Operation
   * @throws KuduException if anything went wrong
   * @see SessionConfiguration.FlushMode FlushMode
   */
  public OperationResponse apply(Operation operation) throws KuduException {
    while (true) {
      try {
        Deferred d = session.apply(operation);
        if (getFlushMode() == FlushMode.AUTO_FLUSH_SYNC) {
          return d.join();
        }
        break;
      } catch (PleaseThrottleException ex) {
        try {
          ex.getDeferred().join();
        } catch (Exception e) {
          // This is the error response from the buffer that was flushing,
          // we can't do much with it at this point.
          LOG.error("Previous batch had this exception", e);
        }
      } catch (Exception e) {
        throw KuduException.transformException(e);
      }
    }
    return null;
  }

  /**
   * Blocking call that force flushes this session's buffers. Data is persisted when this call
   * returns, else it will throw an exception.
   * @return a list of OperationResponse, one per operation that was flushed
   * @throws KuduException if anything went wrong
   */
  public List flush() throws KuduException {
    return KuduClient.joinAndHandleException(session.flush());
  }

  /**
   * Blocking call that flushes the buffers (see {@link #flush()}) and closes the sessions.
   * @return List of OperationResponse, one per operation that was flushed
   * @throws KuduException if anything went wrong
   */
  public List close() throws KuduException {
    return KuduClient.joinAndHandleException(session.close());
  }

  @Override
  public FlushMode getFlushMode() {
    return session.getFlushMode();
  }

  @Override
  public void setFlushMode(FlushMode flushMode) {
    session.setFlushMode(flushMode);
  }

  @Override
  public void setMutationBufferSpace(int numOps) {
    session.setMutationBufferSpace(numOps);
  }

  @Override
  public void setErrorCollectorSpace(int size) {
    session.setErrorCollectorSpace(size);
  }

  /**
   * @deprecated
   */
  @Override
  @Deprecated
  public void setMutationBufferLowWatermark(float mutationBufferLowWatermarkPercentage) {
    LOG.warn("setMutationBufferLowWatermark is deprecated");
  }

  @Override
  public void setFlushInterval(int intervalMillis) {
    session.setFlushInterval(intervalMillis);
  }

  @Override
  public long getTimeoutMillis() {
    return session.getTimeoutMillis();
  }

  @Override
  public void setTimeoutMillis(long timeout) {
    session.setTimeoutMillis(timeout);
  }

  @Override
  public boolean isClosed() {
    return session.isClosed();
  }

  @Override
  public boolean hasPendingOperations() {
    return session.hasPendingOperations();
  }

  @Override
  public void setExternalConsistencyMode(ExternalConsistencyMode consistencyMode) {
    session.setExternalConsistencyMode(consistencyMode);
  }

  @Override
  public boolean isIgnoreAllDuplicateRows() {
    return session.isIgnoreAllDuplicateRows();
  }

  @Override
  public void setIgnoreAllDuplicateRows(boolean ignoreAllDuplicateRows) {
    session.setIgnoreAllDuplicateRows(ignoreAllDuplicateRows);
  }

  @Override
  public boolean isIgnoreAllNotFoundRows() {
    return session.isIgnoreAllNotFoundRows();
  }

  @Override
  public void setIgnoreAllNotFoundRows(boolean ignoreAllNotFoundRows) {
    session.setIgnoreAllNotFoundRows(ignoreAllNotFoundRows);
  }

  @Override
  public int countPendingErrors() {
    return session.countPendingErrors();
  }

  @Override
  public RowErrorsAndOverflowStatus getPendingErrors() {
    return session.getPendingErrors();
  }

  @Override
  public ResourceMetrics getWriteOpMetrics() {
    return session.getWriteOpMetrics();
  }
}