io.edurt.datacap.pinot.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 datacap-shaded-pinot Show documentation
Show all versions of datacap-shaded-pinot Show documentation
DataCap - Shaded for Apache Pinot
/*
* Copyright (c) 2014 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 io.edurt.datacap.pinot.com.ning.http.client.providers.netty.handler;
import static io.edurt.datacap.pinot.com.ning.http.client.providers.netty.util.HttpUtils.HTTP;
import static io.edurt.datacap.pinot.com.ning.http.client.providers.netty.util.HttpUtils.WEBSOCKET;
import static io.edurt.datacap.pinot.com.ning.http.util.AsyncHttpProviderUtils.*;
import static io.edurt.datacap.pinot.org.jboss.netty.handler.codec.http.HttpHeaders.Names.*;
import static io.edurt.datacap.pinot.org.jboss.netty.handler.codec.http.HttpResponseStatus.*;
import io.edurt.datacap.pinot.org.jboss.netty.channel.Channel;
import io.edurt.datacap.pinot.org.jboss.netty.handler.codec.http.HttpHeaders;
import io.edurt.datacap.pinot.org.jboss.netty.handler.codec.http.HttpResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import io.edurt.datacap.pinot.com.ning.http.client.AsyncHandler;
import io.edurt.datacap.pinot.com.ning.http.client.AsyncHttpClientConfig;
import io.edurt.datacap.pinot.com.ning.http.client.FluentCaseInsensitiveStringsMap;
import io.edurt.datacap.pinot.com.ning.http.client.HttpResponseHeaders;
import io.edurt.datacap.pinot.com.ning.http.client.HttpResponseStatus;
import io.edurt.datacap.pinot.com.ning.http.client.MaxRedirectException;
import io.edurt.datacap.pinot.com.ning.http.client.Realm;
import io.edurt.datacap.pinot.com.ning.http.client.Realm.AuthScheme;
import io.edurt.datacap.pinot.com.ning.http.client.Request;
import io.edurt.datacap.pinot.com.ning.http.client.RequestBuilder;
import io.edurt.datacap.pinot.com.ning.http.client.cookie.Cookie;
import io.edurt.datacap.pinot.com.ning.http.client.cookie.CookieDecoder;
import io.edurt.datacap.pinot.com.ning.http.client.filter.FilterContext;
import io.edurt.datacap.pinot.com.ning.http.client.filter.FilterException;
import io.edurt.datacap.pinot.com.ning.http.client.filter.ResponseFilter;
import io.edurt.datacap.pinot.com.ning.http.client.providers.netty.NettyAsyncHttpProviderConfig;
import io.edurt.datacap.pinot.com.ning.http.client.providers.netty.channel.ChannelManager;
import io.edurt.datacap.pinot.com.ning.http.client.providers.netty.future.NettyResponseFuture;
import io.edurt.datacap.pinot.com.ning.http.client.providers.netty.request.NettyRequestSender;
import io.edurt.datacap.pinot.com.ning.http.client.uri.Uri;
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 originalHeaders = request.getHeaders();
originalHeaders.remove(HOST);
originalHeaders.remove(CONTENT_LENGTH);
if (switchToGet)
originalHeaders.remove(CONTENT_TYPE);
if (realm != null && realm.getScheme() == AuthScheme.NTLM) {
originalHeaders.remove(AUTHORIZATION);
originalHeaders.remove(PROXY_AUTHORIZATION);
}
return originalHeaders;
}
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);
HttpHeaders responseHeaders = response.headers();
String location = responseHeaders.get(HttpHeaders.Names.LOCATION);
Uri uri = Uri.create(future.getUri(), location);
if (!uri.equals(future.getUri())) {
final RequestBuilder requestBuilder = new RequestBuilder(future.getRequest());
// if we are to strictly handle 302, we should keep the original method (which browsers don't)
// 303 must force GET
boolean switchToGet = !request.getMethod().equals("GET") && (statusCode == 303 || (statusCode == 302 && !config.isStrict302Handling()));
if (switchToGet)
requestBuilder.setMethod("GET");
// in case of a redirect from HTTP to HTTPS, future attributes might change
final boolean initialConnectionKeepAlive = future.isKeepAlive();
final String initialPartition = future.getPartition();
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, initialPartition);
}
} 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;
}
}