Subversion Repositories SmartDukaan

Rev

Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
339 rajveer 1 
/**
2 
 *
3 
 */
4 
package in.shop2020.serving.controllers;
5 
 
517 rajveer 6 
import in.shop2020.serving.pages.PageContentKeys;
7 
import in.shop2020.serving.pages.PageEnum;
8 
import in.shop2020.serving.pages.PageManager;
9 
import in.shop2020.serving.services.PageLoaderHandler;
339 rajveer 10 
import in.shop2020.serving.services.SolrSearchService;
517 rajveer 11 
import in.shop2020.serving.utils.FileUtils;
12 
import in.shop2020.util.Utils;
339 rajveer 13 
 
517 rajveer 14 
import java.io.File;
339 rajveer 15 
import java.io.UnsupportedEncodingException;
16 
import java.net.URLEncoder;
17 
import java.util.ArrayList;
18 
import java.util.Arrays;
354 rajveer 19 
import java.util.HashMap;
339 rajveer 20 
import java.util.LinkedHashMap;
21 
import java.util.List;
22 
import java.util.Map;
536 rajveer 23 
import java.util.StringTokenizer;
339 rajveer 24 
 
25 
import javax.servlet.http.HttpServletRequest;
26 
 
27 
import org.apache.commons.lang.StringUtils;
28 
import org.apache.juli.logging.Log;
29 
import org.apache.juli.logging.LogFactory;
30 
import org.apache.struts2.interceptor.ServletRequestAware;
31 
import org.apache.struts2.rest.DefaultHttpHeaders;
32 
import org.apache.struts2.rest.HttpHeaders;
33 
 
34 
/**
545 rajveer 35 
 * @author rajveer
339 rajveer 36 
 *
37 
 */
