/*
* 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.base.Preconditions;
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 javax.validation.constraints.NotNull;
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(), checkNotNull(uri).getPort()));
checkArgument(uri.isAbsolute() && !uri.isOpaque(),
"HttpClientConnector requires an absolute URI with a path");
checkArgument(uri.getScheme().compareTo("http") != 0 &&
uri.getScheme().compareTo("https") != 0,
"HttpClientConnector only connects to HTTP/HTTPS URIs");
this.endpointUri = uri;
}
@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();
}
}