Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.datalogger.EventType;

import in.shop2020.model.v1.catalog.Category;

import in.shop2020.serving.cache.EhcacheWrapper.CacheKeys;

import in.shop2020.serving.cache.SnippetCacheWrapper;

import in.shop2020.serving.services.SolrSearchService;

import in.shop2020.serving.utils.Utils;

import in.shop2020.utils.CategoryManager;

import in.shop2020.utils.DataLogger;

import java.io.UnsupportedEncodingException;

import java.net.URLEncoder;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import org.apache.commons.lang.StringUtils;

import org.apache.log4j.Logger;

import org.apache.struts2.convention.annotation.Result;

import com.google.gson.Gson;

/**

 * @author rajveer

 *

 */

@Result(name="redirect", location="${location}", type="redirect")

public class SearchController extends BaseController {

	private static final long serialVersionUID = -8392433517042806559L;

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

	private List<String> results;

	private Map<String, Double> dynamicSearchMap;

	/**

	 * 

	 */

	private Map<String, String> snippets;

	/**

	 * 

	 */

	private Map<String, List<String[]>> facets = null;

	/**

	 * 

	 */

	private Map<String, List<List<String>>> crumbs;

	private String  query;

	private String sortOrder = null;

	private Double minPrice = null;

	private Double maxPrice = null;

	private long windowSize = 20;

	private long page = 1;

	private long totalResults;

	private long beginIndex = 0;

	private String url;

	private String priceUrl = "?";

	private String sortUrl;

	private String location;

	private static HashMap<String, List<String>> categoriesChildren = null;

	private static Map<Long, String> categoryLabelMap = null;

	/**

	 * 

	 * @return

	 * @throws UnsupportedEncodingException

	 */

    // GET /query

