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;
637 rajveer 12
import in.shop2020.serving.utils.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[]> results;
46
 
339 rajveer 47
	/**
48
	 * 
49
	 */
517 rajveer 50
	private Map<String, String> snippets;
339 rajveer 51
 
52
	/**
53
	 * 
54
	 */
545 rajveer 55
	private Map<String, List<String[]>> facets = null;
339 rajveer 56
 
57
	/**
58
	 * 
59
	 */
60
	private List<String[]> crumbs;
61
 
517 rajveer 62
	private String  query;
569 rajveer 63
	private String sortOrder = null;
545 rajveer 64
	private Double minPrice = null;
65
	private Double maxPrice = null;
354 rajveer 66
 
545 rajveer 67
	private long windowSize = 10;
68
	private long page = 1;
69
	private long totalResults;
70
	private long beginIndex = 0;
71
	private String url;
72
	private String priceUrl;
569 rajveer 73
	private long categoryId=10000;
74
	private String sortUrl;
339 rajveer 75
	/**
76
	 * 
77
	 * @return
78
	 * @throws UnsupportedEncodingException
79
	 */
80
    // GET /query
81
    public HttpHeaders index() throws UnsupportedEncodingException {
517 rajveer 82
 
650 rajveer 83
    	log.info("this.request=" + this.request);
637 rajveer 84
 
545 rajveer 85
    	//url = this.request.getRequestURL();
86
		query = this.request.getParameter("q");
87
		url = "q="+ URLEncoder.encode(query, "UTF-8");
569 rajveer 88
 
89
 
90
		if(this.request.getParameter("category") != null){
91
    		this.categoryId = Long.parseLong(this.request.getParameter("category"));
92
    		url= url + "&category=" + this.request.getParameter("category");
93
    	}
94
		sortUrl = url;
95
		if(this.request.getParameter("sort") != null){
96
    		url= url + "&sort=" + this.request.getParameter("sort");
97
    		sortOrder = this.request.getParameter("sort");
98
    	}
545 rajveer 99
		priceUrl = url;
100
    	if(this.request.getParameter("page") != null){
101
    		this.page = Long.parseLong(this.request.getParameter("page"));
102
    		this.beginIndex = this.windowSize * (this.page-1);
536 rajveer 103
    	}
545 rajveer 104
    	if(this.request.getParameter("min-price") != null){
105
    		this.minPrice = (new Double(this.request.getParameter("min-price")));
106
    		url= url + "&min-price=" + this.request.getParameter("min-price");
107
    	}
108
    	if(this.request.getParameter("max-price") != null){
109
    		this.maxPrice = (new Double(this.request.getParameter("max-price")));
110
    		url= url + "&max-price=" + this.request.getParameter("max-price");
111
    	}    	
339 rajveer 112
 
569 rajveer 113
 
339 rajveer 114
    	String[] fqrys = this.request.getParameterValues("fq");
545 rajveer 115
 
339 rajveer 116
    	this.crumbs = new ArrayList<String[]>();
117
 
569 rajveer 118
    	String[] facetDefIDs = new String[] {"F_50010","F_50011","F_50002","F_50001",  "F_50006", "F_50007" };
583 rajveer 119
    	String[] facetLabels = new String[] {"Category","Sub Category","Price",
569 rajveer 120
    	    	"Brand", "Data Connectivity", "Camera Resolution"
121
        	};
122
 
123
    	String urlCrumb = "";
124
    	if(query != null) {
125
	    	urlCrumb = url; 
126
	    }else {
127
 
128
    	}
129
 
339 rajveer 130
    	if(fqrys != null) {
131
	    	log.info("fqrys=" + Arrays.toString(fqrys));
569 rajveer 132
	    	List<String> arrList = Arrays.asList(facetDefIDs);
133
	    	for(int i=0; i<fqrys.length; i++){
134
	    		urlCrumb += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");
135
	    		priceUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");
136
	    		sortUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");
137
	    		String filterUrl = url;
545 rajveer 138
	    		String facetName = StringUtils.split(fqrys[i], ":")[0];
339 rajveer 139
	    		String facetValue = StringUtils.split(fqrys[i], ":")[1];
569 rajveer 140
	    		String facetLabel = facetLabels[arrList.indexOf(facetName)];  
141
	    		for(int j=0; j<fqrys.length; j++) {
142
	    			if(i==j){
143
	    				continue;
144
	    			}
145
	    			filterUrl += "&fq=" + URLEncoder.encode(fqrys[j], "UTF-8");
146
		    	}
147
	    		String[] acrumb = new String[] { facetLabel, facetValue, filterUrl };
148
	    		log.info("acrumb=" + Arrays.toString(acrumb));
149
	    		this.crumbs.add(acrumb);
150
	    	}
151
    	}
152
    	url = urlCrumb;
517 rajveer 153
 
154
 
339 rajveer 155
 
569 rajveer 156
    	SolrSearchService search = new SolrSearchService(query, fqrys, facetDefIDs, (page-1)*windowSize, windowSize, minPrice, maxPrice, categoryId, sortOrder);
354 rajveer 157
    	this.results = new LinkedHashMap<String, String[]>();
158
    	this.results =  search.getResultMap(); 
339 rajveer 159
 
545 rajveer 160
    	// Facets
339 rajveer 161
 
162
    	String qryString = this.request.getQueryString();
163
    	log.info("qryString=" + qryString);
164
 
545 rajveer 165
		this.facets = new LinkedHashMap<String, List<String[]>>();
339 rajveer 166
    	for (int i=0; i<facetDefIDs.length; i++) {
354 rajveer 167
    		String facetDefID = facetDefIDs[i];
168
    		String facetLabel = facetLabels[i];
169
    		HashMap<String, Integer> facetDetailMap = search.getFacetDetails(facetDefID);
170
    		if(facetDetailMap==null)
171
    			continue;
172
    		List<String[]> values = new ArrayList<String[]>();
173
    		for(String facet: facetDetailMap.keySet()){
545 rajveer 174
    			String drilldownURL = url;  
175
      			drilldownURL += "&fq=" + facetDefID + ":" + 
354 rajveer 176
    				URLEncoder.encode(facet, "UTF-8"); 
177
      			String[] afacet = new String[] { facet, 
178
      					facetDetailMap.get(facet).toString(), drilldownURL  };
179
      			values.add(afacet);    
180
 
181
    		}
182
 
183
    		this.facets.put(facetLabel, values);
184
    	}
185
 
186
    	Map<String, Double> priceMap = search.getPriceStatsMap();
517 rajveer 187
    	if(priceMap != null){
188
    		this.minPrice = priceMap.get("min");
189
    		this.maxPrice = priceMap.get("max");
190
    	}else{
191
    		this.minPrice = 0.0;
192
    		this.maxPrice = 0.0;
193
    	}
354 rajveer 194
 
545 rajveer 195
    	this.totalResults = search.getTotalResults();
517 rajveer 196
    	return new DefaultHttpHeaders("index")
339 rajveer 197
            .disableCaching();
198
    }
