com.websudos.phantom.dsl.package.scala Maven / Gradle / Ivy
/*
* Copyright 2013-2015 Websudos, Limited.
*
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* - Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
*
* - Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* - Explicit consent must be obtained from the copyright owner, Outworkers Limited before any redistribution is made.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
package com.websudos.phantom
import java.net.InetAddress
import java.nio.ByteBuffer
import java.util.{Date, Random}
import com.datastax.driver.core.utils.UUIDs
import com.datastax.driver.core.{VersionNumber, ConsistencyLevel => CLevel}
import com.websudos.phantom.batch.Batcher
import com.websudos.phantom.builder.QueryBuilder
import com.websudos.phantom.builder.clauses.{UpdateClause, UsingClauseOperations, WhereClause}
import com.websudos.phantom.builder.ops._
import com.websudos.phantom.builder.primitives.{DefaultPrimitives, Primitive}
import com.websudos.phantom.builder.query.{CQLQuery, CreateImplicits, DeleteImplicits, SelectImplicits}
import com.websudos.phantom.builder.syntax.CQLSyntax
import shapeless.{::, HNil}
import scala.concurrent.ExecutionContextExecutor
import scala.util.Try
import scala.util.control.NoStackTrace
package object dsl extends ImplicitMechanism with CreateImplicits
with DefaultPrimitives
with SelectImplicits
with Operators
with UsingClauseOperations
with DeleteImplicits {
type CassandraTable[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.CassandraTable[Owner, Record]
type Column[Owner <: CassandraTable[Owner, Record], Record, T] = com.websudos.phantom.column.Column[Owner, Record, T]
type PrimitiveColumn[Owner <: CassandraTable[Owner, Record], Record, T] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, T]
type OptionalColumn[Owner <: CassandraTable[Owner, Record], Record, T] = com.websudos.phantom.column.OptionalColumn[Owner, Record, T]
type OptionalPrimitiveColumn[Owner <: CassandraTable[Owner, Record], Record, T] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, T]
type BigDecimalColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, BigDecimal]
type BlobColumn[Owner <: CassandraTable[Owner, Record], Record, T] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, ByteBuffer]
type BigIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, BigInt]
type BooleanColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Boolean]
type DateColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Date]
type DateTimeColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, DateTime]
type LocalDateColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, LocalDate]
type DoubleColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Double]
type FloatColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Float]
type IntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Int]
type SmallIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Short]
type TinyIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Byte]
type InetAddressColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, InetAddress]
type LongColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, Long]
type StringColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, String]
type UUIDColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.PrimitiveColumn[Owner, Record, UUID]
type CounterColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.CounterColumn[Owner, Record]
type TimeUUIDColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.TimeUUIDColumn[Owner, Record]
type OptionalBlobColumn[Owner <: CassandraTable[Owner, Record], Record, T] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, ByteBuffer]
type OptionalBigDecimalColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, BigDecimal]
type OptionalBigIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, BigInt]
type OptionalBooleanColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Boolean]
type OptionalDateColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Date]
type OptionalDateTimeColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, DateTime]
type OptionalLocalDateColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, LocalDate]
type OptionalDoubleColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Double]
type OptionalFloatColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Float]
type OptionalIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Int]
type OptionalSmallIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Short]
type OptionalTinyIntColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Byte]
type OptionalInetAddressColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, InetAddress]
type OptionalLongColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, Long]
type OptionalStringColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, String]
type OptionalUUIDColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalPrimitiveColumn[Owner, Record, UUID]
type OptionalTimeUUIDColumn[Owner <: CassandraTable[Owner, Record], Record] = com.websudos.phantom.column.OptionalTimeUUIDColumn[Owner, Record]
type ClusteringOrder[ValueType] = com.websudos.phantom.keys.ClusteringOrder[ValueType]
type Ascending = com.websudos.phantom.keys.Ascending
type Descending = com.websudos.phantom.keys.Descending
type PartitionKey[ValueType] = com.websudos.phantom.keys.PartitionKey[ValueType]
type PrimaryKey[ValueType] = com.websudos.phantom.keys.PrimaryKey[ValueType]
type Index[ValueType] = com.websudos.phantom.keys.Index[ValueType]
type Keys = com.websudos.phantom.keys.Keys
type Entries = com.websudos.phantom.keys.Entries
type StaticColumn[ValueType] = com.websudos.phantom.keys.StaticColumn[ValueType]
type Database = com.websudos.phantom.db.DatabaseImpl
type DateTime = org.joda.time.DateTime
type LocalDate = org.joda.time.LocalDate
type DateTimeZone = org.joda.time.DateTimeZone
type UUID = java.util.UUID
type Row = com.datastax.driver.core.Row
type ResultSet = com.datastax.driver.core.ResultSet
type Session = com.datastax.driver.core.Session
type KeySpace = com.websudos.phantom.connectors.KeySpace
val KeySpace = com.websudos.phantom.connectors.KeySpace
type RootConnector = com.websudos.phantom.connectors.RootConnector
val Version = com.websudos.phantom.connectors.DefaultVersions
type ListResult[R] = com.websudos.phantom.builder.query.ListResult[R]
type IteratorResult[R] = com.websudos.phantom.builder.query.IteratorResult[R]
type RecordResult[R] = com.websudos.phantom.builder.query.RecordResult[R]
case object Batch extends Batcher
object ConsistencyLevel {
val ALL = CLevel.ALL
val Any = CLevel.ANY
val ONE = CLevel.ONE
val TWO = CLevel.TWO
val THREE = CLevel.THREE
val QUORUM = CLevel.QUORUM
val LOCAL_QUORUM = CLevel.LOCAL_QUORUM
val EACH_QUORUM = CLevel.EACH_QUORUM
val LOCAL_SERIAL = CLevel.LOCAL_SERIAL
val LOCAL_ONE = CLevel.LOCAL_ONE
val SERIAL = CLevel.SERIAL
}
type KeySpaceDef = com.websudos.phantom.connectors.KeySpaceDef
val ContactPoint = com.websudos.phantom.connectors.ContactPoint
val ContactPoints = com.websudos.phantom.connectors.ContactPoints
implicit def enumToQueryConditionPrimitive[T <: Enumeration](enum: T): Primitive[T#Value] = {
new Primitive[T#Value] {
override type PrimitiveType = java.lang.String
override def cassandraType: String = Primitive[String].cassandraType
override def fromRow(name: String, row: Row): Try[T#Value] = {
nullCheck(name, row) {
r => enum.values.find(_.toString == r.getString(name)) match {
case Some(value) => value
case _ => throw new Exception(s"Value $name not found in enumeration") with NoStackTrace
}
}
}
override def asCql(value: T#Value): String = Primitive[String].asCql(value.toString)
override def fromString(value: String): T#Value = enum.values.find(value == _.toString).getOrElse(None.orNull)
override def clz: Class[String] = classOf[java.lang.String]
}
}
implicit class RichNumber(val percent: Int) extends AnyVal {
def percentile: CQLQuery = CQLQuery(percent.toString).append(CQLSyntax.CreateOptions.percentile)
}
implicit def primitiveToTokenOp[RR : Primitive](value: RR): TokenConstructor[RR :: HNil, TokenTypes.ValueToken] = {
new TokenConstructor(Seq(Primitive[RR].asCql(value)))
}
implicit lazy val context: ExecutionContextExecutor = Manager.scalaExecutor
implicit class PartitionTokenHelper[T](val p: Column[_, _, T] with PartitionKey[T]) extends AnyVal {
def ltToken (value: T): WhereClause.Condition = {
new WhereClause.Condition(
QueryBuilder.Where.lt(
QueryBuilder.Where.token(p.name).queryString,
QueryBuilder.Where.fcall(CQLSyntax.token, p.asCql(value)).queryString
)
)
}
def lteToken (value: T): WhereClause.Condition = {
new WhereClause.Condition(
QueryBuilder.Where.lte(
QueryBuilder.Where.token(p.name).queryString,
QueryBuilder.Where.fcall(CQLSyntax.token, p.asCql(value)).queryString
)
)
}
def gtToken (value: T): WhereClause.Condition = {
new WhereClause.Condition(
QueryBuilder.Where.gt(
QueryBuilder.Where.token(p.name).queryString,
QueryBuilder.Where.fcall(CQLSyntax.token, p.asCql(value)).queryString
)
)
}
def gteToken (value: T): WhereClause.Condition = {
new WhereClause.Condition(
QueryBuilder.Where.gte(
QueryBuilder.Where.token(p.name).queryString,
QueryBuilder.Where.fcall(CQLSyntax.token, p.asCql(value)).queryString
)
)
}
def eqsToken (value: T): WhereClause.Condition = {
new WhereClause.Condition(
QueryBuilder.Where.eqs(
QueryBuilder.Where.token(p.name).queryString,
QueryBuilder.Where.fcall(CQLSyntax.token, p.asCql(value)).queryString
)
)
}
}
implicit class CounterOperations[Owner <: CassandraTable[Owner, Record], Record](val col: CounterColumn[Owner, Record]) extends AnyVal {
final def +=[T : Numeric](value: T): UpdateClause.Condition = {
new UpdateClause.Condition(QueryBuilder.Update.increment(col.name, value.toString))
}
final def increment[T : Numeric](value: T): UpdateClause.Condition = +=(value)
final def -=[T : Numeric](value: T): UpdateClause.Condition = {
new UpdateClause.Condition(QueryBuilder.Update.decrement(col.name, value.toString))
}
final def decrement[T : Numeric](value: T): UpdateClause.Condition = -=(value)
}
/**
* Augments Cassandra VersionNumber descriptors to support simple comparison of versions.
* This allows for operations that can differ based on the Cassandra version used by the session.
*
* @param version The Cassandra version number.
*/
implicit class VersionAugmenter(val version: VersionNumber) extends AnyVal {
def <(other: VersionNumber): Boolean = version.compareTo(other) == -1
def ===(other: VersionNumber): Boolean = version.compareTo(other) == 0
def > (other: VersionNumber): Boolean = version.compareTo(other) == 1
def >= (other: VersionNumber): Boolean = {
version.compareTo(other) >= 0
}
}
implicit class DateTimeAugmenter(val date: DateTime) extends AnyVal {
def timeuuid(): UUID = {
val random = new Random()
new UUID(UUIDs.startOf(date.getMillis).getMostSignificantBits, random.nextLong())
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy