io.hyperfoil.http.connection.Http2ConnectionHandlerBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hyperfoil-http Show documentation

There is a newer version: 0.27.1

package io.hyperfoil.http.connection;

import java.util.function.BiConsumer;

import io.hyperfoil.http.api.HttpClientPool;
import io.hyperfoil.http.api.HttpConnection;
import io.netty.handler.codec.http2.AbstractHttp2ConnectionHandlerBuilder;
import io.netty.handler.codec.http2.Http2ConnectionDecoder;
import io.netty.handler.codec.http2.Http2ConnectionEncoder;
import io.netty.handler.codec.http2.Http2EventAdapter;
import io.netty.handler.codec.http2.Http2Settings;

class Http2ConnectionHandlerBuilder extends AbstractHttp2ConnectionHandlerBuilder {

   private final HttpClientPool clientPool;
   private final boolean isUpgrade;
   private final BiConsumer requestHandler;

   Http2ConnectionHandlerBuilder(HttpClientPool clientPool, boolean isUpgrade, BiConsumer requestHandler) {
      this.clientPool = clientPool;
      this.isUpgrade = isUpgrade;
      this.requestHandler = requestHandler;
   }

   @Override
   protected CustomHttp2ConnectionHandler build(Http2ConnectionDecoder decoder, Http2ConnectionEncoder encoder, Http2Settings initialSettings) {
      return new CustomHttp2ConnectionHandler(clientPool, requestHandler, decoder, encoder, initialSettings, isUpgrade);
   }

   public CustomHttp2ConnectionHandler build(io.netty.handler.codec.http2.Http2Connection conn) {
      connection(conn);
      frameListener(new Http2EventAdapter() { /* Dunno why this is needed */});
      return super.build();
   }
}