io.airlift.drift.transport.netty.ThriftClientInitializer.java Source code

Introduction

Here is the source code for io.airlift.drift.transport.netty.ThriftClientInitializer.java

Source

/*
 * Copyright (C) 2013 Facebook, Inc.
 *
 * 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.airlift.drift.transport.netty;

import com.google.common.net.HostAndPort;
import io.airlift.units.Duration;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.proxy.Socks4ProxyHandler;
import io.netty.handler.ssl.SslContext;

import java.net.InetSocketAddress;
import java.util.Optional;

class ThriftClientInitializer extends ChannelInitializer<SocketChannel> {
    private final MessageFraming messageFraming;
    private final Duration requestTimeout;
    private final MessageEncoding messageEncoding;
    private final Optional<HostAndPort> socksProxyAddress;
    private final Optional<SslContext> sslContext;

    public ThriftClientInitializer(MessageFraming messageFraming, MessageEncoding messageEncoding,
            Duration requestTimeout, Optional<HostAndPort> socksProxyAddress, Optional<SslContext> sslContext) {
        this.messageFraming = messageFraming;
        this.requestTimeout = requestTimeout;
        this.messageEncoding = messageEncoding;
        this.socksProxyAddress = socksProxyAddress;
        this.sslContext = sslContext;
    }

    @Override
    protected void initChannel(SocketChannel channel) {
        ChannelPipeline pipeline = channel.pipeline();

        messageFraming.addFrameHandlers(pipeline);

        pipeline.addLast(new ThriftClientHandler(requestTimeout, messageEncoding));

        sslContext.ifPresent(sslContext -> pipeline.addFirst(sslContext.newHandler(channel.alloc())));

        socksProxyAddress.ifPresent(socks -> pipeline
                .addFirst(new Socks4ProxyHandler(new InetSocketAddress(socks.getHost(), socks.getPort()))));
    }
}