raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Voldemort Tests now working as well as Riak

Browse source
This commit is contained in:
ticktock 2010-10-21 17:53:44 -04:00
parent 68d832e97b
commit c00aef3591
1 changed files with 26 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

38
akka-persistence/akka-persistence-voldemort/src/main/scala/VoldemortStorageBackend.scala
View file

@ -22,8 +22,10 @@ import voldemort.client.protocol.admin.{AdminClientConfig, AdminClient}
*/
private[akka] object VoldemortStorageBackend extends KVStorageBackend {
import KVAccess._
import VoldemortAccess._
val bootstrapUrlsProp = "bootstrap_urls"
val clientConfig = config.getConfigMap("akka.storage.voldemort.client") match {
case Some(configMap) => getClientConfig(configMap.asMap)
@ -35,12 +37,12 @@ private[akka] object VoldemortStorageBackend extends KVStorageBackend {
val queueStore = config.getString("akka.storage.voldemort.store.queue", "Queues")
var storeClientFactory: StoreClientFactory = initStoreClientFactory
var storeClientFactory: StoreClientFactory = null
var refs: KVAccess = null
var maps: KVAccess = null
var vectors: KVAccess = null
var queues: KVAccess = null
initKVAccess
resetAccess
def refAccess = refs
@ -52,11 +54,13 @@ private[akka] object VoldemortStorageBackend extends KVStorageBackend {
object VoldemortAccess {
val admin = new AdminClient(VoldemortStorageBackend.clientConfig.getProperty(VoldemortStorageBackend.bootstrapUrlsProp), new AdminClientConfig)
var admin: AdminClient = null
}
class VoldemortAccess(val store: String) extends KVAccess {
import VoldemortAccess._
val client: StoreClient[Array[Byte], Array[Byte]] = VoldemortStorageBackend.storeClientFactory.getStoreClient(store)
def put(key: Array[Byte], value: Array[Byte]) = {
@ -72,7 +76,7 @@ private[akka] object VoldemortStorageBackend extends KVStorageBackend {
}
def getAll(keys: Iterable[Array[Byte]]): Map[Array[Byte], Array[Byte]] = {
JavaConversions.asMap(client.getAll(JavaConversions.asIterable(keys))).map {
JavaConversions.asMap(client.getAll(JavaConversions.asIterable(keys))).map{
kv =>
kv match {
case (key: Array[Byte], versioned: Versioned[Array[Byte]]) => (key -> versioned.getValue)
@ -92,7 +96,7 @@ private[akka] object VoldemortStorageBackend extends KVStorageBackend {
def getClientConfig(configMap: Map[String, String]): Properties = {
val properites = new Properties
configMap.foreach {
configMap.foreach{
keyval => keyval match {
case (key, value) => properites.setProperty(key.asInstanceOf[java.lang.String], value.asInstanceOf[java.lang.String])
}
@ -114,15 +118,25 @@ private[akka] object VoldemortStorageBackend extends KVStorageBackend {
}
}
def initKVAccess = {
refs = new VoldemortAccess(refStore)
maps = new VoldemortAccess(mapStore)
vectors = new VoldemortAccess(vectorStore)
queues = new VoldemortAccess(queueStore)
def initAdminClient(): AdminClient = {
if (VoldemortAccess.admin ne null) {
VoldemortAccess.admin.stop
}
new AdminClient(VoldemortStorageBackend.clientConfig.getProperty(VoldemortStorageBackend.bootstrapUrlsProp), new AdminClientConfig)
}
def resetAccess(){
initStoreClientFactory
def initKVAccess = {
refs = new VoldemortAccess(refStore)
maps = new VoldemortAccess(mapStore)
vectors = new VoldemortAccess(vectorStore)
queues = new VoldemortAccess(queueStore)
}
def resetAccess() {
storeClientFactory = initStoreClientFactory
VoldemortAccess.admin = initAdminClient
initKVAccess
}