• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• gatling-amqp-master
  • src
    • main
      • scala
        • io
          • gatling
            • amqp
              • AmqpDsl.scala
              • check
                • AmqpCheckSupport.scala
              • package.scala
              • Predef.scala
              • request
                • builder
                  • Publishing.scala
                  • Consuming.scala
                  • AmqpRequestBuilder.scala
                  • Amqp.scala
              • config
                • AmqpVariables.scala
                • AmqpRunner.scala
                • AmqpProtocol.scala
                • AmqpProtocolBuilder.scala
                • AmqpPreparation.scala
                • AmqpTermination.scala
              • event
                • AmqpEvent.scala
                • AmqpEventBus.scala
              • data
                • Exchange.scala
                • BindQueue.scala
                • PublishRequest.scala
                • package.scala
                • PublishAcked.scala
                • AmqpChannelCommand.scala
                • Message.scala
                • AmqpExchange.scala
                • WaitTermination.scala
                • Connection.scala
                • AmqpQueue.scala
                • AwaitTermination.scala
                • DeclareQueue.scala
                • AmqpRequest.scala
                • PublishNacked.scala
                • DeclareExchange.scala
                • ConsumeRequest.scala
              • action
                • AmqpConsumeAction.scala
                • AmqpActionBuilder.scala
                • AmqpPublishAction.scala
              • infra
                • AmqpTracer.scala
                • AmqpConsumer.scala
                • Stats.scala
                • AmqpPublisher.scala
                • AmqpRouter.scala
                • AmqpNacker.scala
                • AmqpManage.scala
                • Logging.scala
                • AmqpActor.scala
    • test
      • scala
        • io
          • gatling
            • amqp
              • PublishingSimulation.scala
              • ConsumingSimulation.scala
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • .travis.yml
  • README.md
  • build.sbt
  • run
  • .gitignore

package io.gatling.amqp.config

import akka.pattern.ask
import akka.util.Timeout
import io.gatling.amqp.data._
import io.gatling.core.session.Session
import pl.project13.scala.rainbow._

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

/**
 * termination for AMQP
 */
trait AmqpTermination { this: AmqpProtocol =>
  private val publishTimeout: Timeout = Timeout(1 hour)
  private val consumeTimeout: Timeout = Timeout(1 hour)

  protected def awaitTerminationFor(session: Session): Unit = {
    // wait nacker to ensure all confirms has been fired
    Await.result((nacker ask WaitTermination(session))(publishTimeout), Duration.Inf) match {
      case Success(m) => logger.debug(s"amqp: $m".green)
      case Failure(e) => throw e
    }

    // wait consumers
    Await.result((router ask WaitTermination(session))(consumeTimeout), Duration.Inf) match {
      case Success(m) => logger.debug(s"amqp: $m".green)
      case Failure(e) => throw e
    }
  }
}