com.thenewmotion.akka.rabbitmq.package.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of akka-rabbitmq_2.10 Show documentation

akka-rabbitmq

The newest version!

package com.thenewmotion.akka

import akka.actor.{ ActorRef, Props }
import akka.util.Timeout
import com.rabbitmq.{ client => rabbit }

import scala.concurrent.Await
import scala.concurrent.duration._

/**
 * @author Yaroslav Klymko
 */
package object rabbitmq {
  type Connection = rabbit.Connection
  type Channel = rabbit.Channel
  type ConnectionFactory = rabbit.ConnectionFactory
  type BasicProperties = rabbit.AMQP.BasicProperties
  type Envelope = rabbit.Envelope
  type DefaultConsumer = rabbit.DefaultConsumer

  type OnChannel = Channel => Any

  case class CreateChannel(props: Props, name: Option[String] = None)
  case class ChannelCreated(channel: ActorRef)

  case class ChannelMessage(onChannel: OnChannel, dropIfNoChannel: Boolean = true)

  implicit class RichConnectionFactory(val self: ConnectionFactory) extends AnyVal {
    def uri: String = "amqp://%s@%s:%d/%s".format(self.getUsername, self.getHost, self.getPort, self.getVirtualHost)
  }

  implicit class RichConnectionActor(val self: ActorRef) extends AnyVal {
    def createChannel(props: Props, name: Option[String] = None)(implicit timeout: Timeout = Timeout(2.seconds)): ActorRef = {
      import akka.pattern.ask
      val future = self ? CreateChannel(props, name)
      Await.result(future, timeout.duration).asInstanceOf[ChannelCreated].channel
    }
  }
}