co.cask.tigon.sql.io.HttpRouterClientService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tigon-sql Show documentation
There is a newer version: 0.2.1
/*
 * Copyright © 2014 Cask Data, 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 co.cask.tigon.sql.io;

import com.google.common.collect.Maps;
import com.google.common.util.concurrent.AbstractIdleService;
import org.jboss.netty.bootstrap.ClientBootstrap;
import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelFactory;
import org.jboss.netty.channel.ChannelFuture;
import org.jboss.netty.channel.ChannelFutureListener;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.ChannelPipeline;
import org.jboss.netty.channel.ChannelPipelineFactory;
import org.jboss.netty.channel.Channels;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelHandler;
import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;

import java.net.InetSocketAddress;
import java.util.Map;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.Executors;

/**
 * HTTP Data Ingestion Router to TCP Server Routing via TCPClientService.
 */
//TODO: Remove this class. Wrap the DataSourceServer which exposes a write method and both the TCP Ingestion Server
//and the DataIngestion Router NettyHTTP Service can write to it directly.
public class HttpRouterClientService extends AbstractIdleService {
  private final Map serverMap;
  private final ConcurrentMap channelList = Maps.newConcurrentMap();
  private final ClientBootstrap clientBootstrap;

  public HttpRouterClientService(Map ingestionServerMap) {
    this.serverMap = ingestionServerMap;
    ChannelFactory factory = new NioClientSocketChannelFactory(Executors.newCachedThreadPool(),
                                                               Executors.newCachedThreadPool());
    this.clientBootstrap = new ClientBootstrap(factory);
  }

  @Override
  protected void startUp() throws Exception {
    setupClientPipeline();
    for (final Map.Entry entry : serverMap.entrySet()) {
      clientBootstrap.connect(entry.getValue()).addListener(new ChannelFutureListener() {
        @Override
        public void operationComplete(ChannelFuture future) throws Exception {
          channelList.put(entry.getKey(), future.getChannel());
        }
      });
    }
  }

  @Override
  protected void shutDown() throws Exception {
    for (Map.Entry clientChannel : channelList.entrySet()) {
      clientChannel.getValue().close();
    }
  }

  public boolean sendData(String channelName, ChannelBuffer data) {
    Channel client = channelList.get(channelName);
    if (client != null && client.isConnected()) {
      client.write(data);
      return true;
    }
    return false;
  }

  private void setupClientPipeline() {
    clientBootstrap.setOption("tcpNoDelay", true);
    clientBootstrap.setOption("keepAlive", true);
    clientBootstrap.setPipelineFactory(new ChannelPipelineFactory() {
      @Override
      public ChannelPipeline getPipeline() throws Exception {
        ChannelPipeline pipeline = Channels.pipeline();
        pipeline.addLast("passHandler", new PassHandler());
        return pipeline;
      }
    });
  }

  private static class PassHandler extends SimpleChannelHandler {

    @Override
    public void writeRequested(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
      super.writeRequested(ctx, e);
    }
  }
}