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.sumologic.client.util.HttpUtils Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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 com.sumologic.client.util;
import com.fasterxml.jackson.core.JsonGenerationException;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.sumologic.client.*;
import com.sumologic.client.model.HttpDeleteRequest;
import com.sumologic.client.model.HttpGetRequest;
import com.sumologic.client.model.HttpPostRequest;
import com.sumologic.client.model.HttpPutRequest;
import org.apache.http.HttpEntity;
import org.apache.http.client.CookieStore;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.client.config.CookieSpecs;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.*;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import java.io.*;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
public class HttpUtils {
public static final int API_VERSION = 1;
private static final String JSON_CONTENT_TYPE = "application/json";
private final CookieStore cookieStore = new BasicCookieStore();
// Public HTTP request methods
public Response
get(ConnectionConfig config, String endpoint,
Request request, Map requestHeaders,
ResponseHandler handler, int expectedStatusCode) {
return get(config, endpoint, request, Defaults.DEFAULT_HTTP_TIMEOUT, requestHeaders, handler, expectedStatusCode);
}
public Response
get(ConnectionConfig config, String endpoint,
Request request, int timeout, Map requestHeaders,
ResponseHandler handler, int expectedStatusCode) {
try {
URI uri = new URIBuilder()
.setScheme(config.getProtocol())
.setHost(config.getHostname())
.setPort(config.getPort())
.setPath(getEndpointURI(endpoint))
.setParameters(request.toUrlParams())
.setCharset(StandardCharsets.UTF_8)
.build();
HttpGet get = new HttpGet(uri);
configureRequest(config, get, timeout);
return doRequest(config, timeout, get, requestHeaders, request, handler, expectedStatusCode);
} catch (URISyntaxException e) {
throw new SumoClientException("URI cannot be generated", e);
}
}
public Response
post(ConnectionConfig config, String endpoint,
Request request, Map requestHeaders,
ResponseHandler handler, int expectedStatusCode) {
try {
URI uri = new URIBuilder()
.setScheme(config.getProtocol())
.setHost(config.getHostname())
.setPort(config.getPort())
.setPath(getEndpointURI(endpoint))
.setCharset(StandardCharsets.UTF_8)
.build();
HttpPost post = new HttpPost(uri);
configureRequest(config, post, Defaults.DEFAULT_HTTP_TIMEOUT);
String body = JacksonUtils.MAPPER.writeValueAsString(request);
StringEntity entity = new StringEntity(body, StandardCharsets.UTF_8);
entity.setContentType(JSON_CONTENT_TYPE);
post.setEntity(entity);
return doRequest(
config, post, requestHeaders, request, handler, expectedStatusCode);
} catch (URISyntaxException e) {
throw new SumoClientException("URI cannot be generated", e);
} catch (JsonMappingException | JsonGenerationException e) {
throw new SumoClientException("Error generating JSON", e);
} catch (IOException e) {
throw new SumoClientException("Error generating JSON", e);
}
}
public Response
put(ConnectionConfig config, String endpoint,
Request request, ResponseHandler handler, int expectedStatusCode) {
try {
URI uri = new URIBuilder()
.setScheme(config.getProtocol())
.setHost(config.getHostname())
.setPort(config.getPort())
.setPath(getEndpointURI(endpoint))
.build();
HttpPut put = new HttpPut(uri);
Map requestHeaders = new HashMap();
if (request.getETag() != null) {
requestHeaders.put("If-Match", request.getETag());
}
String body = JacksonUtils.MAPPER.writeValueAsString(request);
StringEntity entity = new StringEntity(body, StandardCharsets.UTF_8);
entity.setContentType(JSON_CONTENT_TYPE);
put.setEntity(entity);
configureRequest(config, put, Defaults.DEFAULT_HTTP_TIMEOUT);
return doRequest(
config, put, requestHeaders, request, handler, expectedStatusCode);
} catch (URISyntaxException ex) {
throw new SumoClientException("URI cannot be generated", ex);
} catch (JsonMappingException | JsonGenerationException e) {
throw new SumoClientException("Error generating JSON", e);
} catch (IOException e) {
throw new SumoClientException("Error generating JSON", e);
}
}
public Response
delete(ConnectionConfig config, String endpoint,
Request request, ResponseHandler handler, int expectedStatusCode) {
try {
URI uri = new URIBuilder()
.setScheme(config.getProtocol())
.setHost(config.getHostname())
.setPort(config.getPort())
.setPath(getEndpointURI(endpoint))
.build();
HttpDelete delete = new HttpDelete(uri);
configureRequest(config, delete, Defaults.DEFAULT_HTTP_TIMEOUT);
Map requestHeaders = HttpUtils.toRequestHeaders();
return doRequest(
config, delete, requestHeaders, request, handler, expectedStatusCode);
} catch (URISyntaxException ex) {
throw new SumoClientException("URI cannot be generated", ex);
}
}
public static Map toRequestHeaders(String... parts) {
Map result = new HashMap();
for (int i = 0; i < parts.length; i++) {
result.put(parts[i], parts[++i]);
}
return result;
}
// Private methods
private CloseableHttpClient getHttpClient(ConnectionConfig config) {
CredentialsProvider provider = new BasicCredentialsProvider();
provider.setCredentials(config.getAuthScope(), config.getUsernamePasswordCredentials());
return HttpClients.custom().setDefaultCookieStore(cookieStore)
.setDefaultCredentialsProvider(provider)
.build();
}
private static String getEndpointURI(String endpoint) {
return "/" + UrlParameters.API_SERVICE +
"/" + UrlParameters.VERSION_PREFIX + API_VERSION +
"/" + endpoint;
}
private Response
doRequest(ConnectionConfig config, HttpUriRequest method, Map requestHeaders,
Request request, ResponseHandler handler, int expectedStatusCode) {
return doRequest(config, Defaults.DEFAULT_HTTP_TIMEOUT, method, requestHeaders, request, handler, expectedStatusCode);
}
private Response
doRequest(ConnectionConfig config, int timeout, HttpUriRequest method, Map requestHeaders,
Request request, ResponseHandler handler, int expectedStatusCode) {
// Set headers
for (Map.Entry header : requestHeaders.entrySet()) {
method.setHeader(header.getKey(), header.getValue());
}
CloseableHttpClient httpClient = getHttpClient(config);
InputStream httpStream = null;
CloseableHttpResponse httpResponse = null;
try {
httpResponse = httpClient.execute(method);
HttpEntity entity = httpResponse.getEntity();
httpStream = entity.getContent();
// Request was ok? yes -> handle http response
if (httpResponse.getStatusLine().getStatusCode() == expectedStatusCode) {
return handler.handle(httpResponse, httpStream, request);
}
// no -> get json error and throw exception
else {
StringWriter writer = new StringWriter();
BufferedReader reader = new BufferedReader(new InputStreamReader(httpStream));
// Convert response to JSON string
for (String s; (s = reader.readLine()) != null; ) {
writer.write(s + "
");
}
String json = writer.toString();
if (JacksonUtils.isValidJson(json))
throw new SumoServerException(method.getURI().toString(), writer.toString());
else
throw new SumoServerException(
method.getURI().toString(),
httpResponse.getStatusLine().getStatusCode());
}
}
// Handle IO exceptions
catch (IOException ex) {
throw new SumoClientException("Error reading server response", ex);
}
// Handle runtime exceptions
catch (RuntimeException ex) {
if (ex instanceof SumoServerException) {
throw ex;
} else {
throw new SumoClientException("Runtime error reading server response", ex);
}
}
// Clean-up
finally {
if (httpStream != null) {
try {
httpStream.close();
} catch (Exception ex) {
}
}
if (method != null) {
try {
method.abort();
} catch (Exception ex) {
}
}
try {
httpResponse.close();
} catch (Exception ex) {
}
try {
httpClient.close();
} catch (Exception ex) {
}
}
}
private void configureRequest(ConnectionConfig config, HttpRequestBase request, int timeout) {
RequestConfig.Builder builder = RequestConfig.custom().setConnectTimeout(timeout).setSocketTimeout(timeout);
builder.setCookieSpec(CookieSpecs.STANDARD);
if (config.getProxy() != null) {
builder.setProxy(config.getProxy()).build();
}
request.setConfig(builder.build());
}
}
