brave.netty.http.TestHttpInitializer.java Source code

Introduction

Here is the source code for brave.netty.http.TestHttpInitializer.java

Source

/*
 * Copyright 2012 The Netty Project
 *
 * The Netty Project licenses this file to you 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 brave.netty.http;

import brave.http.HttpTracing;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpRequestDecoder;
import io.netty.handler.codec.http.HttpResponseEncoder;

class TestHttpInitializer extends ChannelInitializer<SocketChannel> {
    private HttpTracing httpTracing;

    public TestHttpInitializer(HttpTracing httpTracing) {
        this.httpTracing = httpTracing;
    }

    @Override
    public void initChannel(SocketChannel ch) throws Exception {
        ChannelPipeline p = ch.pipeline();

        p.addLast("decoder", new HttpRequestDecoder());
        p.addLast("encoder", new HttpResponseEncoder());
        p.addLast("aggregator", new HttpObjectAggregator(1048576));
        //add brave tracing
        p.addLast("braveResponse", NettyHttpTracing.create(httpTracing).channelOutboundHandler());
        p.addLast("braveRequest", NettyHttpTracing.create(httpTracing).channelInboundHandler());

        p.addLast("handler", new TestHttpHandler(httpTracing));

    }
}