com.bld.commons.connection.client.impl.RestClientConnectionImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-rest-connection Show documentation
There is a newer version: 2.0.6
Show newest version
/**
 * @author Francesco Baldi
 * @mail [email protected]
 * @class com.bld.commons.connection.client.impl.RestClientConnectionImpl.java
 */
package com.bld.commons.connection.client.impl;

import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.Proxy.Type;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.client.DefaultResponseErrorHandler;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;

import com.bld.commons.connection.client.RestClientConnection;
import com.bld.commons.connection.model.BasicRequest;
import com.bld.commons.connection.model.MapRequest;
import com.bld.commons.connection.model.ObjectRequest;
import com.bld.commons.connection.utils.RestConnectionMapper;

/**
 * The Class ClientConnectionImpl.
 */
@Component
public class RestClientConnectionImpl implements RestClientConnection {

	/** The port proxy. */
	@Value("${com.dxc.connection.proxy.port:}")
	private Integer portProxy;

	/** The proxy. */
	@Value("${com.dxc.connection.proxy.ip:}")
	private String proxy;
	
	
	/** The Constant HTTP_METHODS. */
	private final static List HTTP_METHODS=Arrays.asList(HttpMethod.POST,HttpMethod.PUT,HttpMethod.PATCH);

	private RestTemplate restTemplate;
	
	


	public RestClientConnectionImpl() {
		super();
		this.restTemplate=this.restTemplate();
	}

	/**
	 * Entity rest template.
	 *
	 * @param  the generic type
	 * @param mapRequest the map request
	 * @param responseClass the response class
	 * @return the t
	 * @throws Exception the exception
	 */
	@Override
	public  T entityRestTemplate(MapRequest mapRequest, Class responseClass) throws Exception {
		MapUtils.unmodifiableMap(mapRequest.getHttpHeaders());
		HttpEntity request = new HttpEntity<>(mapRequest.getHttpHeaders());
		String url=mapRequest.getUrl();
		if(HTTP_METHODS.contains(mapRequest.getMethod())) {
			Object body = RestConnectionMapper.mapToMultiValueMap(mapRequest.getData());
			request = new HttpEntity<>(body, mapRequest.getHttpHeaders());
		}else {
			UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(mapRequest.getUrl());
			RestConnectionMapper.builderQuery(builder, mapRequest.getData());
			url = builder.toUriString();
		}
		RestBuilder restBuilder = new RestBuilder(url, request);
		ResponseEntity response=this.getResponseEntity(mapRequest, restBuilder, responseClass);
		return response.getBody();
	}
	
	/**
	 * Entity rest template.
	 *
	 * @param  the generic type
	 * @param objectRequest the object request
	 * @param responseClass the response class
	 * @return the t
	 * @throws Exception the exception
	 */
	@Override
	public  T entityRestTemplate(ObjectRequest objectRequest, Class responseClass) throws Exception {
		HttpEntity request = new HttpEntity<>(objectRequest.getData(), objectRequest.getHttpHeaders());
		RestBuilder restBuilder = new RestBuilder(objectRequest.getUrl(), request);
		ResponseEntity response=this.getResponseEntity(objectRequest, restBuilder, responseClass);
		return response.getBody();
	}
	
	/**
	 * List rest template.
	 *
	 * @param  the generic type
	 * @param mapRequest the map request
	 * @param responseClass the response class
	 * @return the list
	 * @throws Exception the exception
	 */
	@Override
	public  List listRestTemplate(MapRequest mapRequest, Class responseClass) throws Exception {
		MapUtils.unmodifiableMap(mapRequest.getHttpHeaders());
		HttpEntity request = new HttpEntity<>(mapRequest.getHttpHeaders());
		String url=mapRequest.getUrl();
		if(HTTP_METHODS.contains(mapRequest.getMethod())) {
			Object body = RestConnectionMapper.mapToMultiValueMap(mapRequest.getData());
			request = new HttpEntity<>(body, mapRequest.getHttpHeaders());
		}else {
			UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(mapRequest.getUrl());
			RestConnectionMapper.builderQuery(builder, mapRequest.getData());
			url = builder.toUriString();
		}
		RestBuilder restBuilder = new RestBuilder(url, request);
		
		ResponseEntity response=this.getResponseEntity(mapRequest, restBuilder, responseClass);
		return Arrays.asList(response.getBody());
	}
	
	/**
	 * List rest template.
	 *
	 * @param  the generic type
	 * @param objectRequest the object request
	 * @param responseClass the response class
	 * @return the list
	 * @throws Exception the exception
	 */
	@Override
	public  List listRestTemplate(ObjectRequest objectRequest, Class responseClass) throws Exception {
		HttpEntity request = new HttpEntity<>(objectRequest.getData(), objectRequest.getHttpHeaders());
		RestBuilder restBuilder = new RestBuilder(objectRequest.getUrl(), request);
		ResponseEntity response=this.getResponseEntity(objectRequest, restBuilder, responseClass);
		return Arrays.asList(response.getBody());
	}
	
	
	/**
	 * Gets the response entity.
	 *
	 * @param  the generic type
	 * @param basicRequest the basic request
	 * @param restBuilder the rest builder
	 * @param responseClass the response class
	 * @return the response entity
	 */
	private  ResponseEntity getResponseEntity(BasicRequest basicRequest,RestBuilder restBuilder,Class responseClass) {
		this.setTimeout(basicRequest);
		return this.restTemplate.exchange(restBuilder.getUrl(), basicRequest.getMethod(), restBuilder.getRequest(), responseClass, basicRequest.getUriParams());
	}
	


	/**
	 * Sets the timeout.
	 *
	 * @param  the key type
	 * @param basicRequest the new timeout
	 */
	private  void setTimeout(BasicRequest basicRequest) {
		restTemplate.setRequestFactory(new SimpleClientHttpRequestFactory());
		if (basicRequest.getTimeout() != null) {
			SimpleClientHttpRequestFactory rf = (SimpleClientHttpRequestFactory) restTemplate.getRequestFactory();
			rf.setReadTimeout(basicRequest.getTimeout());
			rf.setConnectTimeout(basicRequest.getTimeout());
		}

	}
	
	private RestTemplate restTemplate() {
		RestTemplate restTemplate = null;
		if (StringUtils.isNotEmpty(this.proxy) && this.portProxy != null) {
			SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
			Proxy proxy = new Proxy(Type.HTTP, new InetSocketAddress(this.proxy, this.portProxy));
			requestFactory.setProxy(proxy);
			restTemplate = new RestTemplate(requestFactory);
		} else {
			restTemplate = new RestTemplate();
			restTemplate.setErrorHandler(new DefaultResponseErrorHandler());
		}
		return restTemplate;
	}
	

	/**
	 * The Class RestBuilder.
	 */
	private class RestBuilder {

		/** The url. */
		private String url;

		/** The request. */
		private HttpEntity request;

		/**
		 * Instantiates a new rest builder.
		 *
		 * @param url the url
		 * @param request the request
		 */
		private RestBuilder(String url, HttpEntity request) {
			super();
			this.url = url;
			this.request = request;
		}

		/**
		 * Gets the url.
		 *
		 * @return the url
		 */
		private String getUrl() {
			return url;
		}

		/**
		 * Gets the request.
		 *
		 * @return the request
		 */
		private HttpEntity getRequest() {
			return request;
		}

	}

}