    public String index() throws Exception {

		query = this.request.getParameter("q");

		url = "q="+ URLEncoder.encode(query, "UTF-8");

		if(query.trim().isEmpty()){

				location = "/";

				return "redirect";

		}

		sortUrl = url;

		if(this.request.getParameter("sort") != null){

    		url= url + "&sort=" + this.request.getParameter("sort");

    		sortOrder = this.request.getParameter("sort");

    	}

		priceUrl += url;

    	if(this.request.getParameter("page") != null){

    		this.page = Long.parseLong(this.request.getParameter("page"));

    		this.beginIndex = this.windowSize * (this.page-1);

    	}

    	if(this.request.getParameter("minPrice") != null){

    		this.minPrice = (new Double(this.request.getParameter("minPrice")));

    		url= url + "&minPrice=" + this.request.getParameter("minPrice");

    	}

    	if(this.request.getParameter("maxPrice") != null){

    		this.maxPrice = (new Double(this.request.getParameter("maxPrice")));

    		url= url + "&maxPrice=" + this.request.getParameter("maxPrice");

    	}    	

    	String[] fqrys = this.request.getParameterValues("fq");

    	this.crumbs = new HashMap<String, List<List<String>>>();

    	String urlCrumb = "";

    	if(query != null) {

	    	urlCrumb = url; 

	    }else {

    	}

    	if(fqrys != null) {

	    	log.info("fqrys=" + Arrays.toString(fqrys));

	    	String filterUrl = "";

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

	    		urlCrumb += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");

	    		priceUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");

	    		sortUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");

	    		String facetName = StringUtils.split(fqrys[i], ":")[0];

	    		String facetValue = StringUtils.split(fqrys[i], ":")[1];

	    		String facetLabel = Utils.FACET_LABEL_MAP.get(facetName);

	    		filterUrl = "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8"); 

	    		List<String> acrumb = Arrays.asList(facetValue, filterUrl);

	    		if(!crumbs.containsKey(facetLabel)) {

	    			crumbs.put(facetLabel, new ArrayList<List<String>>());

	    		}

	    		this.crumbs.get(facetLabel).add(acrumb);

	    	}

    	}

    	url = urlCrumb;

    	log.info("Queried term:" + query);

    	SolrSearchService search = new SolrSearchService(URLEncoder.encode(query, "UTF-8"), fqrys, (page-1)*windowSize, windowSize, minPrice, maxPrice, sortOrder, sourceId);

    	this.results =  search.getResultMap(); 

    	this.dynamicSearchMap = search.getDynamicPriceMap();

    	// Facets

    	List<String> fqs = null;

    	if (fqrys != null) {

    		fqs = Arrays.asList(fqrys);

    	}else {

    		fqs = new ArrayList<String>();

    	}

		setFacet(fqs, search);

		//If 

    	Map<String, Double> priceMap = search.getPriceStatsMap();

    	if(priceMap != null){

    		if (this.minPrice == null){

    			this.minPrice = priceMap.get("min");

    		}

    		if(this.maxPrice == null) {

    			this.maxPrice = priceMap.get("max");

    		}

    	} else{

    		this.minPrice = 0.0;

    		this.maxPrice = 0.0;

    	}

    	this.totalResults = search.getTotalResults();

        DataLogger.logData(EventType.PRODUCT_SEARCH, getSessionId(), userinfo.getUserId(), userinfo.getEmail(),

                query, Long.toString(Utils.ROOT_CATEGORY), Long.toString(totalResults));

    	return "index";

    }

    private void setFacet(List<String> fqs, SolrSearchService search) throws Exception{

    	this.facets = new LinkedHashMap<String, List<String[]>>();

    	List<String> filtrableFacets = search.getFilterableFacets();

    	for (String filtrableFacet : filtrableFacets) {

    		Map<String, Integer> facetDetailMap = search.getFacetDetails(filtrableFacet);

    		if(facetDetailMap==null)

    			continue;

    		List<String[]> values = new ArrayList<String[]>();

    		for(Entry<String, Integer> facetEntry : facetDetailMap.entrySet()){

    			String selected = "";

    			if(fqs.contains(filtrableFacet + ":" + facetEntry.getKey())) {

    				selected = "checked=\"checked\"";

    			}

    		    String facet = facetEntry.getKey();

      			String drilldownURL = "&fq=" +  

    				URLEncoder.encode( filtrableFacet + ":" + facet, "UTF-8"); 

      			String[] afacet = new String[] { facet, 

      			      facetEntry.getValue().toString(), drilldownURL,  selected};

      			values.add(afacet);    

    		}

    		this.facets.put(Utils.FACET_LABEL_MAP.get(filtrableFacet), values);

    	}

	}

	public Map<String, String> getSnippets() throws Exception {

        if(results != null){

            snippets = new HashMap<String, String>();   

            for(String docId: results){

                String snippet = (String)SnippetCacheWrapper.getSnippet(CacheKeys.CATEGORY_SNIPPET_CACHE_KEY, docId, sourceId);

                if (snippet != null) {

                    snippets.put(docId, snippet);

                }

            }

        }

        return snippets;

    }

    /**

     * 

     * @return

     */

    public String getQuery() {

    	return this.query;

    }

    /**

     * 

     * @return

     */

    public List<String> getResults() {

    	return this.results;

    }

    /**

     * 

     * @return

     */

    public Map<String, List<String[]>> getFacets() {

    	return this.facets;

    }

    public long getTotalResults(){

    	return totalResults;

    }

    public String getUrl(){

    	return this.url;

    }

    public String getPriceUrl(){

    	return this.priceUrl;

    }

    public String getSortUrl(){

    	return this.sortUrl;

    }

	public long getBeginIndex(){

		if(totalResults>0)

			return beginIndex+1;

		return beginIndex;

	}

	public long getTotalPages() {

		return 1 + (totalResults-1)/windowSize;

	}

	public long getCurrentPage() {

		return this.page;

	}

    public Double getMinPrice() {

    	return this.minPrice;

    }

    public Double getMaxPrice() {

    	return this.maxPrice;

    }

    public Map<String, List<List<String>>> getCrumbs() {

    	return this.crumbs;

    }

    public String getSortOrder(){

    	return this.sortOrder;

    }

    public String getLocation(){

    	return this.location;

    }

    public List<String> getChildren(String categoryLabel) {

    	return getCategoriesChildren().get(categoryLabel);

    }

    public static Map<String, List<String>> getCategoriesChildren(){

    	Map<Long,String> idLabelMap = new HashMap<Long, String>(); 

    	if (categoriesChildren == null) {

    		categoriesChildren = new HashMap<String, List<String>>();

    		for (Category category: CategoryManager.getCategoryManager().getCategories().values()){

    			idLabelMap.put(category.getId(), category.getLabel());

    		}

    		for (Category category: CategoryManager.getCategoryManager().getCategories().values()){

    			if(category.getParent_category_id() != 10000 && category.getParent_category_id() != 0 &&

    					category.getParent_category_id() != Utils.MOBILE_PHONES_CATEGORY){

    				String parentLabel = idLabelMap.get(category.getParent_category_id());

    				if(!categoriesChildren.containsKey(parentLabel)){

    					categoriesChildren.put(parentLabel, new ArrayList<String>());

    				}

    				categoriesChildren.get(parentLabel).add(category.getLabel());

    			}

    		}

    	}

    	return categoriesChildren;

    }

    public static String getCategoryLabel(Long id){

    	if(categoryLabelMap == null){

    		Map<Long,String> idLabelMap = new HashMap<Long, String>(); 

    		for (Category category: CategoryManager.getCategoryManager().getCategories().values()){

    			idLabelMap.put(category.getId(), category.getLabel());

    		}

    		categoryLabelMap = idLabelMap;

    	} 

    	return categoryLabelMap.get(id);

    }

	public String getDynamicSearchMap(){

    	if (this.dynamicSearchMap == null) {

    		return "{}";

    	} else {

    		return new Gson().toJson(this.dynamicSearchMap);

    	}

    }

}