code.grnet.gr Git - aquarium/blob - src/main/scala/gr/grnet/aquarium/actor/service/user/UserActorsLRU.scala

   1 /*
   2  * Copyright 2011-2012 GRNET S.A. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or
   5  * without modification, are permitted provided that the following
   6  * conditions are met:
   7  *
   8  *   1. Redistributions of source code must retain the above
   9  *      copyright notice, this list of conditions and the following
  10  *      disclaimer.
  11  *
  12  *   2. Redistributions in binary form must reproduce the above
  13  *      copyright notice, this list of conditions and the following
  14  *      disclaimer in the documentation and/or other materials
  15  *      provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
  18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
  21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  24  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGE.
  29  *
  30  * The views and conclusions contained in the software and
  31  * documentation are those of the authors and should not be
  32  * interpreted as representing official policies, either expressed
  33  * or implied, of GRNET S.A.
  34  */
  35
  36 package gr.grnet.aquarium.actor
  37 package service
  38 package user
  39
  40 import org.apache.solr.util.ConcurrentLRUCache
  41 import akka.actor.ActorRef
  42 import gr.grnet.aquarium.util.{Loggable, Lifecycle}
  43 import collection.JavaConversions._
  44
  45 /**
  46  * This class holds an LRU cache for the user actors.
  47  *
  48  * The underlying implementation is borrowed from the Apache lucene+solr project(s).
  49  *
  50  * The provided collections-like API is neither Java- nor Scala-oriented.
  51  *
  52  * @author Christos KK Loverdos <loverdos@gmail.com>
  53  */
  54
  55 class UserActorsLRU(val upperWaterMark: Int, val lowerWatermark: Int) extends Lifecycle {
  56   private[this] val _cache = new ConcurrentLRUCache[String, ActorRef](
  57     upperWaterMark,
  58     lowerWatermark,
  59     ((upperWaterMark + lowerWatermark).toLong / 2).toInt,
  60     (3L * upperWaterMark / 4).toInt,
  61     true,
  62     false,
  63     EvictionListener)
  64
  65   def put(userId: String, userActor: ActorRef): Unit = {
  66     _cache.put(userId, userActor)
  67   }
  68
  69   def get(userId: String): Option[ActorRef] = {
  70     _cache.get(userId) match {
  71       case null ⇒ None
  72       case actorRef ⇒ Some(actorRef)
  73     }
  74   }
  75
  76   def shutdownAll() = {
  77     val accessed = mapAsScalaMap(_cache.getLatestAccessedItems(_cache.size()))
  78
  79     //Send the poison pill and make sure that all futures have been returned
  80     accessed.keysIterator.map {
  81       x =>
  82         UserActorSupervisor.supervisor.unlink(_cache.get(x))
  83         _cache.get(x).stop()
  84     }
  85   }
  86
  87   def size: Int = _cache.size()
  88
  89   def clear: Unit = _cache.clear()
  90
  91   def start() = {}
  92
  93   def stop() = {
  94     _cache.destroy()
  95   }
  96
  97   private[this] object EvictionListener extends ConcurrentLRUCache.EvictionListener[String, ActorRef] with Loggable {
  98     def evictedEntry(userId: String, userActor: ActorRef): Unit = {
  99       logger.debug("Parking UserActor for userId = %s".format(userId))
 100       UserActorSupervisor.supervisor.unlink(userActor)
 101       // Check this is received after any currently servicing business logic message.
 102       userActor.stop()
 103       // Hopefully no need to further track these actors as they will now cause their own death.
 104     }
 105   }
 106
 107 }
 108