Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.web.client;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.util.Map;

import java.util.Set;

import org.apache.http.HttpResponse;

import org.apache.http.client.ClientProtocolException;

import org.apache.http.client.HttpClient;

import org.apache.http.client.methods.HttpGet;

import org.apache.http.impl.client.HttpClients;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.http.HttpHeaders;

import org.springframework.http.HttpStatus;

import org.springframework.http.MediaType;

import org.springframework.web.util.UriComponentsBuilder;

import com.spice.profitmandi.common.ResponseCodeHolder;

import com.spice.profitmandi.common.enumuration.SchemeType;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.model.ProfitMandiConstants;

public class RestClient {

	private static final Logger LOGGER = LoggerFactory.getLogger(RestClient.class);

	private HttpClient httpClient;

	private SchemeType scheme;

	private String hostName;

	private int port;

	private final String url;

	private HttpHeaders headers;

	public RestClient(SchemeType scheme, String hostName, int port){

		this.scheme = scheme;

		this.hostName = hostName;

		this.port = port;

		this.url = scheme.getValue()==null ? SchemeType.HTTP.toString() : scheme.getValue()  + hostName + ":" + port + "/";

		this.headers = new HttpHeaders();

		this.httpClient = HttpClients.createDefault();

		headers.add("Content-Type", MediaType.APPLICATION_JSON_VALUE);

	}

	public SchemeType getScheme() {

		return scheme;

	}

	public String getHostName() {

		return hostName;

	}

	public int getPort() {

		return port;

	}

	public String getUrl() {

		return url;

	}

	public String get(String uri, Map<String, String> params)

		throws ProfitMandiBusinessException{

		UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(url + uri);

		Set<String> keys = params.keySet();

		for(String key : keys){

			builder.queryParam(key, params.get(key));

		}

		HttpGet request = new HttpGet(builder.build().toUri());

		request.setHeader("Content-Type", MediaType.APPLICATION_JSON_VALUE);

		LOGGER.info("Connecting to google server for token validation at url {}",request.getURI());

		try {

			HttpResponse response = httpClient.execute(request);

			String responseString = this.toString(response.getEntity().getContent());

			LOGGER.info("Got response for token validation with responseCode {} and responseBody {}", response.getStatusLine().getStatusCode(), responseString);

			if(response.getStatusLine().getStatusCode() == HttpStatus.OK.value()){

				return responseString;

			}else{

				throw new ProfitMandiBusinessException(ProfitMandiConstants.TOKEN, params.get(ProfitMandiConstants.ACCESS_TOKEN), "USR_1005");

			}

		} catch (ClientProtocolException e) {

			throw new RuntimeException(ResponseCodeHolder.getMessage("USR_1006"));

		} catch (IOException e) {

			throw new RuntimeException(ResponseCodeHolder.getMessage("USR_1006"));

		}

	}

	private String toString(InputStream inputStream){

		BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));

		StringBuilder responseString = new StringBuilder();

		String line = null;

		try {

			while((line = reader.readLine()) != null){

				responseString.append(line);

			}

			inputStream.close();

		} catch (IOException e) {

			throw new RuntimeException();

		}

		return responseString.toString();

	}

}