com.linkedin.mitm.proxy.ProxyInitializer.java Source code

Introduction

Here is the source code for com.linkedin.mitm.proxy.ProxyInitializer.java

Source

/*
 * Copyright (c) LinkedIn Corporation. All rights reserved. Licensed under the BSD-2 Clause license.
 * See LICENSE in the project root for license information.
 */

package com.linkedin.mitm.proxy;

import com.linkedin.mitm.proxy.channel.ChannelMediator;
import com.linkedin.mitm.proxy.channel.ClientChannelHandler;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpRequestDecoder;
import io.netty.handler.codec.http.HttpResponseEncoder;
import io.netty.handler.timeout.IdleStateHandler;

/**
 * Initialize client to proxy channel and channel pipeline.
 *
 * @author shfeng
 */
public class ProxyInitializer extends ChannelInitializer<SocketChannel> {
    private final ProxyServer _proxyServer;

    public ProxyInitializer(ProxyServer proxyServer) {
        _proxyServer = proxyServer;
    }

    @Override
    public void initChannel(SocketChannel socketChannel) {
        socketChannel.pipeline().addLast("decoder", new HttpRequestDecoder())
                .addLast("encoder", new HttpResponseEncoder())
                .addLast("idle", new IdleStateHandler(0, 0, _proxyServer.getClientConnectionIdleTimeout()))
                .addLast("handler", new ClientChannelHandler(
                        new ChannelMediator(socketChannel, _proxyServer.getProxyModeControllerFactory(),
                                _proxyServer.getDownstreamWorkerGroup(),
                                _proxyServer.getServerConnectionIdleTimeout(), _proxyServer.getAllChannels()),
                        _proxyServer.getConnectionFlowRegistry()));
    }
}