Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.thrift.columns
import scala.annotation.implicitNotFound
import scala.collection.JavaConverters._
import scala.util.{Success, Try}
import com.websudos.phantom.builder.QueryBuilder
import com.websudos.phantom.builder.QueryBuilder.Utils
import com.websudos.phantom.builder.query.CQLQuery
import com.websudos.phantom.builder.syntax.CQLSyntax
import com.datastax.driver.core.{GettableData, Row}
import com.twitter.scrooge.{CompactThriftSerializer, ThriftStruct, ThriftStructSerializer}
import com.websudos.phantom.builder.primitives.Primitive
import com.websudos.phantom.column.{AbstractListColumn, AbstractMapColumn, AbstractSetColumn, CollectionValueDefinition, Column, OptionalColumn}
import com.websudos.phantom.CassandraTable
trait ThriftColumnDefinition[ValueType <: ThriftStruct] {
/**
* The Thrift serializer to use.
* This must always be overriden before usage.
* It is a simple forwarding from a concrete Thrift Struct.
*/
implicit def serializer: CompactThriftSerializer[ValueType]
/**
* This converts a value to the appropiate Cassandra type.
* All Thrift structs are serialized to strings.
*
* @param v The Thrift struct to convert.
* @return A string containing the compact Thrift serialization.
*/
def asCql(v: ValueType): String = {
CQLQuery.empty.singleQuote(serializer.toString(v))
}
def valueAsCql(v: ValueType): String = asCql(v)
val primitive = implicitly[Primitive[String]]
}
trait CollectionThriftColumnDefinition[ValueType <: ThriftStruct] extends ThriftColumnDefinition[ValueType] with CollectionValueDefinition[ValueType] {
def fromString(c: String): ValueType = serializer.fromString(c)
}
abstract class ThriftColumn[T <: CassandraTable[T, R], R, ValueType <: ThriftStruct](table: CassandraTable[T, R])
extends Column[T, R, ValueType](table) with ThriftColumnDefinition[ValueType] {
val cassandraType = CQLSyntax.Types.Text
def parse(r: Row): Try[ValueType] = {
Try(serializer.fromString(r.getString(name)))
}
}
abstract class OptionalThriftColumn[T <: CassandraTable[T, R], R, ValueType <: ThriftStruct](table: CassandraTable[T, R])
extends OptionalColumn[T, R, ValueType](table)
with ThriftColumnDefinition[ValueType] {
val cassandraType = CQLSyntax.Types.Text
def asCql(v: Option[ValueType]): String = {
v.map(item => CQLQuery.empty.singleQuote(serializer.toString(item))).orNull
}
def optional(r: Row): Try[ValueType] = {
Try(serializer.fromString(r.getString(name)))
}
}
abstract class ThriftSetColumn[T <: CassandraTable[T, R], R, ValueType <: ThriftStruct](table: CassandraTable[T, R])
extends AbstractSetColumn[T, R, ValueType](table) with CollectionThriftColumnDefinition[ValueType] {
override val cassandraType = QueryBuilder.Collections.setType(CQLSyntax.Types.Text).queryString
override def asCql(v: Set[ValueType]): String = Utils.set(v.map(valueAsCql)).queryString
override def parse(r: Row): Try[Set[ValueType]] = {
if (r.isNull(name)) {
Success(Set.empty[ValueType])
} else {
Success(r.getSet(name, Primitive[String].clz.asInstanceOf[Class[String]]).asScala.map(fromString).toSet[ValueType])
}
}
}
abstract class ThriftListColumn[T <: CassandraTable[T, R], R, ValueType <: ThriftStruct](table: CassandraTable[T, R])
extends AbstractListColumn[T, R, ValueType](table) with CollectionThriftColumnDefinition[ValueType] {
override val cassandraType = QueryBuilder.Collections.listType(CQLSyntax.Types.Text).queryString
override def asCql(v: List[ValueType]): String = Utils.collection(v.map(valueAsCql)).queryString
override def parse(r: Row): Try[List[ValueType]] = {
if (r.isNull(name)) {
Success(Nil)
} else {
Success(r.getList(name, Primitive[String].clz.asInstanceOf[Class[String]]).asScala.map(fromString).toList)
}
}
}
@implicitNotFound(msg = "Type ${KeyType} must be a Cassandra primitive")
abstract class ThriftMapColumn[T <: CassandraTable[T, R], R, KeyType : Primitive, ValueType <: ThriftStruct](table: CassandraTable[T, R])
extends AbstractMapColumn[T, R, KeyType, ValueType](table) with CollectionThriftColumnDefinition[ValueType] {
val keyPrimitive = Primitive[KeyType]
override val cassandraType = QueryBuilder.Collections.mapType(keyPrimitive.cassandraType, CQLSyntax.Types.Text).queryString
override def keyAsCql(v: KeyType): String = keyPrimitive.asCql(v)
override def keyFromCql(c: String): KeyType = keyPrimitive.fromString(c)
override def parse(r: Row): Try[Map[KeyType, ValueType]] = {
if (r.isNull(name)) {
Success(Map.empty[KeyType, ValueType])
} else {
Try(
r.getMap(name, keyPrimitive.clz.asInstanceOf[Class[KeyType]],
Primitive[String].clz.asInstanceOf[Class[String]]
).asScala.toMap map {
case (k, v) => (k, serializer.fromString(v))
}
)
}
}
}
abstract class RootThriftPrimitive[T <: ThriftStruct] extends Primitive[T] {
def serializer: ThriftStructSerializer[T]
override type PrimitiveType = java.lang.String
override def fromRow(column: String, row: GettableData): Try[T] = nullCheck(column, row) {
existing => serializer.fromString(row.getString(column))
}
override def cassandraType: String = CQLSyntax.Types.Text
override def fromString(value: String): T = serializer.fromString(value)
override def asCql(value: T): String = Primitive[String].asCql(serializer.toString(value))
override def clz: Class[java.lang.String] = classOf[java.lang.String]
}
