com.datastax.oss.simulacron.common.request.Query Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of simulacron-common Show documentation
There is a newer version: 0.12.0
/*
 * Copyright (C) 2017-2017 DataStax Inc.
 *
 * Licensed 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 com.datastax.oss.simulacron.common.request;

import static com.fasterxml.jackson.annotation.JsonInclude.Include.NON_EMPTY;

import com.datastax.oss.protocol.internal.Frame;
import com.datastax.oss.protocol.internal.request.Execute;
import com.datastax.oss.protocol.internal.request.Prepare;
import com.datastax.oss.protocol.internal.request.query.QueryOptions;
import com.datastax.oss.simulacron.common.codec.ConsistencyLevel;
import com.datastax.oss.simulacron.common.codec.CqlMapper;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.math.BigInteger;
import java.nio.ByteBuffer;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

public final class Query extends Request {
  public final String query;
  public final transient List consistencyEnum;

  @JsonProperty("param_types")
  @JsonInclude(NON_EMPTY)
  public final Map paramTypes;

  @JsonInclude(NON_EMPTY)
  public final Map params;

  public Query(String query) {
    this(query, Collections.emptyList(), null, null);
  }

  @JsonCreator
  public Query(
      @JsonProperty("query") String query,
      @JsonProperty("consistency_level") String[] consistency,
      @JsonProperty("params") Map params,
      @JsonProperty("param_types") Map paramTypes) {
    this(query, createEnumFromConsistency(consistency), params, paramTypes);
  }

  public Query(
      String query,
      List consistencies,
      Map params,
      Map paramTypes) {
    this.query = query;
    this.consistencyEnum = consistencies;
    this.params = params;
    this.paramTypes = paramTypes;
  }

  @JsonProperty("consistency_level")
  @JsonInclude(NON_EMPTY)
  public String[] getConsistency() {
    String[] consistency = new String[consistencyEnum.size()];
    for (int i = 0; i < consistencyEnum.size(); i++) {
      consistency[i] = consistencyEnum.get(i).toString();
    }
    return consistency;
  }

  public boolean matches(Frame frame) {

    if (frame.message instanceof Prepare) {
      Prepare prepare = (Prepare) frame.message;
      if (query.equals(prepare.cqlQuery)) {
        return true;
      }
    }
    if (frame.message instanceof Execute) {
      Execute execute = (Execute) frame.message;
      Integer queryIdInt = new BigInteger(execute.queryId).intValue();
      if (getQueryId() == queryIdInt) {
        return checkParamsMatch(execute.options, frame);
      }
    }
    if (frame.message instanceof com.datastax.oss.protocol.internal.request.Query) {
      // If the query expected matches the primed query example and CL levels match. Return true;
      com.datastax.oss.protocol.internal.request.Query query =
          (com.datastax.oss.protocol.internal.request.Query) frame.message;
      if (this.query.equals(query.query)) {
        ConsistencyLevel level = ConsistencyLevel.fromCode(query.options.consistency);
        // NOTE: Absent CL level means it will match all CL levels
        if (this.consistencyEnum.contains(level) || this.consistencyEnum.size() == 0) {
          // Id any parameters are present we must make sure they match those in the primed queries
          return checkParamsMatch(query.options, frame);
        }
      }
    }
    return false;
  }

  /**
   * * A method that will check to see if primed query and actual query parameters match
   *
   * @param options query options to check.
   * @param frame The incoming frame.
   * @return True if the parameters match;
   */
  private boolean checkParamsMatch(QueryOptions options, Frame frame) {
    // I don't like this, but I can't see a way to simplify it.

    // No params match criteria was specified, simply return true.
    if (params == null || params.size() == 0) {
      return true;
    }

    if (options.namedValues.size() != params.size()
        && options.positionalValues.size() != params.size()) {
      return false;
    }
    if ((options.namedValues.size() != 0 || options.positionalValues.size() != 0)
        && params.size() != 0) {

      CqlMapper mapper = CqlMapper.forVersion(frame.protocolVersion);
      // Positional parameter case, ensure number of parameters is the same.
      if (options.positionalValues.size() != 0) {
        if (options.positionalValues.size() != params.size()) {
          return false;
        }
        Iterator