517 rajveer 38 
public class SearchController extends BaseController {
339 rajveer 39 
 
40 
	/**
41 
	 *
42 
	 */
43 
	private static Log log = LogFactory.getLog(SearchController.class);
44 
 
517 rajveer 45 
	private Map<String,String> htmlSnippets = new HashMap<String, String>();
339 rajveer 46 
	/**
47 
	 *
48 
	 */
517 rajveer 49 
	private Map<String, String[]> results;
50 
 
339 rajveer 51 
	/**
52 
	 *
53 
	 */
517 rajveer 54 
	private Map<String, String> snippets;
339 rajveer 55 
 
56 
	/**
57 
	 *
58 
	 */
545 rajveer 59 
	private Map<String, List<String[]>> facets = null;
339 rajveer 60 
 
61 
	/**
62 
	 *
63 
	 */
64 
	private List<String[]> crumbs;
65 
 
517 rajveer 66 
	private String  query;
67 
 
545 rajveer 68 
	private Double minPrice = null;
69 
	private Double maxPrice = null;
354 rajveer 70 
 
545 rajveer 71 
	private long windowSize = 10;
72 
	private long page = 1;
73 
	private long totalResults;
74 
	private long beginIndex = 0;
75 
	private String url;
76 
	private String priceUrl;
339 rajveer 77 
	/**
78 
	 *
79 
	 * @return
80 
	 * @throws UnsupportedEncodingException
81 
	 */
82 
    // GET /query
83 
    public HttpHeaders index() throws UnsupportedEncodingException {
517 rajveer 84 
 
545 rajveer 85 
    	// Something we need to figure out
517 rajveer 86 
    	PageLoaderHandler pageLoader = new PageLoaderHandler();
87 
		htmlSnippets.put("HEADER", pageLoader.getHeaderHtml(userinfo.getUserId(), userinfo.isSessionId(), userinfo.getNameOfUser()));
88 
		htmlSnippets.put("MAIN_MENU", pageLoader.getMainMenuHtml());
89 
		htmlSnippets.put("SEARCH_BAR", pageLoader.getSearchBarHtml(userinfo.getTotalItems(), 10000));
90 
		htmlSnippets.put("CUSTOMER_SERVICE", pageLoader.getCustomerServiceHtml());
91 
		htmlSnippets.put("MY_RESEARCH", pageLoader.getMyResearchHtml(userinfo.getUserId(), userinfo.isSessionId()));
92 
		htmlSnippets.put("FOOTER",pageLoader.getFooterHtml());
93 
		htmlSnippets.put("FOOTER",pageLoader.getFooterHtml());
94 
		htmlSnippets.put("SEARCH_HEADER","");
95 
		htmlSnippets.put("SEARCH_DETAILS","");
96 
 
545 rajveer 97 
 
98 
		log.info("this.request=" + this.request);
339 rajveer 99 
 
545 rajveer 100 
    	//url = this.request.getRequestURL();
101 
		query = this.request.getParameter("q");
102 
		url = "q="+ URLEncoder.encode(query, "UTF-8");
103 
		priceUrl = url;
104 
    	if(this.request.getParameter("page") != null){
105 
    		this.page = Long.parseLong(this.request.getParameter("page"));
106 
    		this.beginIndex = this.windowSize * (this.page-1);
536 rajveer 107 
    	}
545 rajveer 108 
    	if(this.request.getParameter("min-price") != null){
109 
    		this.minPrice = (new Double(this.request.getParameter("min-price")));
110 
    		url= url + "&min-price=" + this.request.getParameter("min-price");
111 
    	}
112 
    	if(this.request.getParameter("max-price") != null){
113 
    		this.maxPrice = (new Double(this.request.getParameter("max-price")));
114 
    		url= url + "&max-price=" + this.request.getParameter("max-price");
115 
    	}
339 rajveer 116 
 
117 
    	String[] fqrys = this.request.getParameterValues("fq");
545 rajveer 118 
 
339 rajveer 119 
    	this.crumbs = new ArrayList<String[]>();
120 
 
121 
    	String strCrumb = "";
122 
    	String urlCrumb = "";
517 rajveer 123 
    	if(query != null) {
124 
	    	strCrumb = query;
545 rajveer 125 
	    	urlCrumb = url;
339 rajveer 126 
	    	this.crumbs.add(new String[] { strCrumb, urlCrumb } );
127 
			log.info("acrumb=" +
128 
					Arrays.toString(new String[] { strCrumb, urlCrumb }));
354 rajveer 129 
    	}else {
545 rajveer 130 
 
339 rajveer 131 
    	}
132 
 
133 
    	if(fqrys != null) {
134 
	    	log.info("fqrys=" + Arrays.toString(fqrys));
135 
 
136 
	    	for(int i=0; i<fqrys.length; i++) {
545 rajveer 137 
	    		String facetName = StringUtils.split(fqrys[i], ":")[0];
339 rajveer 138 
	    		String facetValue = StringUtils.split(fqrys[i], ":")[1];
354 rajveer 139 
	//    		facetqueries.put(StringUtils.split(fqrys[i], ":")[0], StringUtils.split(fqrys[i], ":")[1]);
140 
	    		//strCrumb += " " + facetValue;
141 
	    		strCrumb =  facetValue;
339 rajveer 142 
	    		urlCrumb += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");
545 rajveer 143 
	    		priceUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");
339 rajveer 144 
	    		String[] acrumb = new String[] { strCrumb, urlCrumb };
145 
	    		log.info("acrumb=" + Arrays.toString(acrumb));
146 
	    		this.crumbs.add(acrumb);
147 
	    	}
148 
    	}
545 rajveer 149 
    	url = urlCrumb;
354 rajveer 150 
    	/**
545 rajveer 151 
		 * FIXME facet definition ids
354 rajveer 152 
		 */
339 rajveer 153 
    	// Hard coded for now
354 rajveer 154 
//    	long[] facetDefIDs = new long[] {50001, 50003, 50004, 50005, 50006,
155 
//    			50007, 50008, 50009};
339 rajveer 156 
 
517 rajveer 157 
    	String[] facetDefIDs = new String[] {"Category","F_50002","F_50001",  "F_50006", "F_50007" };
158 
 
159 
    	/* All facets. Removing 5 of them
354 rajveer 160 
    	String[] facetDefIDs = new String[] {"Category","F_50002","F_50001", "F_50003", "F_50004", "F_50005", "F_50006",
161 
    			"F_50007", "F_50008", "F_50009"};
517 rajveer 162 
    	*/
163 
 
164 
 
354 rajveer 165 
    	/**
166 
		 * FIXME facet labels
167 
		 */
339 rajveer 168 
    	// Hard-coded for now
517 rajveer 169 
    	/*
354 rajveer 170 
    	String[] facetLabels = new String[] {"Category","Price",
517 rajveer 171 
    	    	"Brand", "Form Factor", "Carry In Pocket", "Cellular Technologies",
172 
    	    	"Data Connectivity", "Camera Resolution", "Built-in Memory",
173 
    	    	"Talk time"
174 
        	};
175 
    	*/
339 rajveer 176 
 
517 rajveer 177 
    	String[] facetLabels = new String[] {"Category","Price",
178 
    	    	"Brand", "Data Connectivity", "Camera Resolution"
179 
        	};
180 
 
545 rajveer 181 
    	SolrSearchService search = new SolrSearchService(query, fqrys, facetDefIDs, (page-1)*windowSize, windowSize, minPrice, maxPrice);
354 rajveer 182 
    	this.results = new LinkedHashMap<String, String[]>();
183 
    	this.results =  search.getResultMap();
339 rajveer 184 
 
545 rajveer 185 
    	// Facets
339 rajveer 186 
 
187 
    	String qryString = this.request.getQueryString();
188 
    	log.info("qryString=" + qryString);
189 
 
545 rajveer 190 
		this.facets = new LinkedHashMap<String, List<String[]>>();
339 rajveer 191 
    	for (int i=0; i<facetDefIDs.length; i++) {
354 rajveer 192 
    		String facetDefID = facetDefIDs[i];
193 
    		String facetLabel = facetLabels[i];
194 
    		HashMap<String, Integer> facetDetailMap = search.getFacetDetails(facetDefID);
195 
    		if(facetDetailMap==null)
196 
    			continue;
197 
    		List<String[]> values = new ArrayList<String[]>();
198 
    		for(String facet: facetDetailMap.keySet()){
545 rajveer 199 
    			String drilldownURL = url;
200 
      			drilldownURL += "&fq=" + facetDefID + ":" +
354 rajveer 201 
    				URLEncoder.encode(facet, "UTF-8");
202 
      			String[] afacet = new String[] { facet,
203 
      					facetDetailMap.get(facet).toString(), drilldownURL  };
204 
      			values.add(afacet);
205 
 
206 
    		}
207 
 
208 
    		this.facets.put(facetLabel, values);
209 
    	}
210 
 
211 
    	Map<String, Double> priceMap = search.getPriceStatsMap();
517 rajveer 212 
    	if(priceMap != null){
213 
    		this.minPrice = priceMap.get("min");
214 
    		this.maxPrice = priceMap.get("max");
215 
    	}else{
216 
    		this.minPrice = 0.0;
217 
    		this.maxPrice = 0.0;
218 
    	}
354 rajveer 219 
 
545 rajveer 220 
    	this.totalResults = search.getTotalResults();
517 rajveer 221 
    	return new DefaultHttpHeaders("index")
339 rajveer 222 
            .disableCaching();
223 
    }
517 rajveer 224 
 
225 
 
226 
    public Map<String, String> getSnippets() throws Exception {
227 
    	if(results != null){
228 
    		snippets = new HashMap<String, String>();
229 
	    	for(String docId: results.keySet()){
230 
				snippets.put(docId, FileUtils.read( Utils.EXPORT_ENTITIES_PATH + docId + File.separator +"SearchSnippet.html"));
231 
			}
232 
    	}
233 
		return snippets;
234 
    }
235 
 
236 
    /**
237 
     *
238 
     * @return
239 
     */
240 
    public String getQuery() {
241 
    	return this.query;
242 
    }
339 rajveer 243 
 
517 rajveer 244 
 
339 rajveer 245 
    /**
246 
     *
247 
     * @return
248 
     */
249 
    public Map<String, String[]> getResults() {
250 
    	return this.results;
251 
    }
252 
 
253 
    /**
254 
     *
255 
     * @return
256 
     */
257 
    public Map<String, List<String[]>> getFacets() {
258 
    	return this.facets;
259 
    }
260 
 
545 rajveer 261 
    public long getTotalResults(){
262 
    	return totalResults;
263 
    }
264 
 
265 
    public String getUrl(){
266 
    	return this.url;
267 
    }
268 
 
269 
    public String getPriceUrl(){
270 
    	return this.priceUrl;
271 
    }
272 
 
273 
	public long getBeginIndex(){
274 
		if(totalResults>0)
275 
			return beginIndex+1;
276 
		return beginIndex;
277 
	}
278 
 
279 
	public long getTotalPages() {
280 
		return 1 + totalResults/windowSize;
281 
	}
282 
 
283 
	public long getCurrentPage() {
284 
		return this.page;
285 
	}
286 
 
354 rajveer 287 
    public Double getMinPrice() {
288 
    	return this.minPrice;
289 
    }
290 
 
291 
    public Double getMaxPrice() {
292 
    	return this.maxPrice;
293 
    }
294 
 
339 rajveer 295 
    /**
296 
     *
297 
     * @return
298 
     */
299 
    public List<String[]> getCrumbs() {
300 
    	return this.crumbs;
301 
    }
302 
 
303 
    /**
304 
     *
305 
     */
517 rajveer 306 
 
307 
    public String getHeaderSnippet(){
308 
		return htmlSnippets.get("HEADER");
339 rajveer 309 
	}
517 rajveer 310 
 
311 
	public String getMainMenuSnippet(){
312 
		return htmlSnippets.get("MAIN_MENU");
313 
	}
314 
 
315 
	public String getSearchBarSnippet(){
316 
		return htmlSnippets.get("SEARCH_BAR");
317 
	}
318 
 
319 
	public String getCustomerServiceSnippet(){
320 
		return htmlSnippets.get("CUSTOMER_SERVICE");
321 
	}
322 
 
323 
	public String getSearchHeaderSnippet(){
324 
		return htmlSnippets.get("SEARCH_HEADER");
325 
	}
326 
 
327 
	public String getSearchDetailsSnippet(){
328 
		return htmlSnippets.get("SEARCH_DETAILS");
329 
	}
330 
 
331 
	public String getMyResearchSnippet(){
332 
		return htmlSnippets.get("MY_RESEARCH");
333 
	}
334 
 
335 
	public String getFooterSnippet(){
336 
		return htmlSnippets.get("FOOTER");
337 
	}
338 
 
339 
	public String getJsFileSnippet(){
340 
		return htmlSnippets.get("JS_FILES");
341 
	}
342 
 
343 
	public String getCssFileSnippet(){
344 
		return htmlSnippets.get("CSS_FILES");
345 
	}
339 rajveer 346 
}