GenericActionBuilder.scala

class GenericActionBuilder[V <: Message](
  attributes: Attributes[V],
  requestActionBuilder: RequestActionBuilder[V],
  responseActionBuilder: ResponseActionBuilder
) extends ActionBuilder with LazyLogging {

  override def build(ctx: ScenarioContext, next: Action): Action = {
    import ctx._

    val requestProtocolComponents = protocolComponentsRegistry.components(requestActionBuilder.key)
    val responseProtocolComponents = protocolComponentsRegistry.components(responseActionBuilder.key)

    val publisher = requestActionBuilder.build(ctx, requestProtocolComponents)
    logger.debug(s"Publisher ${publisher.name} has been instantiated")

    val consumer = responseActionBuilder.build(ctx, responseProtocolComponents)
    logger.debug(s"Consumer ${consumer.name} has been instantiated")

    system.registerOnTermination {
      safe(publisher.close())
      safe(consumer.close())
    }

    new GenericAction[V](
      coreComponents,
      attributes,
      publisher,
      consumer,
      throttled,
      next
    )
  }

  private def safe(f: => Unit): Unit = {
    var r: Unit = () //to prevent JIT from skipping execution of f
    try {
      r = f
    } catch {case NonFatal(e) => e.printStackTrace()}
  }
}

case class Attributes[V](
  requestName: Expression[String],
  payload: Expression[V]
)