Subversion Repositories SmartDukaan

package in.shop2020.dtrapi.services;

import in.shop2020.config.ConfigException;

import in.shop2020.thrift.clients.config.ConfigClient;

import java.io.IOException;

import java.io.InputStream;

import java.net.MalformedURLException;

import java.net.URISyntaxException;

import java.net.URL;

import org.apache.commons.io.IOUtils;

import org.apache.http.client.utils.URIBuilder;

import org.apache.log4j.Logger;

public class SolrService{

	private static URIBuilder autoSuggestUrl;

	private static URIBuilder generalSearchUrl;

	private static String SOLR_URL;

	private static Logger log = Logger.getLogger(Class.class);

	static{

		synchronized(SolrService.class){

			try {

				SOLR_URL = ConfigClient.getClient().get("dtr_solr_url");

			} catch (ConfigException e) {

				log.error("Error while gettting dtr_solr_url param from config service", e);

				SOLR_URL = "http://localhost:8983/solr/collection1/select";

			}

			try {

				autoSuggestUrl = new URIBuilder(SOLR_URL);

			} catch (URISyntaxException e) {

				log.error("Error while building solr_url "+e);

			}

			autoSuggestUrl.addParameter("fl", "title,subCategoryId,category_id,category,subCategory,score");  //Fields to choose

			autoSuggestUrl.addParameter("wt", "json"); 					  //Output format

			autoSuggestUrl.addParameter("rows", "20");             //search limit for each grouped field or query

			generalSearchUrl.addParameter("fl", "title,skuBundleId,id,thumbnail");

			generalSearchUrl.addParameter("wt", "json");

			autoSuggestUrl.addParameter("rows", "20");

		}

	}

	public String getSuggestions(String search_text) throws URISyntaxException, IOException{

		autoSuggestUrl.setParameter("q", "suggest:("+search_text+")");

		URL url = autoSuggestUrl.build().toURL();

		log.info("Search Url "+url.toString());

		InputStream is = url.openStream();

		String jsonString;

		try{

			jsonString = IOUtils.toString(is, "UTF-8");

		}

		finally{

			is.close();

		}

		return jsonString; 

	}

	public String getSearchResults(String search_text, String offset) throws URISyntaxException, IOException{

		generalSearchUrl.setParameter("q", search_text);

		if (offset!=null && !offset.isEmpty()){

			generalSearchUrl.setParameter("start", offset);

		}

		URL url = generalSearchUrl.build().toURL();

		log.info("Search Url "+url.toString());

		InputStream is = url.openStream();

		String jsonString;

		try{

			jsonString = IOUtils.toString(is, "UTF-8");

		}

		finally{

			is.close();

		}

		return jsonString; 

	}

}