171 lines
6.1 KiB
HTML
171 lines
6.1 KiB
HTML
|
|
||
|
|
<html>
|
||
|
|
<head>
|
||
|
|
<link href='./_highlighter/SyntaxHighlighter.css' rel='stylesheet' type='text/css'/>
|
||
|
|
<script language='javascript' src='./_highlighter/shAll.js'></script>
|
||
|
|
</head>
|
||
|
|
<body>
|
||
|
|
<pre name="code" class="scala" style="width:100%">
|
||
|
|
/**
|
||
|
|
* Copyright (C) 2009 Scalable Solutions.
|
||
|
|
*/
|
||
|
|
|
||
|
|
package se.scalablesolutions.akka.state
|
||
|
|
|
||
|
|
import java.io.{Flushable, Closeable}
|
||
|
|
|
||
|
|
import util.Logging
|
||
|
|
import util.Helpers._
|
||
|
|
import serialization.Serializer
|
||
|
|
import akka.Config.config
|
||
|
|
|
||
|
|
import scala.collection.mutable.Map
|
||
|
|
|
||
|
|
import org.apache.cassandra.db.ColumnFamily
|
||
|
|
import org.apache.cassandra.service._
|
||
|
|
|
||
|
|
import org.apache.thrift.transport._
|
||
|
|
import org.apache.thrift.protocol._
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
||
|
|
*/
|
||
|
|
trait CassandraSession extends Closeable with Flushable {
|
||
|
|
import scala.collection.jcl.Conversions._
|
||
|
|
import org.scala_tools.javautils.Imports._
|
||
|
|
import java.util.{Map => JMap, List => JList}
|
||
|
|
|
||
|
|
protected val client: Cassandra.Client
|
||
|
|
protected val keyspace: String
|
||
|
|
|
||
|
|
val obtainedAt: Long
|
||
|
|
val consistencyLevel: Int
|
||
|
|
val schema: JMap[String, JMap[String, String]]
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Count is always the max number of results to return.
|
||
|
|
|
||
|
|
So it means, starting with `start`, or the first one if start is
|
||
|
|
empty, go until you hit `finish` or `count`, whichever comes first.
|
||
|
|
Empty is not a legal column name so if finish is empty it is ignored
|
||
|
|
and only count is used.
|
||
|
|
|
||
|
|
We don't offer a numeric offset since that can't be supported
|
||
|
|
efficiently with a log-structured merge disk format.
|
||
|
|
*/
|
||
|
|
|
||
|
|
// ====================================
|
||
|
|
// ====== Scala-style API names
|
||
|
|
// ====================================
|
||
|
|
|
||
|
|
def /(key: String, columnParent: ColumnParent, start: Array[Byte], end: Array[Byte], ascending: Boolean, count: Int): List[ColumnOrSuperColumn] =
|
||
|
|
/(key, columnParent, start, end, ascending, count, consistencyLevel)
|
||
|
|
|
||
|
|
def /(key: String, columnParent: ColumnParent, start: Array[Byte], end: Array[Byte], ascending: Boolean, count: Int, consistencyLevel: Int): List[ColumnOrSuperColumn] =
|
||
|
|
client.get_slice(keyspace, key, columnParent, new SlicePredicate(null, new SliceRange(start, end, ascending, count)), consistencyLevel).toList
|
||
|
|
|
||
|
|
def /(key: String, columnParent: ColumnParent, slicePredicate: SlicePredicate, consistencyLevel: Int): List[ColumnOrSuperColumn] =
|
||
|
|
client.get_slice(keyspace, key, columnParent, slicePredicate, consistencyLevel).toList
|
||
|
|
|
||
|
|
def |(key: String, colPath: ColumnPath): Option[ColumnOrSuperColumn] =
|
||
|
|
|(key, colPath, consistencyLevel)
|
||
|
|
|
||
|
|
def |(key: String, colPath: ColumnPath, consistencyLevel: Int): Option[ColumnOrSuperColumn] =
|
||
|
|
client.get(keyspace, key, colPath, consistencyLevel)
|
||
|
|
|
||
|
|
def |#(key: String, columnParent: ColumnParent): Int =
|
||
|
|
|#(key, columnParent, consistencyLevel)
|
||
|
|
|
||
|
|
def |#(key: String, columnParent: ColumnParent, consistencyLevel: Int): Int =
|
||
|
|
client.get_count(keyspace, key, columnParent, consistencyLevel)
|
||
|
|
|
||
|
|
def ++|(key: String, colPath: ColumnPath, value: Array[Byte]): Unit =
|
||
|
|
++|(key, colPath, value, obtainedAt, consistencyLevel)
|
||
|
|
|
||
|
|
def ++|(key: String, colPath: ColumnPath, value: Array[Byte], timestamp: Long): Unit =
|
||
|
|
++|(key, colPath, value, timestamp, consistencyLevel)
|
||
|
|
|
||
|
|
def ++|(key: String, colPath: ColumnPath, value: Array[Byte], timestamp: Long, consistencyLevel: Int) =
|
||
|
|
client.insert(keyspace, key, colPath, value, timestamp, consistencyLevel)
|
||
|
|
|
||
|
|
def ++|(key: String, batch: Map[String, List[ColumnOrSuperColumn]]): Unit =
|
||
|
|
++|(key, batch, consistencyLevel)
|
||
|
|
|
||
|
|
def ++|(key: String, batch: Map[String, List[ColumnOrSuperColumn]], consistencyLevel: Int): Unit = {
|
||
|
|
val jmap = new java.util.HashMap[String, JList[ColumnOrSuperColumn]]
|
||
|
|
for (entry <- batch; (key, value) = entry) jmap.put(key, value.asJava)
|
||
|
|
client.batch_insert(keyspace, key, jmap, consistencyLevel)
|
||
|
|
}
|
||
|
|
|
||
|
|
def --(key: String, columnPath: ColumnPath, timestamp: Long): Unit =
|
||
|
|
--(key, columnPath, timestamp, consistencyLevel)
|
||
|
|
|
||
|
|
def --(key: String, columnPath: ColumnPath, timestamp: Long, consistencyLevel: Int): Unit =
|
||
|
|
client.remove(keyspace, key, columnPath, timestamp, consistencyLevel)
|
||
|
|
|
||
|
|
// ====================================
|
||
|
|
// ====== Java-style API names
|
||
|
|
// ====================================
|
||
|
|
}
|
||
|
|
|
||
|
|
class CassandraSessionPool[T <: TTransport](
|
||
|
|
space: String,
|
||
|
|
transportPool: Pool[T],
|
||
|
|
inputProtocol: Protocol,
|
||
|
|
outputProtocol: Protocol,
|
||
|
|
consistency: Int) extends Closeable with Logging {
|
||
|
|
|
||
|
|
def this(space: String, transportPool: Pool[T], ioProtocol: Protocol, consistency: Int) =
|
||
|
|
this (space, transportPool, ioProtocol, ioProtocol, consistency)
|
||
|
|
|
||
|
|
def newSession: CassandraSession = newSession(consistency)
|
||
|
|
|
||
|
|
def newSession(consistencyLevel: Int): CassandraSession = {
|
||
|
|
val socket = transportPool.borrowObject
|
||
|
|
val cassandraClient = new Cassandra.Client(inputProtocol(socket), outputProtocol(socket))
|
||
|
|
val cassandraSchema = cassandraClient.describe_keyspace(space)
|
||
|
|
new CassandraSession {
|
||
|
|
val keyspace = space
|
||
|
|
val client = cassandraClient
|
||
|
|
val obtainedAt = System.currentTimeMillis
|
||
|
|
val consistencyLevel = consistency
|
||
|
|
val schema = cassandraSchema
|
||
|
|
log.debug("Creating %s", toString)
|
||
|
|
|
||
|
|
def flush = socket.flush
|
||
|
|
def close = transportPool.returnObject(socket)
|
||
|
|
override def toString = "[CassandraSession]\n\tkeyspace = " + keyspace + "\n\tschema = " + schema
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
def withSession[T](body: CassandraSession => T) = {
|
||
|
|
val session = newSession(consistency)
|
||
|
|
try {
|
||
|
|
val result = body(session)
|
||
|
|
session.flush
|
||
|
|
result
|
||
|
|
} finally {
|
||
|
|
session.close
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
def close = transportPool.close
|
||
|
|
}
|
||
|
|
|
||
|
|
sealed abstract class Protocol(val factory: TProtocolFactory) {
|
||
|
|
def apply(transport: TTransport) = factory.getProtocol(transport)
|
||
|
|
}
|
||
|
|
|
||
|
|
object Protocol {
|
||
|
|
object Binary extends Protocol(new TBinaryProtocol.Factory)
|
||
|
|
object SimpleJSON extends Protocol(new TSimpleJSONProtocol.Factory)
|
||
|
|
object JSON extends Protocol(new TJSONProtocol.Factory)
|
||
|
|
}
|
||
|
|
</pre>
|
||
|
|
<script language='javascript'>
|
||
|
|
dp.SyntaxHighlighter.ClipboardSwf = './_highlighter/clipboard.swf';
|
||
|
|
dp.SyntaxHighlighter.HighlightAll('code');
|
||
|
|
</script>
|
||
|
|
</body>
|
||
|
|
</html>
|