io.getquill.context.cassandra.Ops.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quill-cassandra_2.12 Show documentation

quill-cassandra

The newest version!

package io.getquill.context.cassandra

import io.getquill.{Action, Delete, EntityQuery, Insert, Query, Update}

//noinspection TypeAnnotation
trait Ops {
  this: CassandraContext[_] =>

  abstract class Options[A](q: A) {
    def usingTimestamp(ts: Int)  = quote(sql"$q USING TIMESTAMP $ts".as[A])
    def usingTtl(ttl: Int)       = quote(sql"$q USING TTL $ttl".as[A])
    def using(ts: Int, ttl: Int) = quote(sql"$q USING TIMESTAMP $ts AND TTL $ttl".as[A])
  }

  implicit final class QueryOps[Q <: Query[_]](q: Q) {
    def allowFiltering = quote(sql"$q ALLOW FILTERING".transparent.pure.as[Q])
  }

  implicit final class EntityOps[A <: EntityQuery[_]](q: A) extends Options(q)

  implicit final class InsertOps[A <: Insert[_]](q: A) extends Options(q) {
    def ifNotExists = quote(sql"$q IF NOT EXISTS".as[A])
  }

  implicit final class UpdateOps[A <: Update[_]](q: A) extends Options(q) {
    def ifExists = quote(sql"$q IF EXISTS".as[A])
  }

  implicit final class DeleteOps[A <: Delete[_]](q: A) extends Options(q) {
    def ifExists = quote(sql"$q IF EXISTS".as[A])
  }

  implicit final class ActionOps[T](q: Action[T]) {
    def ifCond(cond: T => Boolean) =
      quote(sql"$q IF $cond".as[Action[T]])
  }

  implicit final class MapOps[K, V](map: Map[K, V]) {
    def containsValue(value: V) = quote(sql"$map CONTAINS $value".as[Boolean])
  }
}