517 rajveer 199
 
200
 
201
    public Map<String, String> getSnippets() throws Exception {
202
    	if(results != null){
203
    		snippets = new HashMap<String, String>();	
204
	    	for(String docId: results.keySet()){
205
				snippets.put(docId, FileUtils.read( Utils.EXPORT_ENTITIES_PATH + docId + File.separator +"SearchSnippet.html"));
206
			}
207
    	}
208
		return snippets;
209
    }
210
 
211
    /**
212
     * 
213
     * @return
214
     */
215
    public String getQuery() {
216
    	return this.query;
217
    }
339 rajveer 218
 
517 rajveer 219
 
339 rajveer 220
    /**
221
     * 
222
     * @return
223
     */
224
    public Map<String, String[]> getResults() {
225
    	return this.results;
226
    }
227
 
228
    /**
229
     * 
230
     * @return
231
     */
232
    public Map<String, List<String[]>> getFacets() {
233
    	return this.facets;
234
    }
235
 
545 rajveer 236
    public long getTotalResults(){
237
    	return totalResults;
238
    }
239
 
240
    public String getUrl(){
241
    	return this.url;
242
    }
243
 
244
    public String getPriceUrl(){
245
    	return this.priceUrl;
246
    }
247
 
569 rajveer 248
    public String getSortUrl(){
249
    	return this.sortUrl;
250
    }
251
 
545 rajveer 252
	public long getBeginIndex(){
253
		if(totalResults>0)
254
			return beginIndex+1;
255
		return beginIndex;
256
	}
257
 
258
	public long getTotalPages() {
259
		return 1 + totalResults/windowSize;
260
	}
261
 
262
	public long getCurrentPage() {
263
		return this.page;
264
	}
265
 
354 rajveer 266
    public Double getMinPrice() {
267
    	return this.minPrice;
268
    }
269
 
270
    public Double getMaxPrice() {
271
    	return this.maxPrice;
272
    }
273
 
339 rajveer 274
    public List<String[]> getCrumbs() {
275
    	return this.crumbs;
276
    }
569 rajveer 277
 
339 rajveer 278
}