org.scassandra.server.priming.query.PrimeQuerySingle.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scassandra-server_2.11 Show documentation
Scassandra server
The newest version!
/*
 * Copyright (C) 2016 Christopher Batey and Dogan Narinc
 *
 * 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 org.scassandra.server.priming.query

import org.scassandra.codec.Consistency.Consistency
import org.scassandra.codec.datatype.DataType
import org.scassandra.server.priming.Defaulter._
import org.scassandra.server.priming.json.ResultJsonRepresentation
import org.scassandra.server.priming.routes.PrimingJsonHelper.extractPrime

/*
  These case classes are for parsing the incoming JSON for /prime-query-single

  Really variable_types make more sense in the When but for reason I put them in the
  Then for prepared statements and I didn't want the two to be inconsistent
 */
case class PrimeQuerySingle(when: When, thenDo: Then) {
  @transient lazy val prime = {
    extractPrime(thenDo, when.keyspace, when.table)
  }

  def withDefaults: PrimeQuerySingle = {
    PrimeQuerySingle(
      when.withDefaults,
      thenDo.withDefaults(when.query)
    )
  }
}

case class When(query: Option[String] = None,
                queryPattern: Option[String] = None,
                consistency: Option[List[Consistency]] = None,
                keyspace: Option[String] = None,
                table: Option[String] = None
               ) {
  def withDefaults: When = copy(consistency = defaultConsistency(consistency))
}

trait ThenProvider {
  val rows: Option[List[Map[String, Any]]]
  val result: Option[ResultJsonRepresentation]
  val column_types: Option[Map[String, DataType]]
  val fixedDelay: Option[Long]
  val config: Option[Map[String, String]]
}

case class Then(rows: Option[List[Map[String, Any]]] = None,
                result: Option[ResultJsonRepresentation] = None,
                column_types: Option[Map[String, DataType]] = None,
                fixedDelay: Option[Long] = None,
                config: Option[Map[String, String]] = None,
                variable_types: Option[List[DataType]] = None) extends ThenProvider {

  def withDefaults(query: Option[String]): Then =
    copy(
      variable_types = defaultVariableTypesToVarChar(query, variable_types),
      column_types = defaultColumnTypesToVarChar(column_types, rows)
    )
}