com.twitter.finagle.http2.transport.common.H2StreamChannelInit.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finagle-http2_2.13 Show documentation

finagle-http2

There is a newer version: 24.2.0

package com.twitter.finagle.http2.transport.common

import com.twitter.finagle.Stack
import com.twitter.finagle.http2.transport.client.Http2ClientEventMapper
import com.twitter.finagle.http2.transport.server.H2UriValidatorHandler
import com.twitter.finagle.netty4.http
import com.twitter.finagle.netty4.param.Allocator
import io.netty.channel._
import io.netty.handler.codec.http2.Http2StreamFrameToHttpObjectCodec

/**
 * Initializes the child channels for HTTP/2 StreamChannels to a state where they look like HTTP/1.x pipelines
 */
private[http2] object H2StreamChannelInit {

  def initClient(params: Stack.Params): ChannelInitializer[Channel] =
    new H2Initializer(None, params, isServer = false)

  def initServer(
    init: ChannelInitializer[Channel],
    params: Stack.Params
  ): ChannelInitializer[Channel] =
    new H2Initializer(Some(init), params, isServer = true)

  private final class H2Initializer(
    init: Option[ChannelInitializer[Channel]],
    params: Stack.Params,
    isServer: Boolean)
      extends ChannelInitializer[Channel] {
    def initChannel(ch: Channel): Unit = {
      val alloc = params[Allocator].allocator
      ch.config.setAllocator(alloc)
      if (isServer) {
        ch.pipeline
          .addLast(new Http2NackHandler)
          .addLast(H2UriValidatorHandler.HandlerName, H2UriValidatorHandler)
      }

      ch.pipeline.addLast(
        new Http2StreamFrameToHttpObjectCodec(
          isServer,
          false /* validateHeaders */
        )
      )
      ch.pipeline.addLast(StripHeadersHandler.HandlerName, StripHeadersHandler)
      ch.pipeline.addLast(Http2StreamMessageHandler(isServer = isServer))

      if (isServer) {
        http.initServer(params)(ch.pipeline)
      } else {
        http.initClient(params)(ch.pipeline)
        ch.pipeline.addLast("event-mapper", Http2ClientEventMapper)
      }

      init match {
        case Some(init) => ch.pipeline.addLast(init)
        case None => () // nop
      }
    }
  }
}