io.gatling.recorder.http.Netty.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gatling-recorder Show documentation

gatling-recorder

There is a newer version: 3.13.1

/*
 * Copyright 2011-2024 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.recorder.http

import scala.util.{ Failure, Success, Try }

import io.gatling.http.client.uri.Uri
import io.gatling.recorder.util.HttpUtils

import io.netty.channel.{ Channel, ChannelFuture, ChannelFutureListener }
import io.netty.handler.codec.http._

object Netty {
  implicit class PimpedChannelFuture(val cf: ChannelFuture) extends AnyVal {
    def addScalaListener(f: Try[Channel] => Unit): ChannelFuture =
      cf.addListener { (future: ChannelFuture) =>
        val outcome =
          if (future.isSuccess) {
            Success(future.channel)
          } else {
            Failure(future.cause)
          }
        f(outcome)
      }
  }

  implicit class PimpedChannel(val channel: Channel) extends AnyVal {
    def reply500AndClose(): Unit =
      channel
        .writeAndFlush(new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.INTERNAL_SERVER_ERROR))
        .addListener(ChannelFutureListener.CLOSE)
  }

  implicit class PimpedFullHttpRequest(val request: FullHttpRequest) extends AnyVal {
    def makeRelative: FullHttpRequest = {
      val relativeUrl = Uri.create(request.uri).toRelativeUrl
      val relativeRequest = new DefaultFullHttpRequest(request.protocolVersion, request.method, relativeUrl, request.content)
      relativeRequest.headers.add(request.headers)
      relativeRequest
    }

    def filterSupportedEncodings: FullHttpRequest = {
      Option(request.headers.get(HttpHeaderNames.ACCEPT_ENCODING))
        .foreach { acceptEncodingValue =>
          request.headers.set(HttpHeaderNames.ACCEPT_ENCODING, HttpUtils.filterSupportedEncodings(acceptEncodingValue))
        }

      request
    }
  }
}