io.stargate.db.query.builder.AbstractBound Maven / Gradle / Ivy

Go to download
package io.stargate.db.query.builder;

import com.datastax.oss.driver.shaded.guava.common.annotations.VisibleForTesting;
import com.datastax.oss.driver.shaded.guava.common.base.Preconditions;
import io.stargate.db.Parameters;
import io.stargate.db.datastore.ResultSet;
import io.stargate.db.query.AsyncQueryExecutor;
import io.stargate.db.query.BoundQuery;
import io.stargate.db.query.QueryType;
import io.stargate.db.query.TypedValue;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.function.UnaryOperator;
import org.apache.cassandra.stargate.db.ConsistencyLevel;

// In theory, we should access this outside of this package. However, some test cast a BoundQuery
// into this class to reach into the underlying BuiltQuery, and this out of convenience. This could
// be made package private by changing those tests, but it's probably not a big deal.
@VisibleForTesting
public class AbstractBound> implements BoundQuery {
  private final Source source;
  private final List values;

  AbstractBound(Q builtQuery, List boundedValues, List values) {
    this.source = new Source<>(builtQuery, boundedValues);
    this.values = values;
  }

  @Override
  public QueryType type() {
    return source.query().type();
  }

  @Override
  public Source source() {
    return source;
  }

  @Override
  public List values() {
    return values;
  }

  private AsyncQueryExecutor executor() {
    Preconditions.checkState(
        source.query().executor() != null, "Cannot execute query: it has no attached executor");
    return source.query().executor();
  }

  /**
   * Executes this query using the underlying executor the query was built with.
   *
   * See the {@link AsyncQueryExecutor#execute(BoundQuery)}.
   */
  public CompletableFuture execute() {
    return executor().execute(this);
  }

  /**
   * Executes this query using the underlying executor the query was built with.
   *
   * 
See the {@link AsyncQueryExecutor#execute(BoundQuery, ConsistencyLevel)}.
   */
  public CompletableFuture execute(ConsistencyLevel consistencyLevel) {
    return executor().execute(this, consistencyLevel);
  }

  /**
   * Executes this query using the underlying executor the query was built with.
   *
   * See the {@link AsyncQueryExecutor#execute(BoundQuery, UnaryOperator)}.
   */
  public CompletableFuture execute(UnaryOperator parametersModifier) {
    return executor().execute(this, parametersModifier);
  }
}