Akka Pub/Sub Extensions

Akka PubSub Extensions.md

Simple helper classes that allow Akka Actors to publish and subscribe to a DistributedPubSub system by automatically determining the "topic" name from the class name.

Publisher.scala

import akka.actor.Actor
import akka.cluster.pubsub.DistributedPubSub
import akka.cluster.pubsub.DistributedPubSubMediator.Publish
import scala.reflect.runtime.universe._

/** Actor that can publish data  */
trait Publisher extends Actor {

  //PubSub for the current actor system
  private val pubsub = DistributedPubSub(context.system).mediator

  /**
    * Send data for the given type of data
    *
    * @param msg Message to send
    * @tparam A Type of message, used as the message topic
    */
  protected def publish[A: TypeTag](msg: A) = pubsub ! Publish(typeOf[A].getClass.getName, msg)
}

Subscriber.scala

import akka.actor.Actor
import akka.cluster.pubsub.DistributedPubSub
import akka.cluster.pubsub.DistributedPubSubMediator.Subscribe
import scala.reflect.runtime.universe._

/** Actor that can subscribe to the pipes */
trait Subscriber extends Actor {

  //PubSub for the current actor system
  private lazy val pubsub = DistributedPubSub(context.system).mediator

  /**
    * Subscribe for the given type of data
    *
    * @tparam A Type that you are interested in receiving
    */
  protected def subscribe[A]()(implicit tag: TypeTag[A]): Unit = pubsub ! Subscribe(typeOf[A].getClass.getName, self)
}