com.ning.http.client.providers.netty.handler.Protocol Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of payment-retries-plugin Show documentation
Show all versions of payment-retries-plugin Show documentation
Kill Bill Payment Retries plugin
The newest version!
/*
* Copyright (c) 2014-2015 AsyncHttpClient Project. All rights reserved.
*
* This program is licensed to you under the Apache License Version 2.0,
* and you may not use this file except in compliance with the Apache License Version 2.0.
* You may obtain a copy of the Apache License Version 2.0 at
* http://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the Apache License Version 2.0 is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the Apache License Version 2.0 for the specific language governing permissions and limitations there under.
*/
package com.ning.http.client.providers.netty.handler;
import static com.ning.http.util.AsyncHttpProviderUtils.*;
import static org.jboss.netty.handler.codec.http.HttpHeaders.Names.*;
import static org.jboss.netty.handler.codec.http.HttpResponseStatus.*;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.handler.codec.http.HttpHeaders;
import org.jboss.netty.handler.codec.http.HttpResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.ning.http.client.AsyncHandler;
import com.ning.http.client.AsyncHttpClientConfig;
import com.ning.http.client.FluentCaseInsensitiveStringsMap;
import com.ning.http.client.HttpResponseHeaders;
import com.ning.http.client.HttpResponseStatus;
import com.ning.http.client.MaxRedirectException;
import com.ning.http.client.Realm;
import com.ning.http.client.Realm.AuthScheme;
import com.ning.http.client.Request;
import com.ning.http.client.RequestBuilder;
import com.ning.http.client.cookie.Cookie;
import com.ning.http.client.cookie.CookieDecoder;
import com.ning.http.client.filter.FilterContext;
import com.ning.http.client.filter.FilterException;
import com.ning.http.client.filter.ResponseFilter;
import com.ning.http.client.providers.netty.NettyAsyncHttpProviderConfig;
import com.ning.http.client.providers.netty.channel.ChannelManager;
import com.ning.http.client.providers.netty.future.NettyResponseFuture;
import com.ning.http.client.providers.netty.request.NettyRequestSender;
import com.ning.http.client.uri.Uri;
import com.ning.http.util.MiscUtils;
import java.io.IOException;
import java.util.HashSet;
import java.util.Set;
public abstract class Protocol {
protected final Logger logger = LoggerFactory.getLogger(getClass());
protected final ChannelManager channelManager;
protected final AsyncHttpClientConfig config;
protected final NettyAsyncHttpProviderConfig nettyConfig;
protected final NettyRequestSender requestSender;
private final boolean hasResponseFilters;
protected final boolean hasIOExceptionFilters;
private final MaxRedirectException maxRedirectException;
public static final Set REDIRECT_STATUSES = new HashSet<>();
static {
REDIRECT_STATUSES.add(MOVED_PERMANENTLY.getCode());
REDIRECT_STATUSES.add(FOUND.getCode());
REDIRECT_STATUSES.add(SEE_OTHER.getCode());
REDIRECT_STATUSES.add(TEMPORARY_REDIRECT.getCode());
}
public Protocol(ChannelManager channelManager, AsyncHttpClientConfig config, NettyAsyncHttpProviderConfig nettyConfig,
NettyRequestSender requestSender) {
this.channelManager = channelManager;
this.config = config;
this.nettyConfig = nettyConfig;
this.requestSender = requestSender;
hasResponseFilters = !config.getResponseFilters().isEmpty();
hasIOExceptionFilters = !config.getIOExceptionFilters().isEmpty();
maxRedirectException = new MaxRedirectException("Maximum redirect reached: " + config.getMaxRedirects());
}
public abstract void handle(Channel channel, NettyResponseFuture future, Object message) throws Exception;
public abstract void onError(NettyResponseFuture future, Throwable e);
public abstract void onClose(NettyResponseFuture future);
private FluentCaseInsensitiveStringsMap propagatedHeaders(Request request, Realm realm, boolean switchToGet) {
FluentCaseInsensitiveStringsMap headers = request.getHeaders()//
.delete(HttpHeaders.Names.HOST)//
.delete(HttpHeaders.Names.CONTENT_LENGTH)//
.delete(HttpHeaders.Names.CONTENT_TYPE);
if (realm != null && realm.getScheme() == AuthScheme.NTLM) {
headers.delete(AUTHORIZATION)//
.delete(PROXY_AUTHORIZATION);
}
return headers;
}
protected boolean exitAfterHandlingRedirect(//
final Channel channel,//
final NettyResponseFuture future,//
HttpResponse response,//
Request request,//
int statusCode,//
Realm realm) throws Exception {
if (followRedirect(config, request) && REDIRECT_STATUSES.contains(statusCode)) {
if (future.incrementAndGetCurrentRedirectCount() >= config.getMaxRedirects()) {
throw maxRedirectException;
} else {
// We must allow 401 handling again.
future.getAndSetAuth(false);
String originalMethod = request.getMethod();
boolean switchToGet = !originalMethod.equals("GET") && (statusCode == 303 || (statusCode == 302 && !config.isStrict302Handling()));
boolean keepBody = statusCode == 307 || (statusCode == 302 && config.isStrict302Handling());
final RequestBuilder requestBuilder = new RequestBuilder(switchToGet ? "GET" : originalMethod)//
.setCookies(request.getCookies())//
.setConnectionPoolKeyStrategy(request.getConnectionPoolPartitioning())//
.setFollowRedirects(true)//
.setLocalInetAddress(request.getLocalAddress())//
.setNameResolver(request.getNameResolver())//
.setProxyServer(request.getProxyServer())//
.setRealm(request.getRealm())//
.setRequestTimeout(request.getRequestTimeout())//
.setVirtualHost(request.getVirtualHost());
if (keepBody) {
requestBuilder.setBodyEncoding(request.getBodyEncoding());
if (MiscUtils.isNonEmpty(request.getFormParams()))
requestBuilder.setFormParams(request.getFormParams());
else if (request.getStringData() != null)
requestBuilder.setBody(request.getStringData());
else if (request.getByteData() != null)
requestBuilder.setBody(request.getByteData());
else if (request.getBodyGenerator() != null)
requestBuilder.setBody(request.getBodyGenerator());
}
requestBuilder.setHeaders(propagatedHeaders(request, realm, switchToGet));
// in case of a redirect from HTTP to HTTPS, future attributes might change
final boolean initialConnectionKeepAlive = future.isKeepAlive();
final Object initialPartitionKey = future.getPartitionKey();
HttpHeaders responseHeaders = response.headers();
String location = responseHeaders.get(HttpHeaders.Names.LOCATION);
Uri uri = Uri.create(future.getUri(), location);
future.setUri(uri);
String newUrl = uri.toUrl();
if (request.getUri().getScheme().startsWith(WEBSOCKET)) {
newUrl = newUrl.replaceFirst(HTTP, WEBSOCKET);
}
logger.debug("Redirecting to {}", newUrl);
for (String cookieStr : responseHeaders.getAll(HttpHeaders.Names.SET_COOKIE)) {
Cookie c = CookieDecoder.decode(cookieStr);
if (c != null)
requestBuilder.addOrReplaceCookie(c);
}
requestBuilder.setHeaders(propagatedHeaders(future.getRequest(), realm, switchToGet));
final Request nextRequest = requestBuilder.setUrl(newUrl).build();
logger.debug("Sending redirect to {}", request.getUri());
if (future.isKeepAlive() && !HttpHeaders.isTransferEncodingChunked(response) && !response.isChunked()) {
if (isSameHostAndProtocol(request.getUri(), nextRequest.getUri())) {
future.setReuseChannel(true);
} else {
channelManager.drainChannelAndOffer(channel, future, initialConnectionKeepAlive, initialPartitionKey);
}
} else {
// redirect + chunking = WAT
channelManager.closeChannel(channel);
}
requestSender.sendNextRequest(nextRequest, future);
return true;
}
}
return false;
}
@SuppressWarnings({ "rawtypes", "unchecked" })
protected boolean exitAfterProcessingFilters(//
Channel channel,//
NettyResponseFuture future,//
AsyncHandler handler, //
HttpResponseStatus status,//
HttpResponseHeaders responseHeaders) throws IOException {
if (hasResponseFilters) {
FilterContext fc = new FilterContext.FilterContextBuilder().asyncHandler(handler).request(future.getRequest())
.responseStatus(status).responseHeaders(responseHeaders).build();
for (ResponseFilter asyncFilter : config.getResponseFilters()) {
try {
fc = asyncFilter.filter(fc);
// FIXME Is it worth protecting against this?
if (fc == null) {
throw new NullPointerException("FilterContext is null");
}
} catch (FilterException efe) {
requestSender.abort(channel, future, efe);
}
}
// The handler may have been wrapped.
future.setAsyncHandler(fc.getAsyncHandler());
// The request has changed
if (fc.replayRequest()) {
requestSender.replayRequest(future, fc, channel);
return true;
}
}
return false;
}
}