com.datastax.oss.simulacron.common.cluster.QueryLog 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.cluster;

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;
import com.datastax.oss.simulacron.common.codec.ConsistencyLevel;
import com.datastax.oss.simulacron.common.stubbing.Prime;
import com.datastax.oss.simulacron.common.stubbing.StubMapping;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.net.SocketAddress;
import java.util.Optional;

public class QueryLog {

  @JsonProperty("type")
  private String type;

  @JsonProperty("query")
  private String query;

  @JsonProperty("consistency_level")
  private ConsistencyLevel consistency;

  @JsonProperty("serial_consistency_level")
  private ConsistencyLevel serialConsistency;

  @JsonProperty("connection")
  private SocketAddress connection;

  @JsonProperty("timestamp")
  private long timestamp;

  @JsonProperty("primed")
  private boolean primed;

  @JsonIgnore private Frame frame;

  @JsonCreator
  public QueryLog(
      @JsonProperty("query") String query,
      @JsonProperty("consistency_level") ConsistencyLevel consistency,
      @JsonProperty("serial_consistency_level") ConsistencyLevel serialConsistency,
      @JsonProperty("connection") SocketAddress connection,
      @JsonProperty("timestamp") long timestamp,
      @JsonProperty("primed") boolean primed) {
    this.query = query;
    this.consistency = consistency;
    this.serialConsistency = serialConsistency;
    this.connection = connection;
    this.timestamp = timestamp;
    this.primed = primed;
  }

  QueryLog(
      Frame frame,
      SocketAddress connection,
      long timestamp,
      boolean primed,
      Optional stubOption) {
    this.frame = frame;
    this.connection = connection;
    this.timestamp = timestamp;
    this.primed = primed;
    this.type = frame.message.getClass().getSimpleName().toUpperCase();

    if (frame.message instanceof Query) {
      Query query = (Query) frame.message;
      this.query = query.query;
      this.consistency = ConsistencyLevel.fromCode(query.options.consistency);
      this.serialConsistency = ConsistencyLevel.fromCode(query.options.serialConsistency);
    } else if (frame.message instanceof Execute) {
      Execute execute = (Execute) frame.message;
      this.consistency = ConsistencyLevel.fromCode(execute.options.consistency);
      this.serialConsistency = ConsistencyLevel.fromCode(execute.options.serialConsistency);
      if (stubOption.isPresent()) {
        StubMapping stub = stubOption.get();
        if (stub instanceof Prime) {
          Prime prime = (Prime) stub;
          if (prime.getPrimedRequest().when
              instanceof com.datastax.oss.simulacron.common.request.Query) {
            com.datastax.oss.simulacron.common.request.Query query =
                (com.datastax.oss.simulacron.common.request.Query) prime.getPrimedRequest().when;
            this.query = query.query;
          }
        }
      }
    } else if (frame.message instanceof Prepare) {
      Prepare prepare = (Prepare) frame.message;
      this.query = prepare.cqlQuery;
    } else {
      // in the case where we don't know how to extract info from the message, just set the query to
      // the type of message.
      this.query = frame.message.getClass().getSimpleName().toUpperCase();
    }
  }

  public String getType() {
    return type;
  }

  public String getQuery() {
    return query;
  }

  public ConsistencyLevel getConsistency() {
    return consistency;
  }

  public ConsistencyLevel getSerialConsistency() {
    return serialConsistency;
  }

  public SocketAddress getConnection() {
    return connection;
  }

  public long getTimestamp() {
    return timestamp;
  }

  public boolean isPrimed() {
    return primed;
  }

  /** @return The frame associated with this log if present. */
  @JsonIgnore
  public Frame getFrame() {
    return this.frame;
  }

  @Override
  public String toString() {
    return "QueryLog{"
        + "type='"
        + type
        + '\''
        + ", query='"
        + query
        + '\''
        + ", consistency="
        + consistency
        + ", serialConsistency="
        + serialConsistency
        + ", connection="
        + connection
        + ", timestamp="
        + timestamp
        + ", primed="
        + primed
        + '}';
  }
}