org.elasticsearch.http.netty4.Netty4HttpRequest Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of transport-netty4 Show documentation
Show all versions of transport-netty4 Show documentation
Elasticsearch module: transport-netty4
/*
* Licensed to Elasticsearch under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch licenses this file to you 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 org.elasticsearch.http.netty4;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.handler.codec.http.DefaultFullHttpRequest;
import io.netty.handler.codec.http.DefaultHttpHeaders;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpHeaderNames;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.cookie.Cookie;
import io.netty.handler.codec.http.cookie.ServerCookieDecoder;
import io.netty.handler.codec.http.cookie.ServerCookieEncoder;
import org.elasticsearch.common.bytes.BytesReference;
import org.elasticsearch.http.HttpRequest;
import org.elasticsearch.rest.RestRequest;
import org.elasticsearch.rest.RestStatus;
import org.elasticsearch.transport.netty4.Netty4Utils;
import java.util.AbstractMap;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;
public class Netty4HttpRequest implements HttpRequest {
private final FullHttpRequest request;
private final BytesReference content;
private final HttpHeadersMap headers;
private final AtomicBoolean released;
private final Exception inboundException;
private final boolean pooled;
Netty4HttpRequest(FullHttpRequest request) {
this(request, new HttpHeadersMap(request.headers()), new AtomicBoolean(false), true,
Netty4Utils.toBytesReference(request.content()));
}
Netty4HttpRequest(FullHttpRequest request, Exception inboundException) {
this(request, new HttpHeadersMap(request.headers()), new AtomicBoolean(false), true,
Netty4Utils.toBytesReference(request.content()), inboundException);
}
private Netty4HttpRequest(FullHttpRequest request, HttpHeadersMap headers, AtomicBoolean released, boolean pooled,
BytesReference content) {
this(request, headers, released, pooled, content, null);
}
private Netty4HttpRequest(FullHttpRequest request, HttpHeadersMap headers, AtomicBoolean released, boolean pooled,
BytesReference content, Exception inboundException) {
this.request = request;
this.headers = headers;
this.content = content;
this.pooled = pooled;
this.released = released;
this.inboundException = inboundException;
}
@Override
public RestRequest.Method method() {
HttpMethod httpMethod = request.method();
if (httpMethod == HttpMethod.GET)
return RestRequest.Method.GET;
if (httpMethod == HttpMethod.POST)
return RestRequest.Method.POST;
if (httpMethod == HttpMethod.PUT)
return RestRequest.Method.PUT;
if (httpMethod == HttpMethod.DELETE)
return RestRequest.Method.DELETE;
if (httpMethod == HttpMethod.HEAD) {
return RestRequest.Method.HEAD;
}
if (httpMethod == HttpMethod.OPTIONS) {
return RestRequest.Method.OPTIONS;
}
if (httpMethod == HttpMethod.PATCH) {
return RestRequest.Method.PATCH;
}
if (httpMethod == HttpMethod.TRACE) {
return RestRequest.Method.TRACE;
}
if (httpMethod == HttpMethod.CONNECT) {
return RestRequest.Method.CONNECT;
}
throw new IllegalArgumentException("Unexpected http method: " + httpMethod);
}
@Override
public String uri() {
return request.uri();
}
@Override
public BytesReference content() {
assert released.get() == false;
return content;
}
@Override
public void release() {
if (pooled && released.compareAndSet(false, true)) {
request.release();
}
}
@Override
public HttpRequest releaseAndCopy() {
assert released.get() == false;
if (pooled == false) {
return this;
}
try {
final ByteBuf copiedContent = Unpooled.copiedBuffer(request.content());
return new Netty4HttpRequest(
new DefaultFullHttpRequest(request.protocolVersion(), request.method(), request.uri(), copiedContent, request.headers(),
request.trailingHeaders()),
headers, new AtomicBoolean(false), false, Netty4Utils.toBytesReference(copiedContent));
} finally {
release();
}
}
@Override
public final Map> getHeaders() {
return headers;
}
@Override
public List strictCookies() {
String cookieString = request.headers().get(HttpHeaderNames.COOKIE);
if (cookieString != null) {
Set cookies = ServerCookieDecoder.STRICT.decode(cookieString);
if (!cookies.isEmpty()) {
return ServerCookieEncoder.STRICT.encode(cookies);
}
}
return Collections.emptyList();
}
@Override
public HttpVersion protocolVersion() {
if (request.protocolVersion().equals(io.netty.handler.codec.http.HttpVersion.HTTP_1_0)) {
return HttpRequest.HttpVersion.HTTP_1_0;
} else if (request.protocolVersion().equals(io.netty.handler.codec.http.HttpVersion.HTTP_1_1)) {
return HttpRequest.HttpVersion.HTTP_1_1;
} else {
throw new IllegalArgumentException("Unexpected http protocol version: " + request.protocolVersion());
}
}
@Override
public HttpRequest removeHeader(String header) {
HttpHeaders headersWithoutContentTypeHeader = new DefaultHttpHeaders();
headersWithoutContentTypeHeader.add(request.headers());
headersWithoutContentTypeHeader.remove(header);
HttpHeaders trailingHeaders = new DefaultHttpHeaders();
trailingHeaders.add(request.trailingHeaders());
trailingHeaders.remove(header);
FullHttpRequest requestWithoutHeader = new DefaultFullHttpRequest(request.protocolVersion(), request.method(), request.uri(),
request.content(), headersWithoutContentTypeHeader, trailingHeaders);
return new Netty4HttpRequest(requestWithoutHeader, new HttpHeadersMap(requestWithoutHeader.headers()), released,
pooled, content);
}
@Override
public Netty4HttpResponse createResponse(RestStatus status, BytesReference content) {
return new Netty4HttpResponse(request.headers(), request.protocolVersion(), status, content);
}
@Override
public Exception getInboundException() {
return inboundException;
}
public FullHttpRequest nettyRequest() {
return request;
}
/**
* A wrapper of {@link HttpHeaders} that implements a map to prevent copying unnecessarily. This class does not support modifications
* and due to the underlying implementation, it performs case insensitive lookups of key to values.
*
* It is important to note that this implementation does have some downsides in that each invocation of the
* {@link #values()} and {@link #entrySet()} methods will perform a copy of the values in the HttpHeaders rather than returning a
* view of the underlying values.
*/
private static class HttpHeadersMap implements Map> {
private final HttpHeaders httpHeaders;
private HttpHeadersMap(HttpHeaders httpHeaders) {
this.httpHeaders = httpHeaders;
}
@Override
public int size() {
return httpHeaders.size();
}
@Override
public boolean isEmpty() {
return httpHeaders.isEmpty();
}
@Override
public boolean containsKey(Object key) {
return key instanceof String && httpHeaders.contains((String) key);
}
@Override
public boolean containsValue(Object value) {
return value instanceof List && httpHeaders.names().stream().map(httpHeaders::getAll).anyMatch(value::equals);
}
@Override
public List get(Object key) {
return key instanceof String ? httpHeaders.getAll((String) key) : null;
}
@Override
public List put(String key, List value) {
throw new UnsupportedOperationException("modifications are not supported");
}
@Override
public List remove(Object key) {
throw new UnsupportedOperationException("modifications are not supported");
}
@Override
public void putAll(Map> m) {
throw new UnsupportedOperationException("modifications are not supported");
}
@Override
public void clear() {
throw new UnsupportedOperationException("modifications are not supported");
}
@Override
public Set keySet() {
return httpHeaders.names();
}
@Override
public Collection> values() {
return httpHeaders.names().stream().map(k -> Collections.unmodifiableList(httpHeaders.getAll(k))).collect(Collectors.toList());
}
@Override
public Set>> entrySet() {
return httpHeaders.names().stream().map(k -> new AbstractMap.SimpleImmutableEntry<>(k, httpHeaders.getAll(k)))
.collect(Collectors.toSet());
}
}
}