akka.dumbdi.ActorModule.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of akka-dumbdi_2.11 Show documentation

akka-dumbdi

There is a newer version: 0.0.2

package akka.dumbdi

import akka.actor.Actor

import scala.reflect.ClassTag
import scala.util.{ Failure, Success, Try }

trait ActorModule { self ⇒
  protected val state = collection.mutable.HashMap.empty[Class[_], Any]

  final def get[A](implicit ct: ClassTag[A]): A = {
    state(ct.runtimeClass).asInstanceOf[A]
  }

  final def ++(other: ActorModule): ActorModule = new ActorModuleConfigurable {
    for (x ← self.state) bindClassOf(x._1, x._2)
    for (x ← other.state) bindClassOf(x._1, x._2) // overwrites
  }

  override def toString: String = state.mkString(",")
}

object EmptyModule extends ActorModule {}

trait ActorModuleConfigurable extends ActorModule {
  def bind[A](instance: A)(implicit ct: ClassTag[A]) =
    state(ct.runtimeClass) = instance

  def bindClassOf(clazz: Class[_], to: Any) =
    state(clazz) = to
}

trait ActorWithModule extends ActorWithNamedModule {
  self: Actor ⇒
  final protected def moduleConfigLocation: String = "/" + context.self.path.elements.drop(1) // drop the parent to look-a-like deployment part
    .mkString("/")
}

trait ActorWithNamedModule {
  self: Actor ⇒
  protected def moduleConfigLocation: String

  private val fromConfigModule: ActorModule = {
    Try {
      context.system.settings.config.getString(s"akka.actor.dependency.${moduleConfigLocation}")
    } match {
      case Success(classname) ⇒ // path was found - if module does not exist, crash early
        Class.forName(classname).newInstance().asInstanceOf[ActorModule]
      case Failure(_) ⇒ // path was not found - return empty module
        EmptyModule
    }
  }

  private val initModule: ActorModuleConfigurable = new ActorModuleConfigurable {}

  protected def initialize(module: ActorModuleConfigurable): Unit
  initialize(initModule)

  lazy protected val module: ActorModule = initModule ++ fromConfigModule
}