io.gatling.jms.action.Send.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2011-2018 GatlingCorp (https://gatling.io)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.gatling.jms.action

import javax.jms.Message
import io.gatling.commons.stats.OK
import io.gatling.commons.util.Clock
import io.gatling.commons.validation._
import io.gatling.core.action._
import io.gatling.core.config.GatlingConfiguration
import io.gatling.core.session._
import io.gatling.core.stats.StatsEngine
import io.gatling.jms.client.JmsConnectionPool
import io.gatling.jms.protocol.JmsProtocol
import io.gatling.jms.request._

/**
 * Core JMS Action to handle Send
 *
 * This handles the core "send"ing of messages. Gatling calls the execute method to trigger a send.
 */
class Send(
    attributes:        JmsAttributes,
    protocol:          JmsProtocol,
    jmsConnectionPool: JmsConnectionPool,
    val statsEngine:   StatsEngine,
    val clock:         Clock,
    configuration:     GatlingConfiguration,
    val next:          Action
)
  extends JmsAction(attributes, protocol, jmsConnectionPool) {

  override val name: String = genName("jmsSend")

  override protected def beforeSend(requestName: String, session: Session): Validation[Message => Unit] =
    ((message: Message) => {
      val now = clock.nowMillis
      if (logger.underlying.isDebugEnabled) {
        logMessage(s"Message sent JMSMessageID=${message.getJMSMessageID}", message)
      }

      configuration.resolve(
        // [fl]
        //
        //
        //
        //
        // [fl]
        statsEngine.logResponse(session, requestName, now, now, OK, None, None)
      )
      next ! session
    }).success
}