com.twitter.storehaus.redis.RedisHashStore.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of storehaus-redis_2.9.3 Show documentation
storehaus-redis
The newest version!
/*
 * Copyright 2013 Twitter Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License. You may obtain
 * a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.twitter.storehaus.redis

import com.twitter.algebird.Monoid
import com.twitter.util.{ Duration, Future, Time }
import com.twitter.finagle.redis.Client
import com.twitter.storehaus.{ Store, UnpivotedStore }
import org.jboss.netty.buffer.{ ChannelBuffer, ChannelBuffers }

/**
 *
 *  @author Doug Tangren
 */

object RedisHashStore {

  def apply(client: Client, ttl: Option[Duration] = RedisStore.Default.TTL) =
    new RedisHashStore(client, ttl)

  def unpivoted(client: Client, ttl: Option[Duration] = RedisStore.Default.TTL) =
    new UnpivotedRedisHashStore(apply(client, ttl))
}
import RedisHashStore._

/**
 * A Store in which keys map to Maps of secondary keys and values backed
 * by a redis hash
 */
class RedisHashStore(val client: Client, ttl: Option[Duration])
  extends Store[ChannelBuffer, Map[ChannelBuffer, ChannelBuffer]] {

  override def get(k: ChannelBuffer): Future[Option[Map[ChannelBuffer, ChannelBuffer]]] =
    client.hGetAll(k).map({ case e if (e.isEmpty) => None case xs => Some(Map(xs:_*)) })

  protected def set(k: ChannelBuffer, v: Map[ChannelBuffer, ChannelBuffer]) = {
    ttl.map(exp => client.expire(k, exp.inSeconds))
    client.hMSet(k, v).unit
  }

  override def put(kv: (ChannelBuffer, Option[Map[ChannelBuffer, ChannelBuffer]])): Future[Unit] =
    kv match {
      case (key, Some(value)) => set(key, value)
      case (key, None) => client.del(Seq(key)).unit
    }

  override def close(t: Time) = client.quit.foreach { _ => client.release }
}

/*
 *  A Store in which K is a tuple of (Key, FieldKey) and V is
 *  the value of associated with FieldKey within the redis hash
 */
class UnpivotedRedisHashStore(hstore: RedisHashStore)
  extends UnpivotedStore[(ChannelBuffer, ChannelBuffer), ChannelBuffer, ChannelBuffer, ChannelBuffer](hstore)(identity)