blended.itestsupport.jms.ScheduledJMSProducer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of blended.itestsupport Show documentation

Define an integration test API for collaborating blended container(s) using docker as a runtime for the container(s) under test and an Akka based Camel framework to perform the integration tests as pure blackbox tests. Container(s) may be prestarted and discovered (for execution speed) or started by the integration test (for reproducability).

The newest version!

package blended.itestsupport.jms

import javax.jms.{ConnectionFactory, DeliveryMode, Message, Session}

import akka.actor.{Actor, ActorLogging, Props}
import blended.jms.utils.{JMSMessageFactory, JMSSupport}
import com.typesafe.config.Config
import scala.collection.JavaConverters._

import scala.concurrent.duration._

object ScheduledJMSProducer {

  def props(cf: ConnectionFactory, cfg: Config) = Props(new ScheduledJMSProducer(cf, cfg) with JMSMessageFactory[Long] {
    override def createMessage(session: Session, content: Long): Message = {
      val msg = session.createTextMessage(s"${content}")

      if (cfg.hasPath("properties")) {
        val props = cfg.getObject("properties").entrySet().asScala
        props.foreach { entry => msg.setStringProperty(entry.getKey(), cfg.getString("properties." + entry.getKey())) }
      }

      msg
    }
  })
}

class ScheduledJMSProducer(cf: ConnectionFactory, cfg: Config) extends Actor with ActorLogging with JMSSupport { this: JMSMessageFactory[Long] =>

  var counter : Long = 0

  implicit val eCtxt = context.system.dispatcher

  val schedule = cfg.getLong("interval")

  case object Tick

  override def preStart(): Unit = context.system.scheduler.scheduleOnce(schedule.millis, self, Tick)

  override def receive: Receive = {
    case Tick =>
      sendMessage(cf, cfg.getString("destination"), counter, this, DeliveryMode.PERSISTENT, 4, 0)
      context.system.scheduler.scheduleOnce(schedule.millis, self, Tick)
  }

}