/*
* Copyright (C) 2012-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 com.facebook.nifty.client;
import com.google.common.net.HostAndPort;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelPipeline;
import org.jboss.netty.channel.ChannelPipelineFactory;
import org.jboss.netty.channel.Channels;
import org.jboss.netty.handler.codec.http.HttpChunkAggregator;
import org.jboss.netty.handler.codec.http.HttpClientCodec;
import org.jboss.netty.util.Timer;
import java.net.InetSocketAddress;
import java.net.URI;
import java.net.URISyntaxException;
import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
public class HttpClientConnector extends AbstractClientConnector<HttpClientChannel>
{
private final URI endpointUri;
public HttpClientConnector(String hostNameAndPort, String servicePath)
throws URISyntaxException
{
super(new InetSocketAddress(HostAndPort.fromString(hostNameAndPort).getHostText(),
HostAndPort.fromString(hostNameAndPort).getPort()));
this.endpointUri = new URI("http", hostNameAndPort, servicePath, null);
}
public HttpClientConnector(URI uri)
{
super(HostAndPort.fromParts(checkNotNull(uri).getHost(), getPort(uri)));
checkArgument(uri.isAbsolute() && !uri.isOpaque(),
"HttpClientConnector requires an absolute URI with a path");
this.endpointUri = uri;
}
public static int getPort(URI uri)
{
URI uriNN = checkNotNull(uri);
if (uri.getScheme().toLowerCase().equals("http")) {
return uriNN.getPort() == -1 ? 80 : uriNN.getPort();
} else if (uri.getScheme().toLowerCase().equals("https")) {
return uriNN.getPort() == -1 ? 443 : uriNN.getPort();
} else {
throw new IllegalArgumentException("HttpClientConnector only connects to HTTP/HTTPS " +
"URIs");
}
}
@Override
public HttpClientChannel newThriftClientChannel(Channel nettyChannel, Timer timer)
{
HttpClientChannel channel =
new HttpClientChannel(nettyChannel,
timer,
endpointUri.getHost(),
endpointUri.getPath());
channel.getNettyChannel().getPipeline().addLast("thriftHandler", channel);
return channel;
}
@Override
public ChannelPipelineFactory newChannelPipelineFactory(final int maxFrameSize)
{
return new ChannelPipelineFactory()
{
@Override
public ChannelPipeline getPipeline()
throws Exception
{
ChannelPipeline cp = Channels.pipeline();
cp.addLast("httpClientCodec", new HttpClientCodec());
cp.addLast("chunkAggregator", new HttpChunkAggregator(maxFrameSize));
return cp;
}
};
}
@Override
public String toString()
{
return endpointUri.toString();
}
}