All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.arangodb.http.HttpConnection Maven / Gradle / Ivy
/*
* DISCLAIMER
*
* Copyright 2017 ArangoDB GmbH, Cologne, Germany
*
* 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.
*
* Copyright holder is ArangoDB GmbH, Cologne, Germany
*/
package com.arangodb.http;
import com.arangodb.*;
import com.arangodb.config.HostDescription;
import com.arangodb.internal.config.ArangoConfig;
import com.arangodb.internal.net.Connection;
import com.arangodb.internal.serde.ContentTypeFactory;
import com.arangodb.internal.util.EncodeUtils;
import com.arangodb.internal.InternalRequest;
import com.arangodb.internal.RequestType;
import com.arangodb.internal.InternalResponse;
import com.arangodb.shaded.netty.handler.ssl.ApplicationProtocolConfig;
import com.arangodb.shaded.netty.handler.ssl.ClientAuth;
import com.arangodb.shaded.netty.handler.ssl.IdentityCipherSuiteFilter;
import com.arangodb.shaded.netty.handler.ssl.JdkSslContext;
import com.arangodb.shaded.vertx.core.Vertx;
import com.arangodb.shaded.vertx.core.VertxOptions;
import com.arangodb.shaded.vertx.core.buffer.Buffer;
import com.arangodb.shaded.vertx.core.http.HttpHeaders;
import com.arangodb.shaded.vertx.core.http.HttpMethod;
import com.arangodb.shaded.vertx.core.http.HttpVersion;
import com.arangodb.shaded.vertx.core.net.JdkSSLEngineOptions;
import com.arangodb.shaded.vertx.core.spi.tls.SslContextFactory;
import com.arangodb.shaded.vertx.ext.auth.authentication.TokenCredentials;
import com.arangodb.shaded.vertx.ext.auth.authentication.UsernamePasswordCredentials;
import com.arangodb.shaded.vertx.ext.web.client.HttpRequest;
import com.arangodb.shaded.vertx.ext.web.client.HttpResponse;
import com.arangodb.shaded.vertx.ext.web.client.WebClient;
import com.arangodb.shaded.vertx.ext.web.client.WebClientOptions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.net.ssl.SSLContext;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;
import java.util.Iterator;
import java.util.Map.Entry;
import java.util.Optional;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.concurrent.atomic.AtomicInteger;
/**
* @author Mark Vollmary
* @author Michele Rastelli
*/
public class HttpConnection implements Connection {
private static final Logger LOGGER = LoggerFactory.getLogger(HttpConnection.class);
private static final String CONTENT_TYPE_APPLICATION_JSON_UTF8 = "application/json; charset=utf-8";
private static final String CONTENT_TYPE_VPACK = "application/x-velocypack";
private static final String USER_AGENT = getUserAgent();
private static final AtomicInteger THREAD_COUNT = new AtomicInteger();
private final ContentType contentType;
private String auth;
private final WebClient client;
private final Integer timeout;
private final Vertx vertx;
private static String getUserAgent() {
return "JavaDriver/" + PackageVersion.VERSION + " (JVM/" + System.getProperty("java.specification.version") + ")";
}
HttpConnection(final ArangoConfig config, final HostDescription host) {
super();
Protocol protocol = config.getProtocol();
contentType = ContentTypeFactory.of(protocol);
timeout = config.getTimeout();
vertx = Vertx.vertx(new VertxOptions().setPreferNativeTransport(true).setEventLoopPoolSize(1));
vertx.runOnContext(e -> {
Thread.currentThread().setName("adb-eventloop-" + THREAD_COUNT.getAndIncrement());
auth = new UsernamePasswordCredentials(
config.getUser(), Optional.ofNullable(config.getPassword()).orElse("")
).toHttpAuthorization();
LOGGER.debug("Created Vert.x context");
});
int intTtl = Optional.ofNullable(config.getConnectionTtl())
.map(ttl -> Math.toIntExact(ttl / 1000))
.orElse(0);
HttpVersion httpVersion = protocol == Protocol.HTTP_JSON || protocol == Protocol.HTTP_VPACK ?
HttpVersion.HTTP_1_1 : HttpVersion.HTTP_2;
WebClientOptions webClientOptions = new WebClientOptions()
.setMaxPoolSize(1)
.setHttp2MaxPoolSize(1)
.setConnectTimeout(timeout)
.setIdleTimeoutUnit(TimeUnit.MILLISECONDS)
.setIdleTimeout(timeout)
.setKeepAliveTimeout(intTtl)
.setHttp2KeepAliveTimeout(intTtl)
.setUserAgentEnabled(false)
.setFollowRedirects(false)
.setLogActivity(true)
.setKeepAlive(true)
.setTcpKeepAlive(true)
.setPipelining(true)
.setReuseAddress(true)
.setReusePort(true)
.setHttp2ClearTextUpgrade(false)
.setProtocolVersion(httpVersion)
.setDefaultHost(host.getHost())
.setDefaultPort(host.getPort());
if (Boolean.TRUE.equals(config.getUseSsl())) {
SSLContext ctx;
if (config.getSslContext() != null) {
ctx = config.getSslContext();
} else {
try {
ctx = SSLContext.getDefault();
} catch (NoSuchAlgorithmException e) {
throw new ArangoDBException(e);
}
}
webClientOptions
.setSsl(true)
.setUseAlpn(true)
.setVerifyHost(config.getVerifyHost())
.setJdkSslEngineOptions(new JdkSSLEngineOptions() {
@Override
public JdkSSLEngineOptions copy() {
return this;
}
@Override
public SslContextFactory sslContextFactory() {
return () -> new JdkSslContext(
ctx,
true,
null,
IdentityCipherSuiteFilter.INSTANCE,
ApplicationProtocolConfig.DISABLED,
ClientAuth.NONE,
null,
false
);
}
});
}
client = WebClient.create(vertx, webClientOptions);
}
private static String buildUrl(final InternalRequest request) {
StringBuilder sb = new StringBuilder();
String dbName = request.getDbName();
if (dbName != null && !dbName.isEmpty()) {
sb.append("/_db/").append(EncodeUtils.encodeURIComponent(dbName));
}
sb.append(request.getPath());
if (!request.getQueryParam().isEmpty()) {
sb.append("?");
for (Iterator> iterator = request.getQueryParam().entrySet().iterator(); iterator.hasNext(); ) {
Entry param = iterator.next();
if (param.getValue() != null) {
sb.append(EncodeUtils.encodeURIComponent(param.getKey()));
sb.append("=");
sb.append(EncodeUtils.encodeURIComponent(param.getValue()));
if (iterator.hasNext()) {
sb.append("&");
}
}
}
}
return sb.toString();
}
private static void addHeader(final InternalRequest request, final HttpRequest httpRequest) {
for (final Entry header : request.getHeaderParam().entrySet()) {
httpRequest.putHeader(header.getKey(), header.getValue());
}
}
@Override
public void close() {
client.close();
vertx.close();
}
private HttpMethod requestTypeToHttpMethod(RequestType requestType) {
switch (requestType) {
case POST:
return HttpMethod.POST;
case PUT:
return HttpMethod.PUT;
case PATCH:
return HttpMethod.PATCH;
case DELETE:
return HttpMethod.DELETE;
case HEAD:
return HttpMethod.HEAD;
case GET:
default:
return HttpMethod.GET;
}
}
public InternalResponse execute(final InternalRequest request) throws IOException {
CompletableFuture rfuture = new CompletableFuture<>();
vertx.runOnContext(e -> doExecute(request, rfuture));
InternalResponse resp;
try {
resp = rfuture.get();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
throw ArangoDBException.wrap(e);
} catch (ExecutionException e) {
Throwable cause = e.getCause();
if (cause instanceof TimeoutException) {
throw ArangoDBException.wrap(cause);
} else if (cause instanceof IOException) {
throw (IOException) cause;
} else {
throw new IOException(cause);
}
}
return resp;
}
public void doExecute(final InternalRequest request, final CompletableFuture rfuture) {
String path = buildUrl(request);
HttpRequest httpRequest = client
.request(requestTypeToHttpMethod(request.getRequestType()), path)
.timeout(timeout);
if (contentType == ContentType.VPACK) {
httpRequest.putHeader("Accept", CONTENT_TYPE_VPACK);
}
addHeader(request, httpRequest);
httpRequest.putHeader(HttpHeaders.AUTHORIZATION.toString(), auth);
httpRequest.putHeader("x-arango-driver", USER_AGENT);
byte[] reqBody = request.getBody();
Buffer buffer;
if (reqBody != null) {
buffer = Buffer.buffer(reqBody);
if (contentType == ContentType.VPACK) {
httpRequest.putHeader(HttpHeaders.CONTENT_TYPE.toString(), CONTENT_TYPE_VPACK);
} else {
httpRequest.putHeader(HttpHeaders.CONTENT_TYPE.toString(), CONTENT_TYPE_APPLICATION_JSON_UTF8);
}
} else {
buffer = Buffer.buffer();
}
httpRequest.sendBuffer(buffer)
.map(this::buildResponse)
.onSuccess(rfuture::complete)
.onFailure(rfuture::completeExceptionally);
}
private InternalResponse buildResponse(final HttpResponse httpResponse) {
final InternalResponse response = new InternalResponse();
response.setResponseCode(httpResponse.statusCode());
Buffer body = httpResponse.body();
if (body != null) {
byte[] bytes = body.getBytes();
if (bytes.length > 0) {
response.setBody(bytes);
}
}
for (Entry header : httpResponse.headers()) {
response.putMeta(header.getKey(), header.getValue());
}
return response;
}
@Override
public void setJwt(String jwt) {
if (jwt != null) {
vertx.runOnContext(e -> auth = new TokenCredentials(jwt).toHttpAuthorization());
}
}
}
