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.visenze.visearch.internal.http.ViSearchHttpClientImpl Maven / Gradle / Ivy
package com.visenze.visearch.internal.http;
import com.google.common.base.Strings;
import com.google.common.collect.Maps;
import com.google.common.collect.Multimap;
import com.visenze.visearch.ClientConfig;
import com.visenze.visearch.ResponseMessages;
import com.visenze.visearch.internal.InternalViSearchException;
import com.visenze.visearch.internal.constant.ViSearchHttpConstants;
import org.apache.http.*;
import org.apache.http.auth.AuthenticationException;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.client.methods.RequestBuilder;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.entity.mime.content.InputStreamBody;
import org.apache.http.impl.auth.BasicScheme;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.message.BasicNameValuePair;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
public class ViSearchHttpClientImpl implements ViSearchHttpClient {
public static final Charset UTF8_CHARSET = Charset.forName("utf-8") ;
private final String endpoint;
CloseableHttpClient httpClient;
private final ClientConfig clientConfig;
private final UsernamePasswordCredentials credentials;
public ViSearchHttpClientImpl(String endpoint, String accessKey, String secretKey, CloseableHttpClient httpClient) {
this.endpoint = endpoint;
credentials = new UsernamePasswordCredentials(accessKey, secretKey);
this.httpClient = httpClient;
this.clientConfig = new ClientConfig();
}
public ViSearchHttpClientImpl(String endpoint, String accessKey, String secretKey) {
this(endpoint, accessKey, secretKey, new ClientConfig());
}
public ViSearchHttpClientImpl(String endpoint, String accessKey, String secretKey, ClientConfig clientConfig) {
this.endpoint = endpoint;
this.clientConfig = clientConfig;
RequestConfig conf = RequestConfig
.custom()
.setConnectTimeout(clientConfig.getConnectionTimeout())
.setSocketTimeout(clientConfig.getSocketTimeout())
.build();
credentials = new UsernamePasswordCredentials(accessKey, secretKey);
this.httpClient = HttpClientBuilder
.create()
.setMaxConnTotal(clientConfig.getMaxConnection())
.setMaxConnPerRoute(clientConfig.getMaxConnection())
.setDefaultRequestConfig(conf)
.build();
}
@Override
public UsernamePasswordCredentials getCredentials() {
return credentials;
}
@Override
public ViSearchHttpResponse get(String path, Multimap params) {
HttpUriRequest request = buildGetRequest(endpoint + path, params);
return getResponse(request);
}
@Override
public ViSearchHttpResponse post(String path, Multimap params) {
HttpUriRequest request = buildPostRequest(endpoint + path, params);
return getResponse(request);
}
@Override
public ViSearchHttpResponse postImage(String path, Multimap params, File file) {
HttpUriRequest request = buildPostRequestForImage(endpoint + path, params, file);
return getResponse(request);
}
@Override
public ViSearchHttpResponse postImage(String path, Multimap params, InputStream inputStream, String filename) {
HttpUriRequest request = buildPostRequestForImage(endpoint + path, params, inputStream, filename);
return getResponse(request);
}
@Override
public ViSearchHttpResponse postImFeature(String path, Multimap params, String imFeature, String transId) {
HttpUriRequest request = buildPostRequestForImFeature(endpoint + path, params, imFeature);
if (!Strings.isNullOrEmpty(transId)) {
request.addHeader(ViSearchHttpConstants.TRANS_ID, transId);
}
return getResponse(request);
}
private HttpUriRequest buildGetRequest(String url, Multimap params) {
return RequestBuilder
.get()
.setUri(buildGetUri(url, mapToNameValuePair(params)))
.build();
}
private static URI buildGetUri(String url, List nameValuePairList) {
try {
return new URIBuilder(url).addParameters(nameValuePairList).build();
} catch (URISyntaxException e) {
throw new InternalViSearchException(ResponseMessages.INVALID_ENDPOINT, e);
//throw new ViSearchException("There was an error parsing the ViSearch endpoint. Please ensure " +
// "that your provided ViSearch endpoint is a well-formed URL and try again.", e);
}
}
private static URI buildPostUri(String url) {
try {
return new URIBuilder(url).build();
} catch (URISyntaxException e) {
throw new InternalViSearchException(ResponseMessages.INVALID_ENDPOINT, e);
//throw new ViSearchException("There was an error parsing the ViSearch endpoint. Please ensure " +
// "that your provided ViSearch endpoint is a well-formed URL and try again.", e);
}
}
private HttpUriRequest buildPostRequest(String url, Multimap params) {
return RequestBuilder
.post()
.setUri(buildPostUri(url))
.setEntity(new UrlEncodedFormEntity(mapToNameValuePair(params), Consts.UTF_8))
.setHeader(HttpHeaders.CONTENT_TYPE, ContentType.APPLICATION_FORM_URLENCODED.withCharset(Consts.UTF_8).toString())
.build();
}
private static HttpUriRequest buildMultipartPostRequest(String url, HttpEntity entity) {
HttpPost httpPost = new HttpPost(url);
httpPost.setEntity(entity);
return httpPost;
}
private static HttpUriRequest buildPostRequestForImage(String url, Multimap params, File file) {
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
builder.setCharset(UTF8_CHARSET);
for (Map.Entry entry : params.entries()) {
builder.addTextBody(entry.getKey(), entry.getValue(), ContentType.TEXT_PLAIN);
}
builder.addBinaryBody(ViSearchHttpConstants.IMAGE, file);
HttpEntity entity = builder.build();
return buildMultipartPostRequest(url, entity);
}
private static HttpUriRequest buildPostRequestForImage(String url, Multimap params, InputStream inputStream, String filename) {
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
ContentType contentType = ContentType.create(ContentType.TEXT_PLAIN.getMimeType(), UTF8_CHARSET);
for (Map.Entry entry : params.entries()) {
builder.addTextBody(entry.getKey(), entry.getValue(), contentType);
}
builder.addPart(ViSearchHttpConstants.IMAGE, new InputStreamBody(inputStream, filename));
HttpEntity entity = builder.build();
return buildMultipartPostRequest(url, entity);
}
private HttpUriRequest buildPostRequestForImFeature(String url, Multimap params, String imFeature) {
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
ContentType contentType = ContentType.create(ContentType.TEXT_PLAIN.getMimeType(), UTF8_CHARSET);
for (Map.Entry entry : params.entries()) {
builder.addTextBody(entry.getKey(), entry.getValue(), contentType);
}
builder.addBinaryBody(ViSearchHttpConstants.IM_FEATURE, imFeature.getBytes(), contentType, ViSearchHttpConstants.IM_FEATURE);
HttpEntity entity = builder.build();
return buildMultipartPostRequest(url, entity);
}
private ViSearchHttpResponse getResponse(HttpUriRequest request) {
addAuthHeader(request);
addOtherHeaders(request);
CloseableHttpResponse response = executeRequest(request);
try {
Map headers = Maps.newHashMap();
Header[] responseHeaders = response.getAllHeaders();
if (responseHeaders != null) {
for (Header header : responseHeaders) {
headers.put(header.getName(), header.getValue());
}
}
ViSearchHttpResponse response1 = new ViSearchHttpResponse(response);
response1.setHeaders(headers);
return response1;
} catch (IllegalArgumentException e) {
throw new InternalViSearchException(ResponseMessages.SYSTEM_ERROR, e);
// throw new NetworkException("A network error occurred when reading response from the ViSearch endpoint. " +
// "Please check your network connectivity and try again.", e);
}
}
private void addAuthHeader(HttpUriRequest request) {
try {
request.addHeader(new BasicScheme().authenticate(credentials, request, null));
} catch (AuthenticationException e) {
throw new InternalViSearchException(ResponseMessages.UNAUTHORIZED, e);
// throw new com.visenze.visearch.internal.AuthenticationException("There was an error generating the " +
// "HTTP basic authentication header. Please check your access key and secret key and try again", e);
}
}
private void addOtherHeaders(HttpUriRequest request) {
// add user agent header
String userAgent = clientConfig.getUserAgent();
if (!userAgent.equals(ClientConfig.DEFAULT_USER_AGENT)) {
userAgent += " " + ClientConfig.DEFAULT_USER_AGENT;
}
request.addHeader(HttpHeaders.USER_AGENT, userAgent);
// add x-request-with header
request.addHeader(ViSearchHttpConstants.X_REQUESTED_WITH, clientConfig.DEFAULT_XREQUEST_WITH);
}
private CloseableHttpResponse executeRequest(HttpUriRequest request) {
try {
return httpClient.execute(request);
} catch (IOException e) {
throw new InternalViSearchException(ResponseMessages.NETWORK_ERROR, e);
// throw new NetworkException("A network error occurred when requesting to the ViSearch endpoint. " +
// "Please check your network connectivity and try again.", e);
}
}
private List mapToNameValuePair(Multimap params) {
List pairs = new ArrayList();
for (Map.Entry entry : params.entries()) {
pairs.add(new BasicNameValuePair(entry.getKey(), entry.getValue().toString()));
}
return pairs;
}
}
