Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.solr;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import org.apache.http.conn.HttpHostConnectException;

import org.json.JSONArray;

import org.json.JSONObject;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.stereotype.Service;

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

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

import com.spice.profitmandi.common.web.client.RestClient;

@Service("solrServiceCommon")

public class SolrService {

	@Value("${new.solr.url}")

	private String solrUrl;

	public String getContent(String queryTerm) throws Exception {

		RestClient rc = new RestClient();

		Map<String, String> params = new HashMap<>();

		List<String> mandatoryQ = new ArrayList<>();

		if (queryTerm != null && !queryTerm.equals("null")) {

			mandatoryQ.add(String.format("+(%s)", queryTerm));

		} else {

			queryTerm = null;

		}

		params.put("q", StringUtils.join(mandatoryQ, " "));

		params.put("fl", "*");

		if (queryTerm == null) {

			params.put("sort", "create_s desc");

		}

		params.put("start", String.valueOf(0));

		params.put("rows", String.valueOf(20));

		params.put("wt", "json");

		String response = null;

		try {

			response = rc.get(SchemeType.HTTP, solrUrl, 8984, "solr/demo/select", params);

		} catch (HttpHostConnectException e) {

			throw new ProfitMandiBusinessException("", "", "Could not connect to host");

		}

		JSONObject solrResponseJSONObj = new JSONObject(response).getJSONObject("response");

		JSONArray docs = solrResponseJSONObj.getJSONArray("docs");

		return docs.toString();

	}

	public Map<Integer, JSONObject> getContentByCatalogIds(List<Integer> catalogIds) throws Exception {

		RestClient rc = new RestClient();

		Map<Integer, JSONObject> documentMap = new HashMap<>();

		Map<String, String> params = new HashMap<>();

		params.put("q", "catalogId_i:" + StringUtils.join(catalogIds, " "));

		params.put("fl", "*");

		params.put("start", String.valueOf(0));

		params.put("rows", String.valueOf(50));

		params.put("wt", "json");

		String response = null;

		try {

			response = rc.get(SchemeType.HTTP, solrUrl, 8984, "solr/demo/select", params);

		} catch (HttpHostConnectException e) {

			throw new ProfitMandiBusinessException("", "", "Could not connect to host");

		}

		JSONObject solrResponseJSONObj = new JSONObject(response).getJSONObject("response");

		JSONArray docs = solrResponseJSONObj.getJSONArray("docs");

		for (int i = 0; i < docs.length(); i++) {

			JSONObject doc = docs.getJSONObject(i);

			documentMap.put(doc.getInt("catalogId_i"), doc);

		}

		return documentMap;

	}

}